diff --git a/pkg/errors/context.go b/pkg/errors/context.go index 28f653c..3d7beca 100644 --- a/pkg/errors/context.go +++ b/pkg/errors/context.go @@ -1,6 +1,8 @@ package errors import ( + "strconv" + "github.com/gofiber/fiber/v2" "go.uber.org/zap" @@ -44,8 +46,8 @@ func FromFiberContext(c *fiber.Ctx) *ErrorContext { // 提取 User ID(如果已认证) if uid := c.Locals(constants.ContextKeyUserID); uid != nil { - if userID, ok := uid.(string); ok { - ctx.UserID = userID + if userID, ok := uid.(uint); ok { + ctx.UserID = strconv.FormatUint(uint64(userID), 10) } } diff --git a/pkg/logger/middleware.go b/pkg/logger/middleware.go index 2a3ec07..79440b7 100644 --- a/pkg/logger/middleware.go +++ b/pkg/logger/middleware.go @@ -54,9 +54,11 @@ func Middleware() fiber.Handler { } // 获取用户 ID(由 auth 中间件设置) - userID := "" + var userID uint if uid := c.Locals(constants.ContextKeyUserID); uid != nil { - userID = uid.(string) + if id, ok := uid.(uint); ok { + userID = id + } } // 获取响应 body @@ -73,7 +75,7 @@ func Middleware() fiber.Handler { zap.String("request_id", requestID), zap.String("ip", c.IP()), zap.String("user_agent", c.Get("User-Agent")), - zap.String(constants.ContextKeyUserID, userID), + zap.Uint("user_id", userID), zap.String("request_body", requestBody), zap.String("response_body", responseBody), )