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 { // 请求日志记录中间件