更新前后端代码

go/internal/middleware/logger.go

@@ -0,0 +1,60 @@
+package middleware
+
+import (
+	"time"
+
+	"github.com/gin-gonic/gin"
+
+	"filestoragesystem/internal/utils"
+)
+
+// RequestID 请求ID中间件
+func RequestID() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		id := c.GetHeader("X-Request-Id")
+		if id == "" {
+			id = utils.RandomKey(8)
+		}
+		c.Set("requestID", id)
+		c.Header("X-Request-Id", id)
+		c.Next()
+	}
+}
+
+// Logger 访问日志中间件
+func Logger() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		start := time.Now()
+		path := c.Request.URL.Path
+
+		c.Next()
+
+		latency := time.Since(start)
+		status := c.Writer.Status()
+
+		args := []interface{}{
+			"status", status,
+			"method", c.Request.Method,
+			"path", path,
+			"ip", utils.ClientIP(c),
+			"latency", latency.String(),
+			"request_id", c.GetString("requestID"),
+		}
+		if uid := CurrentUserID(c); uid > 0 {
+			args = append(args, "user_id", uid)
+		}
+		if err := c.Errors.ByType(gin.ErrorTypePrivate).Last(); err != nil {
+			args = append(args, "error", err.Error())
+		}
+
+		sugar := utils.Logger().Sugar()
+		switch {
+		case status >= 500:
+			sugar.Errorw("access", args...)
+		case status >= 400:
+			sugar.Warnw("access", args...)
+		default:
+			sugar.Infow("access", args...)
+		}
+	}
+}