109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
package log
|
||
|
||
import (
|
||
"os"
|
||
"time"
|
||
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
"gopkg.in/natefinch/lumberjack.v2"
|
||
)
|
||
|
||
var (
|
||
logger *zap.Logger
|
||
sugar *zap.SugaredLogger
|
||
)
|
||
|
||
// 通用时间格式
|
||
const (
|
||
TimeFormatDate = "2006-01-02"
|
||
TimeFormatDateTime = "2006-01-02 15:04:05"
|
||
)
|
||
|
||
type Config struct {
|
||
Level string
|
||
Filepath string
|
||
MaxSizeMB int
|
||
MaxAgeDay int
|
||
Backups int
|
||
Compress bool
|
||
}
|
||
|
||
func init() {
|
||
encoderConfig := zap.NewDevelopmentEncoderConfig()
|
||
core := zapcore.NewCore(
|
||
zapcore.NewConsoleEncoder(encoderConfig),
|
||
zapcore.AddSync(os.Stdout),
|
||
zap.DebugLevel,
|
||
)
|
||
core = zapcore.NewTee(core, NewMemoryCore(zap.DebugLevel))
|
||
|
||
logger = zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
|
||
sugar = logger.Sugar()
|
||
}
|
||
|
||
func Init(cfg Config) {
|
||
var zapLevel zapcore.Level
|
||
|
||
// 日志等级解析
|
||
switch cfg.Level {
|
||
case "debug":
|
||
zapLevel = zap.DebugLevel
|
||
case "info":
|
||
zapLevel = zap.InfoLevel
|
||
case "warning", "warn":
|
||
zapLevel = zap.WarnLevel
|
||
case "error":
|
||
zapLevel = zap.ErrorLevel
|
||
default:
|
||
zapLevel = zap.InfoLevel
|
||
}
|
||
|
||
// lumberjack 日志切割配置
|
||
writeSyncer := zapcore.AddSync(&lumberjack.Logger{
|
||
Filename: cfg.Filepath,
|
||
MaxSize: cfg.MaxSizeMB,
|
||
MaxBackups: cfg.Backups,
|
||
MaxAge: cfg.MaxAgeDay,
|
||
Compress: cfg.Compress,
|
||
})
|
||
|
||
// 基础编码配置 (提取公共部分)
|
||
baseEencoderConfig := zapcore.EncoderConfig{
|
||
TimeKey: "time",
|
||
LevelKey: "level",
|
||
NameKey: "logger",
|
||
CallerKey: "caller",
|
||
MessageKey: "msg",
|
||
StacktraceKey: "Stacktrace",
|
||
LineEnding: zapcore.DefaultLineEnding,
|
||
EncodeLevel: zapcore.CapitalLevelEncoder,
|
||
EncodeTime: func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||
enc.AppendString(t.Format(TimeFormatDateTime))
|
||
},
|
||
EncodeDuration: zapcore.StringDurationEncoder,
|
||
EncodeCaller: zapcore.ShortCallerEncoder,
|
||
}
|
||
|
||
// 1. 控制台编码器（开启颜色）
|
||
consoleEncoderConfig := baseEencoderConfig
|
||
consoleEncoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder
|
||
encoderConsole := zapcore.NewConsoleEncoder(consoleEncoderConfig)
|
||
|
||
// 2.文件编码器（json 格式, 无色）
|
||
fileEncoderConfig := baseEencoderConfig
|
||
fileEncoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
|
||
encoderJson := zapcore.NewJSONEncoder(fileEncoderConfig)
|
||
|
||
core := zapcore.NewTee(
|
||
zapcore.NewCore(encoderJson, writeSyncer, zapLevel),
|
||
zapcore.NewCore(encoderConsole, zapcore.AddSync(os.Stdout), zapLevel),
|
||
NewMemoryCore(zapLevel),
|
||
)
|
||
|
||
logger = zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
|
||
sugar = logger.Sugar()
|
||
|
||
zap.ReplaceGlobals(logger)
|
||
}
|