u

code/allapp/allapp-go/config/logger.go

@@ -1 +1,153 @@
 package config
+
+import (
+	"fmt"
+	"os"
+	"sync"
+	"time"
+
+	fiberZap "github.com/gofiber/contrib/v3/zap"
+	"github.com/gofiber/fiber/v3/log"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	"gopkg.in/natefinch/lumberjack.v2"
+)
+
+var (
+	mu          sync.Mutex
+	currentDay  = make(map[string]string)
+	currentSync = make(map[string]zapcore.WriteSyncer)
+)
+
+func initLogger() {
+	dynamicLevel := zap.NewAtomicLevel()
+	dynamicLevel.SetLevel(LogLevel())
+
+	logger := fiberZap.NewLogger(fiberZap.LoggerConfig{
+		ExtraKeys: []string{"request_id"},
+		ZapOptions: []zap.Option{
+			zap.AddCaller(),      // 打开 caller
+			zap.AddCallerSkip(2), // 跳过一层 Fiber 调用
+		},
+		CoreConfigs: []fiberZap.CoreConfig{
+			{
+				Encoder:      zapcore.NewConsoleEncoder(consoleEncoderConfig()),
+				WriteSyncer:  zapcore.AddSync(os.Stdout),
+				LevelEncoder: dynamicLevel,
+			},
+			{
+				Encoder:     zapcore.NewJSONEncoder(humanEncoderConfig()),
+				WriteSyncer: getWriteSyncer(fmt.Sprintf("%s/debug", conf.Log.Dir)),
+				LevelEncoder: zap.LevelEnablerFunc(func(lev zapcore.Level) bool {
+					return lev == zap.DebugLevel
+				}),
+			},
+			{
+				Encoder:     zapcore.NewJSONEncoder(humanEncoderConfig()),
+				WriteSyncer: getWriteSyncer(fmt.Sprintf("%s/info", conf.Log.Dir)),
+				LevelEncoder: zap.LevelEnablerFunc(func(lev zapcore.Level) bool {
+					return lev == zap.InfoLevel
+				}),
+			},
+			{
+				Encoder:     zapcore.NewJSONEncoder(humanEncoderConfig()),
+				WriteSyncer: getWriteSyncer(fmt.Sprintf("%s/warn", conf.Log.Dir)),
+				LevelEncoder: zap.LevelEnablerFunc(func(lev zapcore.Level) bool {
+					return lev == zap.WarnLevel
+				}),
+			},
+			{
+				Encoder:     zapcore.NewJSONEncoder(humanEncoderConfig()),
+				WriteSyncer: getWriteSyncer(fmt.Sprintf("%s/error", conf.Log.Dir)),
+				LevelEncoder: zap.LevelEnablerFunc(func(lev zapcore.Level) bool {
+					return lev >= zap.ErrorLevel
+				}),
+			},
+		},
+	})
+
+	log.SetLogger(logger)
+
+	// 退出时同步日志
+	defer func() {
+		_ = logger.Sync()
+	}()
+}
+
+func getWriteSyncer(basePath string) zapcore.WriteSyncer {
+	mu.Lock()
+	defer mu.Unlock()
+
+	today := time.Now().Format("2006-01-02")
+	if day, ok := currentDay[basePath]; !ok || day != today {
+		currentDay[basePath] = today
+		file := fmt.Sprintf("%s/%s.log", basePath, today)
+
+		currentSync[basePath] = zapcore.AddSync(&lumberjack.Logger{
+			Filename:   file,
+			MaxSize:    conf.Log.MaxSize,
+			MaxBackups: conf.Log.MaxBackups,
+			MaxAge:     conf.Log.MaxAge,
+			Compress:   conf.Log.Compress,
+			LocalTime:  true,
+		})
+	}
+	return currentSync[basePath]
+}
+
+func LogLevel() zapcore.Level {
+	level := conf.Log.Level
+
+	switch level {
+	case "trace":
+		return zapcore.DebugLevel
+	case "debug":
+		return zapcore.DebugLevel
+	case "info":
+		return zapcore.InfoLevel
+	case "warn":
+		return zapcore.WarnLevel
+	case "error":
+		return zapcore.ErrorLevel
+	case "dpanic":
+		return zapcore.DPanicLevel
+	case "panic":
+		return zapcore.PanicLevel
+	case "fatal":
+		return zapcore.FatalLevel
+	default:
+		return zapcore.InfoLevel
+	}
+}
+
+func consoleEncoderConfig() zapcore.EncoderConfig {
+	return zapcore.EncoderConfig{
+		MessageKey:   "msg",
+		LevelKey:     "level",
+		TimeKey:      "ts",
+		EncodeTime:   customTimeEncoder,
+		EncodeLevel:  zapcore.CapitalColorLevelEncoder,
+		EncodeCaller: zapcore.ShortCallerEncoder,
+	}
+}
+
+func humanEncoderConfig() zapcore.EncoderConfig {
+	return zapcore.EncoderConfig{
+		MessageKey:     "message",
+		LevelKey:       "level",
+		NameKey:        "mame",
+		TimeKey:        "ts",
+		CallerKey:      "caller",
+		FunctionKey:    "func",
+		StacktraceKey:  "stacktrace",
+		LineEnding:     "\n",
+		EncodeTime:     customTimeEncoder,
+		EncodeLevel:    zapcore.LowercaseLevelEncoder,
+		EncodeDuration: zapcore.SecondsDurationEncoder,
+		EncodeCaller:   zapcore.ShortCallerEncoder,
+	}
+}
+
+func customTimeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
+	enc.AppendString(t.Format("2006-01-02 15:04:05.000"))
+}