85 lines
2.8 KiB
Go
85 lines
2.8 KiB
Go
|
// Package log define log struct of wave record project
|
|||
|
|
package log
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"os"
|
|||
|
|
"sync"
|
|||
|
|
|
|||
|
|
"modelRT/constant"
|
|||
|
|
|
|||
|
|
"github.com/natefinch/lumberjack"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
"go.uber.org/zap/zapcore"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
logger *zap.Logger
|
|||
|
|
once sync.Once
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// CutLogConfig define log config of wave record project
|
|||
|
|
type CutLogConfig struct {
|
|||
|
|
Mode string `json:"mode"` // Mode 日志模式 development、production
|
|||
|
|
Level string `json:"level"` // Level 最低日志等级，DEBUG<INFO<WARN<ERROR<FATAL INFO-->收集INFO等级以上的日志
|
|||
|
|
FileName string `json:"file_name"` // FileName 日志文件位置
|
|||
|
|
MaxSize int `json:"max_size"` // MaxSize 进行切割之前,日志文件的最大大小(MB为单位)默认为100MB
|
|||
|
|
MaxAge int `json:"max_age"` // MaxAge 是根据文件名中编码的时间戳保留旧日志文件的最大天数。
|
|||
|
|
MaxBackups int `json:"max_backups"` // MaxBackups 是要保留的旧日志文件的最大数量。默认是保留所有旧的日志文件(尽管 MaxAge 可能仍会导致它们被删除)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getEncoder responsible for setting the log format for encoding
|
|||
|
|
func getEncoder() zapcore.Encoder {
|
|||
|
|
encoderConfig := zap.NewProductionEncoderConfig()
|
|||
|
|
// serialization time eg:2006-01-02 15:04:05
|
|||
|
|
encoderConfig.EncodeTime = zapcore.TimeEncoderOfLayout("2006-01-02 15:04:05")
|
|||
|
|
encoderConfig.TimeKey = "time"
|
|||
|
|
encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
|
|||
|
|
encoderConfig.EncodeCaller = zapcore.ShortCallerEncoder
|
|||
|
|
return zapcore.NewJSONEncoder(encoderConfig)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getLogWriter responsible for setting the location of log storage
|
|||
|
|
func getLogWriter(mode, filename string, maxsize, maxBackup, maxAge int) zapcore.WriteSyncer {
|
|||
|
|
lumberJackLogger := &lumberjack.Logger{
|
|||
|
|
Filename: filename, // log file position
|
|||
|
|
MaxSize: maxsize, // log file maxsize
|
|||
|
|
MaxAge: maxAge, // maximum number of day files retained
|
|||
|
|
MaxBackups: maxBackup, // maximum number of old files retained
|
|||
|
|
Compress: false, // whether to compress
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
syncConsole := zapcore.AddSync(os.Stderr)
|
|||
|
|
if mode == constant.DevelopmentLogMode {
|
|||
|
|
return syncConsole
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
syncFile := zapcore.AddSync(lumberJackLogger)
|
|||
|
|
return zapcore.NewMultiWriteSyncer(syncFile, syncConsole)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// initLogger return successfully initialized zap logger
|
|||
|
|
func initLogger(lCfg CutLogConfig) *zap.Logger {
|
|||
|
|
writeSyncer := getLogWriter(lCfg.Mode, lCfg.FileName, lCfg.MaxSize, lCfg.MaxBackups, lCfg.MaxAge)
|
|||
|
|
encoder := getEncoder()
|
|||
|
|
|
|||
|
|
l := new(zapcore.Level)
|
|||
|
|
err := l.UnmarshalText([]byte(lCfg.Level))
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
core := zapcore.NewCore(encoder, writeSyncer, l)
|
|||
|
|
logger = zap.New(core, zap.AddCaller())
|
|||
|
|
zap.ReplaceGlobals(logger)
|
|||
|
|
|
|||
|
|
return logger
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetLoggerInstance return instance of zap logger
|
|||
|
|
func GetLoggerInstance(lCfg CutLogConfig) *zap.Logger {
|
|||
|
|
once.Do(func() {
|
|||
|
|
logger = initLogger(lCfg)
|
|||
|
|
})
|
|||
|
|
return logger
|
|||
|
|
}
|