55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"github.com/duke-git/lancet/v2/cryptor"
|
||
|
|
"github.com/duke-git/lancet/v2/random"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
"io"
|
||
|
|
"nunu-layout-admin/pkg/log"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RequestLogMiddleware(logger *log.Logger) gin.HandlerFunc {
|
||
|
|
return func(ctx *gin.Context) {
|
||
|
|
// The configuration is initialized once per request
|
||
|
|
uuid, err := random.UUIdV4()
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
trace := cryptor.Md5String(uuid)
|
||
|
|
logger.WithValue(ctx, zap.String("trace", trace))
|
||
|
|
logger.WithValue(ctx, zap.String("request_method", ctx.Request.Method))
|
||
|
|
logger.WithValue(ctx, zap.Any("request_headers", ctx.Request.Header))
|
||
|
|
logger.WithValue(ctx, zap.String("request_url", ctx.Request.URL.String()))
|
||
|
|
if ctx.Request.Body != nil {
|
||
|
|
bodyBytes, _ := ctx.GetRawData()
|
||
|
|
ctx.Request.Body = io.NopCloser(bytes.NewBuffer(bodyBytes)) // 关键点
|
||
|
|
logger.WithValue(ctx, zap.String("request_params", string(bodyBytes)))
|
||
|
|
}
|
||
|
|
logger.WithContext(ctx).Info("Request")
|
||
|
|
ctx.Next()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func ResponseLogMiddleware(logger *log.Logger) gin.HandlerFunc {
|
||
|
|
return func(ctx *gin.Context) {
|
||
|
|
blw := &bodyLogWriter{body: bytes.NewBufferString(""), ResponseWriter: ctx.Writer}
|
||
|
|
ctx.Writer = blw
|
||
|
|
startTime := time.Now()
|
||
|
|
ctx.Next()
|
||
|
|
duration := time.Since(startTime).String()
|
||
|
|
logger.WithContext(ctx).Info("Response", zap.Any("response_body", blw.body.String()), zap.Any("time", duration))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
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)
|
||
|
|
}
|