init

serve/internal/middleware/request_id.go

@@ -0,0 +1,27 @@
+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()
+	}
+}