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 初始化请求
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

2
router/server.go
View File

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