From a5597d50b80f7d14df8413e21656048ec8b2b72b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?= Date: Sat, 3 Jan 2026 13:04:42 +0800 Subject: [PATCH] feat: update code --- middleware/init_request.go | 18 ++++++++---------- router/server.go | 2 +- 2 files changed, 9 insertions(+), 11 deletions(-) diff --git a/middleware/init_request.go b/middleware/init_request.go index d3771cc..b786ce7 100644 --- a/middleware/init_request.go +++ b/middleware/init_request.go @@ -21,16 +21,14 @@ import ( ) // InitRequest 初始化请求 -func InitRequest() gin.HandlerFunc { - return func(ctx *gin.Context) { - httpHandleConfig := define.GetHttpHandleConfig() - requestID := getRequestID(ctx) - traceID := getTraceID(ctx, requestID) - ctx.Set(httpHandleConfig.TraceIDField, traceID) - ctx.Set(httpHandleConfig.RequestIDField, requestID) - ctx.Set(httpHandleConfig.StartRequestTimeField, time.Now().UnixMilli()) - ctx.Next() - } +func InitRequest(ctx *gin.Context) { + httpHandleConfig := define.GetHttpHandleConfig() + requestID := getRequestID(ctx) + traceID := getTraceID(ctx, requestID) + ctx.Set(httpHandleConfig.TraceIDField, traceID) + ctx.Set(httpHandleConfig.RequestIDField, requestID) + ctx.Set(httpHandleConfig.StartRequestTimeField, time.Now().UnixMilli()) + ctx.Next() } // getRequestID 生成 request id diff --git a/router/server.go b/router/server.go index 2b5801b..d9352ef 100644 --- a/router/server.go +++ b/router/server.go @@ -114,7 +114,7 @@ func (s *server) getGlobalMiddlewareList(option *serverOption) []gin.HandlerFunc globalMiddlewareList = append(globalMiddlewareList, option.initContextData) // 初始化一些全局的变量 } if !option.disableInitRequest { // 启用了初始化请求 - globalMiddlewareList = append(globalMiddlewareList, middleware.InitRequest()) // 初始化请求 + globalMiddlewareList = append(globalMiddlewareList, middleware.InitRequest) // 初始化请求 } if nil != option.loggerCfg { // 请求日志记录中间件