38 lines
1018 B
Go
38 lines
1018 B
Go
|
package logger
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
"go.uber.org/zap/zapcore"
|
|||
|
|
"os"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 初始化打印到控制台的ZapCore
|
|||
|
|
func initConsoleCore() zapcore.Core {
|
|||
|
|
// 配置 sugaredLogger
|
|||
|
|
writer := zapcore.AddSync(os.Stdout)
|
|||
|
|
|
|||
|
|
// 自定义时间输出格式
|
|||
|
|
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.CapitalColorLevelEncoder
|
|||
|
|
encoder := zapcore.NewConsoleEncoder(encoderConfig)
|
|||
|
|
|
|||
|
|
// 设置日志等级，如果是release模式，控制台只打印Error级别以上的日志
|
|||
|
|
logLevel := zapcore.DebugLevel
|
|||
|
|
if config.Mode == Prod {
|
|||
|
|
logLevel = zapcore.ErrorLevel
|
|||
|
|
}
|
|||
|
|
// 设置日志级别
|
|||
|
|
core := zapcore.NewCore(encoder, writer, logLevel)
|
|||
|
|
return core
|
|||
|
|
}
|