fix: logger add response trace-id

2024-07-19 10:56:31 +08:00
parent 19f683caae
commit 6c5e74525e
1 changed files with 3 additions and 0 deletions
--- a/internal/middleware/logger/logger.go
+++ b/internal/middleware/logger/logger.go
@@ -23,6 +23,9 @@ func New() nf.HandlerFunc {
 		c.Locals(nf.TraceKey, traceId)

 		err := c.Next()
+
+		c.Writer.Header().Add(nf.TraceKey, fmt.Sprint(traceId))
+
 		status, _ := strconv.Atoi(c.Writer.Header().Get(resp.RealStatusHeader))
 		duration := time.Since(now)