loveuer/ultone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: add module header

Browse Source
This commit is contained in:
loveuer 2024-07-25 14:13:15 +08:00
parent 1b1143e93e
commit cee0bfb43d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/middleware/logger/logger.go
View File

@ -7,6 +7,7 @@ import (
"github.com/loveuer/nf/nft/resp" "github.com/loveuer/nf/nft/resp"
"strconv" "strconv"
"time" "time"
"ultone/internal/opt"
"ultone/internal/tool" "ultone/internal/tool"
) )
@ -24,7 +25,8 @@ func New() nf.HandlerFunc {
err := c.Next() 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)) status, _ := strconv.Atoi(c.Writer.Header().Get(resp.RealStatusHeader))
duration := time.Since(now) duration := time.Since(now)