103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"service/library/logger"
|
|
"time"
|
|
)
|
|
|
|
var defaultLogFormatter = func(param gin.LogFormatterParams) string {
|
|
if param.Latency > time.Minute {
|
|
// Truncate in a golang < 1.8 safe way
|
|
param.Latency = param.Latency - param.Latency%time.Second
|
|
}
|
|
return fmt.Sprintf("|%3d|%13v|%15s|%-7s %#v\t%s",
|
|
param.StatusCode,
|
|
param.Latency,
|
|
param.ClientIP,
|
|
param.Method,
|
|
param.Path,
|
|
param.ErrorMessage,
|
|
)
|
|
}
|
|
|
|
type bodyLogWriter struct {
|
|
gin.ResponseWriter
|
|
body *bytes.Buffer
|
|
}
|
|
|
|
func (w bodyLogWriter) Write(b []byte) (int, error) {
|
|
w.body.Write(b)
|
|
return w.ResponseWriter.Write(b)
|
|
}
|
|
|
|
func LoggerMiddleware(conf gin.LoggerConfig) gin.HandlerFunc {
|
|
formatter := conf.Formatter
|
|
if formatter == nil {
|
|
formatter = defaultLogFormatter
|
|
}
|
|
|
|
return func(c *gin.Context) {
|
|
// Start timer
|
|
start := time.Now()
|
|
path := c.Request.URL.Path
|
|
raw := c.Request.URL.RawQuery
|
|
|
|
//body writer
|
|
blw := &bodyLogWriter{body: bytes.NewBufferString(""), ResponseWriter: c.Writer}
|
|
c.Writer = blw
|
|
|
|
// Process request
|
|
c.Next()
|
|
|
|
notlogged := conf.SkipPaths
|
|
var skip map[string]struct{}
|
|
if length := len(notlogged); length > 0 {
|
|
skip = make(map[string]struct{}, length)
|
|
|
|
for _, path := range notlogged {
|
|
skip[path] = struct{}{}
|
|
}
|
|
}
|
|
|
|
// Log only when path is not being skipped
|
|
if _, ok := skip[path]; !ok {
|
|
param := gin.LogFormatterParams{
|
|
Request: c.Request,
|
|
Keys: c.Keys,
|
|
}
|
|
// Stop timer
|
|
param.TimeStamp = time.Now()
|
|
param.Latency = param.TimeStamp.Sub(start)
|
|
|
|
param.ClientIP = c.ClientIP()
|
|
param.Method = c.Request.Method
|
|
param.StatusCode = c.Writer.Status()
|
|
param.ErrorMessage = c.Errors.ByType(gin.ErrorTypePrivate).String()
|
|
|
|
param.BodySize = c.Writer.Size()
|
|
|
|
if raw != "" {
|
|
path = path + "?" + raw
|
|
}
|
|
|
|
param.Path = path
|
|
|
|
var bodyStr string
|
|
body, ok := c.Get(gin.BodyBytesKey)
|
|
if ok {
|
|
bodyStr = string(body.([]byte))
|
|
}
|
|
logger.Info("%s params: %s", formatter(param), bodyStr)
|
|
//logger.Info("%s response: %s", formatter(param), blw.body.String())
|
|
|
|
//logger.WithCtx(c.Request.Context()).Info("%s params: %s", formatter(param), bodyStr)
|
|
//if accessConf.LogResponseEnabled {
|
|
// logger.WithCtx(c.Request.Context()).Info("%s response: %s", formatter(param), blw.body.String())
|
|
//}
|
|
}
|
|
}
|
|
}
|