feat: add gls

pkg/middleware/requestid.go

@@ -1,19 +1,23 @@
 package middleware
 
 import (
+	"gitea.com/bitwsd/document_ai/pkg/requestid"
+
 	"github.com/gin-gonic/gin"
 	"github.com/google/uuid"
 )
 
 func RequestID() gin.HandlerFunc {
 	return func(c *gin.Context) {
-		requestID := c.Request.Header.Get("X-Request-ID")
-		if requestID == "" {
-			requestID = uuid.New().String()
+		reqID := c.Request.Header.Get("X-Request-ID")
+		if reqID == "" {
+			reqID = uuid.New().String()
 		}
-		c.Request.Header.Set("X-Request-ID", requestID)
-		c.Set("request_id", requestID)
-		c.Next()
+		c.Request.Header.Set("X-Request-ID", reqID)
+		c.Set("request_id", reqID)
+
+		requestid.SetRequestID(reqID, func() {
+			c.Next()
+		})
 	}
 }
-