wip: 完成 client api 分析

internal/middleware/logger/logger.go

@@ -0,0 +1,71 @@
+package logger
+
+import (
+	"fmt"
+	"strconv"
+	"time"
+
+	"esway/internal/opt"
+	"esway/internal/tool"
+
+	"github.com/loveuer/esgo2dump/log"
+	"github.com/loveuer/nf"
+	"github.com/loveuer/nf/nft/resp"
+)
+
+type Config struct {
+	IgnoreFn func(c *nf.Ctx) bool
+}
+
+var defaultConfig = Config{
+	IgnoreFn: func(c *nf.Ctx) bool { return false },
+}
+
+func New(configs ...Config) nf.HandlerFunc {
+	return func(c *nf.Ctx) error {
+		var (
+			now   = time.Now()
+			logFn func(msg string, data ...any)
+			ip    = c.IP()
+			cfg   Config
+		)
+
+		if len(configs) > 0 {
+			cfg = configs[0]
+		}
+
+		if cfg.IgnoreFn == nil {
+			cfg.IgnoreFn = defaultConfig.IgnoreFn
+		}
+
+		traceId := c.Context().Value(nf.TraceKey)
+		c.Locals(nf.TraceKey, traceId)
+
+		err := c.Next()
+
+		c.Writer.Header().Set(nf.TraceKey, fmt.Sprint(traceId))
+		c.Writer.Header().Add("X-NF-Module", opt.Cfg.Name)
+
+		if cfg.IgnoreFn(c) {
+			return err
+		}
+
+		status, _ := strconv.Atoi(c.Writer.Header().Get(resp.RealStatusHeader))
+		duration := time.Since(now)
+
+		msg := fmt.Sprintf("%s | %15s | %d[%3d] | %s | %6s | %s", traceId, ip, c.StatusCode, status, tool.HumanDuration(duration.Nanoseconds()), c.Method(), c.Path())
+
+		switch {
+		case status >= 500:
+			logFn = log.Error
+		case status >= 400:
+			logFn = log.Warn
+		default:
+			logFn = log.Info
+		}
+
+		logFn(msg)
+
+		return err
+	}
+}