67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package logger
|
||
|
||
import (
|
||
"os"
|
||
"time"
|
||
|
||
"github.com/spf13/viper"
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
"gopkg.in/natefinch/lumberjack.v2"
|
||
)
|
||
|
||
// Init 初始化日志（依赖配置文件已加载）
|
||
func Init() {
|
||
// 日志级别转换
|
||
level := zap.InfoLevel
|
||
switch viper.GetString("logger.level") {
|
||
case "debug":
|
||
level = zap.DebugLevel
|
||
case "warn":
|
||
level = zap.WarnLevel
|
||
case "error":
|
||
level = zap.ErrorLevel
|
||
}
|
||
|
||
// 日志轮转配置（lumberjack）
|
||
hook := lumberjack.Logger{
|
||
Filename: viper.GetString("logger.path") + "logs/app.log", // 日志文件路径
|
||
MaxSize: viper.GetInt("logger.max_size"), // 单个文件最大大小（MB）
|
||
MaxBackups: viper.GetInt("logger.max_backup"), // 最大备份数
|
||
MaxAge: viper.GetInt("logger.max_age"), // 最大保留天数
|
||
Compress: true, // 是否压缩
|
||
}
|
||
|
||
// 编码器配置
|
||
encoderConfig := zapcore.EncoderConfig{
|
||
TimeKey: "time",
|
||
LevelKey: "level",
|
||
NameKey: "logger",
|
||
CallerKey: "caller",
|
||
MessageKey: "msg",
|
||
StacktraceKey: "stacktrace",
|
||
LineEnding: zapcore.DefaultLineEnding,
|
||
EncodeLevel: zapcore.CapitalLevelEncoder, // 日志级别大写（DEBUG/INFO）
|
||
EncodeTime: customTimeEncoder, // 自定义时间格式
|
||
EncodeDuration: zapcore.SecondsDurationEncoder,
|
||
EncodeCaller: zapcore.ShortCallerEncoder, // 精简调用者路径
|
||
}
|
||
|
||
// 输出配置（控制台+文件）
|
||
core := zapcore.NewTee(
|
||
zapcore.NewCore(zapcore.NewConsoleEncoder(encoderConfig), zapcore.AddSync(os.Stdout), level),
|
||
zapcore.NewCore(zapcore.NewJSONEncoder(encoderConfig), zapcore.AddSync(&hook), level),
|
||
)
|
||
|
||
// 创建logger实例（开启调用者信息和堆栈跟踪）
|
||
logger := zap.New(core, zap.AddCaller(), zap.AddStacktrace(zap.ErrorLevel))
|
||
zap.ReplaceGlobals(logger)
|
||
|
||
zap.L().Info("✅ 日志初始化成功", zap.String("level", level.String()))
|
||
}
|
||
|
||
// customTimeEncoder 自定义时间格式
|
||
func customTimeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||
enc.AppendString(t.Format("2006-01-02 15:04:05.000"))
|
||
}
|