/
zap.go
120 lines (111 loc) · 3.04 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
115
116
117
118
119
120
package logger
import (
"context"
"os"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
type ZapLogger struct {
Log *zap.SugaredLogger
Config *zap.Config
}
func NewZapLogger(loggerName, levelStr string) (*ZapLogger, error) {
// Retrieve name of current environment
envName := os.Getenv("CURRENT_ENV_NAME")
if envName == "" {
envName = "default"
}
// If it doesn't exist, make a new one
config := zap.NewProductionConfig()
config.Encoding = "json"
config.EncoderConfig.MessageKey = "msg"
config.EncoderConfig.LevelKey = "level"
config.EncoderConfig.TimeKey = "ts"
config.EncoderConfig.NameKey = "logger"
config.EncoderConfig.CallerKey = "caller"
config.EncoderConfig.StacktraceKey = "stacktrace"
config.EncoderConfig.EncodeLevel = zapcore.LowercaseLevelEncoder
config.EncoderConfig.EncodeTime = zapcore.RFC3339TimeEncoder
config.InitialFields = map[string]interface{}{
"env": envName,
}
logger, err := config.Build(zap.AddCaller(), zap.AddCallerSkip(1))
if err != nil {
return nil, err
}
sugaredLogger := logger.Named(loggerName).Sugar()
zapLogger := &ZapLogger{
Log: sugaredLogger,
Config: &config,
}
zapLogger.SetLevel(levelStr)
return zapLogger, nil
}
func (z *ZapLogger) Debug(args ...interface{}) {
z.Log.Debug(args...)
}
func (z *ZapLogger) Debugf(template string, args ...interface{}) {
z.Log.Debugf(template, args...)
}
func (z *ZapLogger) Debugw(msg string, keysAndValues ...interface{}) {
z.Log.Debugw(msg, keysAndValues...)
}
func (z *ZapLogger) Error(args ...interface{}) {
z.Log.Error(args...)
}
func (z *ZapLogger) Errorf(template string, args ...interface{}) {
z.Log.Errorf(template, args...)
}
func (z *ZapLogger) Errorw(msg string, keysAndValues ...interface{}) {
z.Log.Errorw(msg, keysAndValues)
}
func (z *ZapLogger) Info(args ...interface{}) {
z.Log.Info(args...)
}
func (z *ZapLogger) Infof(template string, args ...interface{}) {
z.Log.Infof(template, args...)
}
func (z *ZapLogger) Infow(msg string, keysAndValues ...interface{}) {
z.Log.Infow(msg, keysAndValues...)
}
func (z *ZapLogger) Named(name string) Logger {
return &ZapLogger{
Log: z.Log.Named(name),
Config: z.Config,
}
}
func (z *ZapLogger) SetLevel(levelStr string) {
level := zapcore.InfoLevel
level.UnmarshalText([]byte(levelStr))
z.Config.Level.SetLevel(level)
}
func (z *ZapLogger) Sync() error {
return z.Log.Sync()
}
func (z *ZapLogger) Warn(args ...interface{}) {
z.Log.Warn(args...)
}
func (z *ZapLogger) Warnf(template string, args ...interface{}) {
z.Log.Warnf(template, args...)
}
func (z *ZapLogger) Warnw(msg string, keysAndValues ...interface{}) {
z.Log.Warnw(msg, keysAndValues...)
}
func (z *ZapLogger) With(args ...interface{}) Logger {
return &ZapLogger{
Log: z.Log.With(args...),
Config: z.Config,
}
}
func (z *ZapLogger) WithTracingParams(ctx context.Context) Logger {
if props, ok := ctx.Value("TracingProperties").(TracingProperties); ok {
return &ZapLogger{
Log: z.Log.With("Properties", props),
Config: z.Config,
}
}
return z
}
func (z *ZapLogger) AsSugaredLogger() *zap.SugaredLogger {
return z.Log
}