122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
/*
|
||
@Author : lanyulei
|
||
*/
|
||
|
||
package logger
|
||
|
||
import (
|
||
"os"
|
||
"time"
|
||
|
||
"gopkg.in/natefinch/lumberjack.v2"
|
||
|
||
"github.com/spf13/viper"
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
)
|
||
|
||
// error logger
|
||
var log *zap.SugaredLogger
|
||
|
||
var levelMap = map[string]zapcore.Level{
|
||
"debug": zapcore.DebugLevel,
|
||
"info": zapcore.InfoLevel,
|
||
"warn": zapcore.WarnLevel,
|
||
"error": zapcore.ErrorLevel,
|
||
"dpanic": zapcore.DPanicLevel,
|
||
"panic": zapcore.PanicLevel,
|
||
"fatal": zapcore.FatalLevel,
|
||
}
|
||
|
||
func Init() {
|
||
var syncWriters []zapcore.WriteSyncer
|
||
level := getLoggerLevel(viper.GetString(`settings.log.level`))
|
||
fileConfig := &lumberjack.Logger{
|
||
Filename: viper.GetString(`settings.log.path`), // 日志文件名
|
||
MaxSize: viper.GetInt(`settings.log.maxsize`), // 日志文件大小
|
||
MaxAge: viper.GetInt(`settings.log.maxAge`), // 最长保存天数
|
||
MaxBackups: viper.GetInt(`settings.log.maxBackups`), // 最多备份几个
|
||
LocalTime: viper.GetBool(`settings.log.localtime`), // 日志时间戳
|
||
Compress: viper.GetBool(`settings.log.compress`), // 是否压缩文件，使用gzip
|
||
}
|
||
encoder := zap.NewProductionEncoderConfig()
|
||
encoder.EncodeTime = func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||
enc.AppendString(t.Format("2006-01-02 15:04:05.000000"))
|
||
}
|
||
if viper.GetBool("settings.log.consoleStdout") {
|
||
syncWriters = append(syncWriters, zapcore.AddSync(os.Stdout))
|
||
}
|
||
if viper.GetBool("settings.log.fileStdout") {
|
||
syncWriters = append(syncWriters, zapcore.AddSync(fileConfig))
|
||
}
|
||
core := zapcore.NewCore(
|
||
zapcore.NewJSONEncoder(encoder),
|
||
zapcore.NewMultiWriteSyncer(syncWriters...),
|
||
zap.NewAtomicLevelAt(level))
|
||
logger := zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
|
||
log = logger.Sugar()
|
||
}
|
||
|
||
func getLoggerLevel(lvl string) zapcore.Level {
|
||
if level, ok := levelMap[lvl]; ok {
|
||
return level
|
||
}
|
||
return zapcore.InfoLevel
|
||
}
|
||
|
||
func Debug(args ...interface{}) {
|
||
log.Debug(args...)
|
||
}
|
||
|
||
func Debugf(format string, args ...interface{}) {
|
||
log.Debugf(format, args...)
|
||
}
|
||
|
||
func Info(args ...interface{}) {
|
||
log.Info(args...)
|
||
}
|
||
|
||
func Infof(format string, args ...interface{}) {
|
||
log.Infof(format, args...)
|
||
}
|
||
|
||
func Warn(args ...interface{}) {
|
||
log.Warn(args...)
|
||
}
|
||
|
||
func Warnf(format string, args ...interface{}) {
|
||
log.Warnf(format, args...)
|
||
}
|
||
|
||
func Error(args ...interface{}) {
|
||
log.Error(args...)
|
||
}
|
||
|
||
func Errorf(format string, args ...interface{}) {
|
||
log.Errorf(format, args...)
|
||
}
|
||
|
||
func DPanic(args ...interface{}) {
|
||
log.DPanic(args...)
|
||
}
|
||
|
||
func DPanicf(format string, args ...interface{}) {
|
||
log.DPanicf(format, args...)
|
||
}
|
||
|
||
func Panic(args ...interface{}) {
|
||
log.Panic(args...)
|
||
}
|
||
|
||
func Panicf(format string, args ...interface{}) {
|
||
log.Panicf(format, args...)
|
||
}
|
||
|
||
func Fatal(args ...interface{}) {
|
||
log.Fatal(args...)
|
||
}
|
||
|
||
func Fatalf(format string, args ...interface{}) {
|
||
log.Fatalf(format, args...)
|
||
}
|