1

common/middleware/request_id.go

@@ -0,0 +1,36 @@
+package middleware
+
+import (
+	"github.com/go-admin-team/go-admin-core/logger"
+	"github.com/go-admin-team/go-admin-core/sdk/pkg"
+	"net/http"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+)
+
+// RequestId 自动增加requestId
+func RequestId(trafficKey string) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		if c.Request.Method == http.MethodOptions {
+			c.Next()
+			return
+		}
+		requestId := c.GetHeader(trafficKey)
+		if requestId == "" {
+			requestId = c.GetHeader(strings.ToLower(trafficKey))
+		}
+		if requestId == "" {
+			requestId = uuid.New().String()
+		}
+		c.Request.Header.Set(trafficKey, requestId)
+		c.Set(trafficKey, requestId)
+		c.Set(pkg.LoggerKey,
+			logger.NewHelper(logger.DefaultLogger).
+				WithFields(map[string]interface{}{
+					trafficKey: requestId,
+				}))
+		c.Next()
+	}
+}