/
zap.go
106 lines (88 loc) · 2.34 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
package log
import (
"io"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
// ZapLoggerAdapter implements Logger for zap.
// It seems somewhat complicated and verbose for wrapping a logger
// but I don't see how to to do it better in Go
type ZapLoggerAdapter struct {
logger *zap.SugaredLogger
}
func (l *ZapLoggerAdapter) Debug(args ...any) {
mapFields(args)
l.logger.Debug(args...)
}
func (l *ZapLoggerAdapter) Debugw(msg string, keysAndValues ...any) {
mapFields(keysAndValues)
l.logger.Debugw(msg, keysAndValues...)
}
func (l *ZapLoggerAdapter) Info(args ...any) {
mapFields(args)
l.logger.Info(args...)
}
func (l *ZapLoggerAdapter) Infow(msg string, keysAndValues ...any) {
mapFields(keysAndValues)
l.logger.Infow(msg, keysAndValues...)
}
func (l *ZapLoggerAdapter) Warn(args ...any) {
mapFields(args)
l.logger.Warn(args...)
}
func (l *ZapLoggerAdapter) Warnw(msg string, keysAndValues ...any) {
mapFields(keysAndValues)
l.logger.Warnw(msg, keysAndValues...)
}
func (l *ZapLoggerAdapter) Error(args ...any) {
mapFields(args)
l.logger.Error(args...)
}
func (l *ZapLoggerAdapter) Errorw(msg string, keysAndValues ...any) {
mapFields(keysAndValues)
l.logger.Errorw(msg, keysAndValues...)
}
func (l *ZapLoggerAdapter) Fatal(args ...any) {
mapFields(args)
l.logger.Fatal(args...)
}
func (l *ZapLoggerAdapter) Fatalw(msg string, keysAndValues ...any) {
mapFields(keysAndValues)
l.logger.Fatalw(msg, keysAndValues...)
}
func (l *ZapLoggerAdapter) Named(s string) Logger {
named := l.logger.Named(s)
return &ZapLoggerAdapter{logger: named}
}
func (l *ZapLoggerAdapter) With(args ...any) Logger {
with := l.logger.With(args...)
return &ZapLoggerAdapter{logger: with}
}
func NewZapLogger(
out io.Writer,
verbose bool,
veryVerbose bool,
) *zap.Logger {
encoder := zapcore.NewJSONEncoder(zap.NewProductionEncoderConfig())
writer := zapcore.Lock(zapcore.AddSync(out))
levelEnabler := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
switch {
case veryVerbose:
return lvl >= zap.DebugLevel
case verbose:
return lvl >= zap.InfoLevel
default:
return lvl >= zap.ErrorLevel
}
})
core := zapcore.NewCore(encoder, writer, levelEnabler)
return zap.New(core)
}
func NewZapLoggerAdapter(
zl *zap.Logger,
) *ZapLoggerAdapter {
return &ZapLoggerAdapter{logger: zl.Sugar()}
}
func RedirectStdLog(l *zap.Logger) {
zap.RedirectStdLog(l)
}