labmem
cd0e13325f
- 配置 slog JSON 输出和日志级别 - 添加请求 ID、访问日志及上下文日志工具 - 拆分启动初始化并统一启动失败处理 BREAKING CHANGE: 配置文件必须提供 log.level，缺失或非法将导致启动失败
28 lines
488 B
Go
28 lines
488 B
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"lyra/internal/pkg/constant"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func RequestIDMiddleware() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
requestID := c.GetHeader("X-Request-ID")
|
|
|
|
if requestID == "" {
|
|
requestID = uuid.NewString()
|
|
}
|
|
|
|
c.Header("X-Request-ID", requestID)
|
|
|
|
ctx := context.WithValue(c.Request.Context(), constant.RequestIDKey, requestID)
|
|
|
|
c.Request = c.Request.WithContext(ctx)
|
|
|
|
c.Next()
|
|
}
|
|
}
|