소스 검색

修改logger

yjp 1 년 전
부모
커밋
86c2b6c725
1개의 변경된 파일7개의 추가작업 그리고 4개의 파일을 삭제
  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
 		},
 	})
 }