//?./utils/vlog
package?vlog
import?(
????"github.com/sirupsen/logrus"
????"os"
)
var?ErrorLog?*logrus.Logger
var?AccessLog?*logrus.Logger
var?errorLogFile?=?"./tmp/log/error.log"
var?accessLogFile?=?"./tmp/log/access.log"
func?init?()?{
????initErrorLog()
????initAccessLog()
}
func?initErrorLog()?{
????ErrorLog?=?logrus.New()
????ErrorLog.SetFormatter(&logrus.JSONFormatter{})
????file?,?err?:=?os.OpenFile(errorLogFile,?os.O_RDWR?|?os.O_CREATE?|?os.O_APPEND,?0755)
????if?err?!=?nil?{
????????panic(err)
????}
????ErrorLog.SetOutput(file)
}
func?initAccessLog()?{
????AccessLog?=?logrus.New()
????AccessLog.SetFormatter(&logrus.JSONFormatter{})
????file?,?err?:=?os.OpenFile(accessLogFile,?os.O_RDWR?|?os.O_CREATE?|?os.O_APPEND,?0755)
????if?err?!=?nil?{
????????panic(err)
????}
????AccessLog.SetOutput(file)
}

type?Server?struct?{
???Addr????string??//?TCP?address?to?listen?on,?":http"?if?empty
???Handler?Handler?//?handler?to?invoke,?http.DefaultServeMux?if?nil
???...
???//?ErrorLog?specifies?an?optional?logger?for?errors?accepting
???//?connections,?unexpected?behavior?from?handlers,?and
???//?underlying?FileSystem?errors.
???//?If?nil,?logging?is?done?via?the?log?package's?standard?logger.
???ErrorLog?*log.Logger
?????...
}

func?main()?{
??...
? ??//?將logrus的Logger轉(zhuǎn)換為io.Writer
????errorWriter?:=?vlog.ErrorLog.Writer()
? ??//?記得關(guān)閉io.Writer
????defer?errorWriter.Close()
????server?:=?&http.Server{
????????Addr:????":8080",
????????Handler:?muxRouter,
? ? ? ??//?用記錄器轉(zhuǎn)換成的io.Writer創(chuàng)建log.Logger
????????ErrorLog:?log.New(vlog.ErrorLog.Writer(),?"",?0),
????}
????...
}

func?(*HelloHandler)?ServeHTTP(w?http.ResponseWriter,?r?*http.Request)?{
???ints?:=?[]int{0,?1,?2}
???fmt.Fprintf(w,?"%v",?ints[0:5])
}

type?ResponseWriter?interface?{
????Header()?Header
????Write([]byte)?(int,?error)
????WriteHeader(statusCode?int)
}

type?ResponseWithRecorder?struct?{
???http.ResponseWriter
???statusCode?int
???body?bytes.Buffer
}
func?(rec?*ResponseWithRecorder)?WriteHeader(statusCode?int)?{
???rec.ResponseWriter.WriteHeader(statusCode)
???rec.statusCode?=?statusCode
}
func?(rec?*ResponseWithRecorder)?Write(d?[]byte)?(n?int,?err?error)?{
???n,?err?=?rec.ResponseWriter.Write(d)
???if?err?!=?nil?{
??????return
???}
???rec.body.Write(d)
???return
}

func?AccessLogging?(f?http.Handler)?http.Handler?{
????//?創(chuàng)建一個(gè)新的handler包裝http.HandlerFunc
????return?http.HandlerFunc(func(w?http.ResponseWriter,?r?*http.Request)?{
????????buf?:=?new(bytes.Buffer)
????????buf.ReadFrom(r.Body)
????????logEntry?:=?vlog.AccessLog.WithFields(logrus.Fields{
????????????"ip":?r.RemoteAddr,
????????????"method":?r.Method,
????????????"path":?r.RequestURI,
????????????"query":?r.URL.RawQuery,
????????????"request_body":?buf.String(),
????????})
????????wc?:=?&ResponseWithRecorder{
????????????ResponseWriter:?w,
????????????statusCode:?http.StatusOK,
????????????body:?bytes.Buffer{},
????????}
????????//?調(diào)用下一個(gè)中間件或者最終的handler處理程序
????????f.ServeHTTP(wc,?r)
????????defer?logEntry.WithFields(logrus.Fields{
????????????"status":?wc.statusCode,
????????????"response_body":?wc.body.String(),
????????}).Info()
????})
}

//?router/router.go
func?RegisterRoutes(r?*mux.Router)?{
????...
????//?apply?Logging?middleware
????r.Use(middleware.Logging(),?middleware.AccessLogging)
????...
}