Merge branch 'add-logger'

AcePointerOSS · Feb 10, 2021 · 36e08f5 · 36e08f5
2 parents 2dbf71f + 9041454
commit 36e08f5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/loggerx/middleware_id.go b/loggerx/middleware_id.go
@@ -6,7 +6,7 @@ import (
 )
 
 func GetReqId(c *gin.Context) string {
-	return c.GetHeader("request-id")
+	return c.Request.Header.Get("request-id")
 }
 
 func UseRequestId() gin.HandlerFunc {
@@ -16,7 +16,6 @@ func UseRequestId() gin.HandlerFunc {
 		if reqId == "" {
 			reqId = uuid.Must(uuid.NewV4()).String()
 		}
-		c.Header("request-id", reqId)
-		c.Next()
+		c.Request.Header.Set("request-id", reqId)
 	}
 }