63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
package log
|
||
|
||
import (
|
||
commonConfig "common/config"
|
||
"github.com/natefinch/lumberjack"
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
"os"
|
||
)
|
||
|
||
func Init(cfg *commonConfig.LoggerConfig) {
|
||
// 格式配置
|
||
jsonConfig := zapcore.EncoderConfig{
|
||
MessageKey: "M",
|
||
LevelKey: "L",
|
||
TimeKey: "T",
|
||
NameKey: "N",
|
||
CallerKey: "C",
|
||
FunctionKey: zapcore.OmitKey,
|
||
StacktraceKey: "S",
|
||
LineEnding: zapcore.DefaultLineEnding,
|
||
EncodeLevel: zapcore.CapitalLevelEncoder,
|
||
EncodeTime: zapcore.TimeEncoderOfLayout("01-02 15:04:05.000"),
|
||
EncodeDuration: zapcore.SecondsDurationEncoder,
|
||
EncodeCaller: zapcore.ShortCallerEncoder,
|
||
EncodeName: nil,
|
||
ConsoleSeparator: " ",
|
||
}
|
||
|
||
// 日志输出到控制台和文件
|
||
writeSyncer := []zapcore.WriteSyncer{zapcore.AddSync(os.Stdout)}
|
||
if !cfg.Debug {
|
||
writeSyncer = append(writeSyncer, zapcore.AddSync(&lumberjack.Logger{
|
||
Filename: "./logs/log.log", // 日志文件位置
|
||
MaxSize: cfg.MaxSize, // 最大文件大小（MB）
|
||
MaxBackups: cfg.MaxBackUp, // 保留旧文件的最大个数
|
||
MaxAge: cfg.MaxAge, // 保留旧文件的最大天数
|
||
Compress: false, // 是否压缩/归档旧文件
|
||
LocalTime: true,
|
||
}))
|
||
}
|
||
|
||
var encoder zapcore.Encoder
|
||
if cfg.Debug {
|
||
encoder = zapcore.NewConsoleEncoder(jsonConfig)
|
||
} else {
|
||
encoder = zapcore.NewJSONEncoder(jsonConfig)
|
||
}
|
||
logger := zap.New(zapcore.NewCore(
|
||
encoder,
|
||
zapcore.NewMultiWriteSyncer(writeSyncer...),
|
||
zap.NewAtomicLevelAt(GetLogLevel(cfg.Level)),
|
||
))
|
||
if cfg.Debug {
|
||
logger = logger.WithOptions(
|
||
zap.AddCaller(),
|
||
zap.AddCallerSkip(1),
|
||
zap.AddStacktrace(zapcore.WarnLevel),
|
||
)
|
||
}
|
||
SetLogger(logger.Sugar())
|
||
}
|