28 lines
530 B
Go
28 lines
530 B
Go
|
package middleware
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/google/uuid"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// RequestID 为每个请求生成唯一ID
|
|||
|
|
func RequestID() gin.HandlerFunc {
|
|||
|
|
return func(c *gin.Context) {
|
|||
|
|
// 尝试从请求头获取Request ID
|
|||
|
|
requestID := c.GetHeader("X-Request-ID")
|
|||
|
|
|
|||
|
|
// 如果没有，生成新的UUID
|
|||
|
|
if requestID == "" {
|
|||
|
|
requestID = uuid.New().String()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置到上下文中
|
|||
|
|
c.Set("request_id", requestID)
|
|||
|
|
|
|||
|
|
// 设置到响应头中
|
|||
|
|
c.Header("X-Request-ID", requestID)
|
|||
|
|
|
|||
|
|
c.Next()
|
|||
|
|
}
|
|||
|
|
}
|