🎨 配置优化

config.go

@@ -2,16 +2,16 @@ package logger
 
 import "fmt"
 
-type mode int
+type mode string
 
-var (
-	Dev  mode = 0
-	Prod mode = 1
+const (
+	Dev  mode = "development"
+	Prod mode = "production"
 )
 
 // LogConfig 日志配置
 type LogConfig struct {
-	Mode       mode   `env:"LOG_MODE"`             // dev, prod
+	Mode       mode   `env:"LOG_MODE" envDefault:"production"` // dev, prod
 	LokiEnable bool   `env:"LOG_LOKI_ENABLE"`                  // 是否启用Loki
 	FileEnable bool   `env:"LOG_FILE_ENABLE"`                  // 是否输出到文件
 	LokiHost   string `env:"LOG_LOKI_HOST"`                    // Loki地址

logger.go

@@ -19,6 +19,10 @@ func init() {
 			fmt.Println("日志配置解析错误: " + err.Error())
 			c = LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false}
 		}
+		// 如果值错了，直接默认为Prod
+		if c.Mode != Dev && c.Mode != Prod {
+			c.Mode = Prod
+		}
 		InitLogger(c)
 	}
 }
@@ -43,7 +47,7 @@ func InitLogger(c LogConfig) {
 
 	// 增加 caller 信息
 	// AddCallerSkip 输出的文件名和行号是调用封装函数的位置，而不是调用日志函数的位置
-	logger := zap.New(zapcore.NewTee(cores...), zap.AddCaller(), zap.AddCallerSkip(1))
+	logger := zap.New(zapcore.NewTee(cores...), zap.AddCaller())
 	Say = logger.Sugar()
 	zap.ReplaceGlobals(logger)
 }