47 lines
1.5 KiB
Go
47 lines
1.5 KiB
Go
package logger
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/natefinch/lumberjack"
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
"os"
|
||
"time"
|
||
)
|
||
|
||
// 初始化LokiCore，使日志可以输出到文件
|
||
func initFileCore() zapcore.Core {
|
||
hostname, _ := os.Hostname()
|
||
lumberJackLogger := &lumberjack.Logger{
|
||
Filename: fmt.Sprintf("logs/runtime-%s.log", hostname), // 日志文件的位置
|
||
MaxSize: 10, // 最大10M
|
||
MaxBackups: 5, // 保留旧文件的最大个数
|
||
MaxAge: 30, // 保留旧文件的最大天数
|
||
Compress: false, // 是否压缩/归档旧文件
|
||
}
|
||
// 配置 sugaredLogger
|
||
writer := zapcore.AddSync(lumberJackLogger)
|
||
|
||
// 自定义时间输出格式
|
||
customTimeEncoder := func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||
enc.AppendString(fmt.Sprintf("[%v]", t.Format("2006-01-02 15:04:05.000")))
|
||
}
|
||
|
||
// 格式相关的配置
|
||
encoderConfig := zap.NewProductionEncoderConfig()
|
||
// 修改时间戳的格式
|
||
encoderConfig.EncodeTime = customTimeEncoder
|
||
// 日志级别使用大写显示
|
||
encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
|
||
encoder := zapcore.NewConsoleEncoder(encoderConfig)
|
||
|
||
// 设置日志等级，如果是release模式，控制台只打印Error级别以上的日志
|
||
logLevel := zapcore.DebugLevel
|
||
if config.Mode == Prod {
|
||
logLevel = zapcore.ErrorLevel
|
||
}
|
||
// 设置日志级别
|
||
core := zapcore.NewCore(encoder, writer, logLevel)
|
||
return core
|
||
}
|