diff --git a/internal/middleware/logger/logger.go b/internal/middleware/logger/logger.go index ef97541..ef96eac 100644 --- a/internal/middleware/logger/logger.go +++ b/internal/middleware/logger/logger.go @@ -7,6 +7,7 @@ import ( "github.com/loveuer/nf/nft/resp" "strconv" "time" + "ultone/internal/opt" "ultone/internal/tool" ) @@ -24,7 +25,8 @@ func New() nf.HandlerFunc { err := c.Next() - c.Writer.Header().Add(nf.TraceKey, fmt.Sprint(traceId)) + c.Writer.Header().Set(nf.TraceKey, fmt.Sprint(traceId)) + c.Writer.Header().Add("X-NF-Module", opt.Cfg.Name) status, _ := strconv.Atoi(c.Writer.Header().Get(resp.RealStatusHeader)) duration := time.Since(now)