135 lines
3.6 KiB
Go
135 lines
3.6 KiB
Go
package log
|
||
|
||
import (
|
||
"fmt"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/config"
|
||
rotatelogs "github.com/lestrrat-go/file-rotatelogs"
|
||
"github.com/natefinch/lumberjack"
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
"io"
|
||
"path/filepath"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
var Logger *zap.Logger
|
||
var once sync.Once
|
||
|
||
func InitLogger() {
|
||
once.Do(func() { Logger = createZapLog() })
|
||
}
|
||
|
||
// initZapLog 初始化 zap 日志
|
||
func createZapLog() *zap.Logger {
|
||
// 开启 debug
|
||
if config.Config.Debug == true {
|
||
if Logger, err := zap.NewDevelopment(); err == nil {
|
||
return Logger
|
||
} else {
|
||
panic("创建zap日志包失败，详情：" + err.Error())
|
||
}
|
||
}
|
||
|
||
encoderConfig := zap.NewProductionEncoderConfig()
|
||
encoderConfig.EncodeTime = func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||
enc.AppendString(t.Format("2006-01-02 15:04:05.000"))
|
||
}
|
||
|
||
// 在日志文件中使用大写字母记录日志级别
|
||
encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
|
||
encoder := zapcore.NewConsoleEncoder(encoderConfig)
|
||
filename := filepath.Join(config.Config.StaticBasePath, "/logs/", config.Config.Logger.Filename)
|
||
var writer zapcore.WriteSyncer
|
||
if config.Config.Logger.DefaultDivision == "size" {
|
||
// 按文件大小切割日志
|
||
writer = zapcore.AddSync(getLumberJackWriter(filename))
|
||
} else {
|
||
// 按天切割日志
|
||
writer = zapcore.AddSync(getRotateWriter(filename))
|
||
}
|
||
|
||
zapCore := zapcore.NewCore(encoder, writer, zap.InfoLevel)
|
||
//zap.AddStacktrace(zap.WarnLevel)
|
||
return zap.New(zapCore, zap.AddCaller())
|
||
}
|
||
|
||
// getRotateWriter 按日期切割日志
|
||
func getRotateWriter(filename string) io.Writer {
|
||
maxAge := time.Duration(config.Config.Logger.DivisionTime.MaxAge)
|
||
rotationTime := time.Duration(config.Config.Logger.DivisionTime.RotationTime)
|
||
hook, err := rotatelogs.New(
|
||
filename+".%Y%m%d",
|
||
rotatelogs.WithLinkName(filename),
|
||
rotatelogs.WithMaxAge(time.Hour*24*maxAge),
|
||
rotatelogs.WithRotationTime(time.Hour*rotationTime), // 默认一天
|
||
)
|
||
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
return hook
|
||
}
|
||
|
||
// getLumberJackWriter 按文件切割日志
|
||
func getLumberJackWriter(filename string) io.Writer {
|
||
// 日志切割配置
|
||
return &lumberjack.Logger{
|
||
Filename: filename, // 日志文件的位置
|
||
MaxSize: config.Config.Logger.DivisionSize.MaxSize, // 在进行切割之前，日志文件的最大大小（以MB为单位）
|
||
MaxBackups: config.Config.Logger.DivisionSize.MaxBackups, // 保留旧文件的最大个数
|
||
MaxAge: config.Config.Logger.DivisionSize.MaxAge, // 保留旧文件的最大天数
|
||
Compress: config.Config.Logger.DivisionSize.Compress, // 是否压缩/归档旧文件
|
||
}
|
||
}
|
||
|
||
func Info(args ...interface{}) {
|
||
Logger.Info(fmt.Sprintf("%s", args...))
|
||
}
|
||
|
||
func Infof(format string, args ...interface{}) {
|
||
Logger.Info(fmt.Sprintf(format, args...))
|
||
}
|
||
|
||
func Warning(args ...interface{}) {
|
||
Logger.Warn(fmt.Sprintf("%s", args...))
|
||
}
|
||
|
||
func Warningf(format string, args ...interface{}) {
|
||
Logger.Warn(fmt.Sprintf(format, args...))
|
||
}
|
||
|
||
func Error(args ...interface{}) {
|
||
Logger.Error(fmt.Sprintf("%s", args...))
|
||
}
|
||
|
||
func Errorf(format string, args ...interface{}) {
|
||
Logger.Error(fmt.Sprintf(format, args...))
|
||
}
|
||
|
||
func Debug(args ...interface{}) {
|
||
Logger.Debug(fmt.Sprintf("%s", args...))
|
||
|
||
}
|
||
|
||
func Debugf(format string, args ...interface{}) {
|
||
Logger.Debug(fmt.Sprintf(format, args...))
|
||
}
|
||
|
||
func Trace(args ...interface{}) {
|
||
Logger.Info(fmt.Sprintf("%s", args...))
|
||
}
|
||
|
||
func Tracef(format string, args ...interface{}) {
|
||
Logger.Info(fmt.Sprintf(format, args...))
|
||
}
|
||
|
||
func Fatal(args ...interface{}) {
|
||
Logger.Fatal(fmt.Sprintf("%s", args...))
|
||
}
|
||
|
||
func Fatalf(format string, args ...interface{}) {
|
||
Logger.Fatal(fmt.Sprintf(format, args...))
|
||
}
|