zhangdeman/gin
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: update code

Browse Source
This commit is contained in:
白茶清欢
2026-01-03 13:04:42 +08:00
parent 28a355dd17
commit a5597d50b8
2 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
middleware/init_request.go
View File

@@ -21,16 +21,14 @@ import (
) )
// InitRequest 初始化请求 // InitRequest 初始化请求
func InitRequest() gin.HandlerFunc { func InitRequest(ctx *gin.Context) {
return func(ctx *gin.Context) { httpHandleConfig := define.GetHttpHandleConfig()
httpHandleConfig := define.GetHttpHandleConfig() requestID := getRequestID(ctx)
requestID := getRequestID(ctx) traceID := getTraceID(ctx, requestID)
traceID := getTraceID(ctx, requestID) ctx.Set(httpHandleConfig.TraceIDField, traceID)
ctx.Set(httpHandleConfig.TraceIDField, traceID) ctx.Set(httpHandleConfig.RequestIDField, requestID)
ctx.Set(httpHandleConfig.RequestIDField, requestID) ctx.Set(httpHandleConfig.StartRequestTimeField, time.Now().UnixMilli())
ctx.Set(httpHandleConfig.StartRequestTimeField, time.Now().UnixMilli()) ctx.Next()
ctx.Next()
}
} }
// getRequestID 生成 request id // getRequestID 生成 request id

2
router/server.go
View File

@@ -114,7 +114,7 @@ func (s *server) getGlobalMiddlewareList(option *serverOption) []gin.HandlerFunc
globalMiddlewareList = append(globalMiddlewareList, option.initContextData) // 初始化一些全局的变量 globalMiddlewareList = append(globalMiddlewareList, option.initContextData) // 初始化一些全局的变量
} }
if !option.disableInitRequest { // 启用了初始化请求 if !option.disableInitRequest { // 启用了初始化请求
globalMiddlewareList = append(globalMiddlewareList, middleware.InitRequest()) // 初始化请求 globalMiddlewareList = append(globalMiddlewareList, middleware.InitRequest) // 初始化请求
} }
if nil != option.loggerCfg { if nil != option.loggerCfg {
// 请求日志记录中间件 // 请求日志记录中间件