From 9041454d1bddbf826ff4046640910be97d3b0dc2 Mon Sep 17 00:00:00 2001
From: JingKai Tan <contact@jingk.ai>
Date: Wed, 10 Feb 2021 17:51:53 +0800
Subject: [PATCH] fix: fixed logger

---
 loggerx/middleware_id.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/loggerx/middleware_id.go b/loggerx/middleware_id.go
index ff28bb4..bc99fe8 100644
--- 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)
 	}
 }