From cee0bfb43d5f7b5b611a38fcceca6caa98de6827 Mon Sep 17 00:00:00 2001 From: loveuer Date: Thu, 25 Jul 2024 14:13:15 +0800 Subject: [PATCH] chore: add module header --- internal/middleware/logger/logger.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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)