/
zap.go
78 lines (69 loc) · 2.09 KB
/
zap.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package zap
import (
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"gopkg.in/natefinch/lumberjack.v2"
"os"
"time"
)
var logger *zap.Logger
func Setup(opts ...Option) {
opt := newOption(opts...)
encoderConfig := zapcore.EncoderConfig{
TimeKey: "datetime",
LevelKey: "level",
NameKey: "logger",
CallerKey: "caller",
MessageKey: "message",
StacktraceKey: "stacktrace",
LineEnding: zapcore.DefaultLineEnding,
EncodeLevel: zapcore.LowercaseLevelEncoder, // 小写编码器
EncodeTime: func(t time.Time, encoder zapcore.PrimitiveArrayEncoder) {
encoder.AppendString(t.Format("2006-01-02 15:04:05"))
},
EncodeDuration: zapcore.SecondsDurationEncoder,
EncodeCaller: zapcore.FullCallerEncoder, // 全路径编码器
EncodeName: zapcore.FullNameEncoder,
}
// 设置日志级别区间
atomicLevel := zap.NewAtomicLevel()
switch opt.LogLevel {
case "debug":
atomicLevel.SetLevel(zap.DebugLevel)
case "info":
atomicLevel.SetLevel(zap.InfoLevel)
case "warn":
atomicLevel.SetLevel(zap.WarnLevel)
case "error":
atomicLevel.SetLevel(zap.ErrorLevel)
default:
atomicLevel.SetLevel(zap.ErrorLevel)
}
// 指定日志等级
//atomicLevel = zap.LevelEnablerFunc(func(level zapcore.Level) bool {
// return level == zap.ErrorLevel
//})
var ws zapcore.WriteSyncer
if opt.LogPath != "" {
hook := lumberjack.Logger{
Filename: opt.LogPath, // 日志文件路径
MaxSize: opt.FileMaxSize, // 每个日志文件保存的最大尺寸 单位:M
MaxBackups: opt.FileMaxBackups, // 日志文件最多保存多少个备份
MaxAge: opt.FileMaxAge, // 文件最多保存多少天
Compress: opt.FileCompress, // 是否压缩
}
ws = zapcore.NewMultiWriteSyncer(zapcore.AddSync(&hook))
} else {
ws = os.Stdout
}
// 一个 NewCore 代表一个日志文件;若需要多个日志文件,则需要多个 NewCore
core := zapcore.NewCore(
zapcore.NewJSONEncoder(encoderConfig), // 编码器配置
ws,
atomicLevel, // 日志级别
)
logger = zap.New(core, zap.AddCaller())
}
func Logger() *zap.Logger {
return logger
}