From 6c5e74525e1ecf64077e525c829d9a4af61073a0 Mon Sep 17 00:00:00 2001
From: loveuer <loveuer@live.com>
Date: Fri, 19 Jul 2024 10:56:31 +0800
Subject: [PATCH] fix: logger add response trace-id

---
 internal/middleware/logger/logger.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/internal/middleware/logger/logger.go b/internal/middleware/logger/logger.go
index 8530030..ef97541 100644
--- 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)