-
Notifications
You must be signed in to change notification settings - Fork 131
/
zap.go
114 lines (96 loc) · 2.74 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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package lumber
import (
"os"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
lumberjack "gopkg.in/natefinch/lumberjack.v2"
)
type zapLogger struct {
sugaredLogger *zap.SugaredLogger
}
const callDepth = 2
func getEncoder(isJSON bool) zapcore.Encoder {
encoderConfig := zap.NewProductionEncoderConfig()
encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
encoderConfig.TimeKey = "time" // This will change the key from ts to time
if isJSON {
return zapcore.NewJSONEncoder(encoderConfig)
}
return zapcore.NewConsoleEncoder(encoderConfig)
}
func getZapLevel(level string) zapcore.Level {
switch level {
case Info:
return zapcore.InfoLevel
case Warn:
return zapcore.WarnLevel
case Debug:
return zapcore.DebugLevel
case Error:
return zapcore.ErrorLevel
case Fatal:
return zapcore.FatalLevel
default:
return zapcore.InfoLevel
}
}
func newZapLogger(config LoggingConfig, verbose bool) Logger {
cores := []zapcore.Core{}
if config.EnableConsole {
level := getZapLevel(config.ConsoleLevel)
// command line args take highest precedence
if verbose {
level = getZapLevel("debug")
}
writer := zapcore.Lock(os.Stdout)
core := zapcore.NewCore(getEncoder(config.ConsoleJSONFormat), writer, level)
cores = append(cores, core)
}
if config.EnableFile {
level := getZapLevel(config.FileLevel)
writer := zapcore.AddSync(&lumberjack.Logger{
Filename: config.FileLocation,
MaxSize: 100,
Compress: true,
MaxAge: 28,
})
core := zapcore.NewCore(getEncoder(config.FileJSONFormat), writer, level)
cores = append(cores, core)
}
combinedCore := zapcore.NewTee(cores...)
// AddCallerSkip skips 2 number of callers, this is important else the file that gets
// logged will always be the wrapped file. In our case zap.go
logger := zap.New(combinedCore,
zap.AddCallerSkip(callDepth),
zap.AddCaller(),
).Sugar()
return &zapLogger{
sugaredLogger: logger,
}
}
func (l *zapLogger) Debugf(format string, args ...interface{}) {
l.sugaredLogger.Debugf(format, args...)
}
func (l *zapLogger) Infof(format string, args ...interface{}) {
l.sugaredLogger.Infof(format, args...)
}
func (l *zapLogger) Warnf(format string, args ...interface{}) {
l.sugaredLogger.Warnf(format, args...)
}
func (l *zapLogger) Errorf(format string, args ...interface{}) {
l.sugaredLogger.Errorf(format, args...)
}
func (l *zapLogger) Fatalf(format string, args ...interface{}) {
l.sugaredLogger.Fatalf(format, args...)
}
func (l *zapLogger) Panicf(format string, args ...interface{}) {
l.sugaredLogger.Panicf(format, args...)
}
func (l *zapLogger) WithFields(fields Fields) Logger {
var f = make([]interface{}, 0, len(fields))
for k, v := range fields {
f = append(f, k, v)
}
newLogger := l.sugaredLogger.With(f...)
return &zapLogger{newLogger}
}