Browse Source

修改logger

yjp 1 năm trước cách đây
mục cha
commit
86c2b6c725
1 tập tin đã thay đổi với 7 bổ sung4 xóa
  1. 7 4
      gin_logger.go

+ 7 - 4
gin_logger.go

@@ -18,16 +18,19 @@ func GinLogger() gin.HandlerFunc {
 		Context: func(c *gin.Context) []zapcore.Field {
 			var body []byte
 			var buf bytes.Buffer
+			fields := make([]zapcore.Field, 0)
 
-			tee := io.TeeReader(c.Request.Body, &buf)
-			body, err := io.ReadAll(tee)
+			body, err := io.ReadAll(c.Request.Body)
 			if err != nil {
 				return make([]zapcore.Field, 0)
 			}
 
-			c.Request.Body = io.NopCloser(&buf)
+			if body != nil && len(body) != 0 {
+				fields = append(fields, zap.String("body", string(body)))
+				c.Request.Body = io.NopCloser(&buf)
+			}
 
-			return []zapcore.Field{zap.String("body", string(body))}
+			return fields
 		},
 	})
 }