/
zap.go
92 lines (78 loc) · 1.91 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
package logger
import (
"io"
"os"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"github.com/hiromaily/go-google-home/pkg/config"
)
// LogEnv dev/prod/custom
type LogEnv string
// log_type
const (
LogDev LogEnv = "dev"
LogProd LogEnv = "prod"
LogCustom LogEnv = "custom"
)
// String converter
func (e LogEnv) String() string {
return string(e)
}
// NewLoggerWithWriter returns *zap.Logger
func NewLoggerWithWriter(w io.Writer, env LogEnv, lv zapcore.LevelEnabler, isStackTrace bool) *zap.Logger {
zap.NewExample()
writer := zapcore.AddSync(w)
var encoderCfg zapcore.EncoderConfig
switch env {
case LogDev:
encoderCfg = zap.NewDevelopmentEncoderConfig()
case LogCustom:
encoderCfg = zapcore.EncoderConfig{
MessageKey: "msg",
LevelKey: "lv",
EncodeLevel: zapcore.LowercaseLevelEncoder,
EncodeTime: zapcore.ISO8601TimeEncoder,
EncodeDuration: zapcore.StringDurationEncoder,
}
// case LogProd:
default:
encoderCfg = zap.NewProductionEncoderConfig()
encoderCfg.TimeKey = "time"
encoderCfg.EncodeTime = zapcore.ISO8601TimeEncoder
}
jsonEncoder := zapcore.NewJSONEncoder(encoderCfg)
cores := []zapcore.Core{
zapcore.NewCore(jsonEncoder, writer, lv),
}
options := []zap.Option{
zap.AddCaller(),
zap.AddCallerSkip(1),
}
if isStackTrace {
options = append(options, zap.AddStacktrace(zap.ErrorLevel))
}
logger := zap.New(zapcore.NewTee(cores...), options...)
return logger
}
func getLogLevel(level string) zapcore.LevelEnabler {
switch level {
case "debug":
return zapcore.DebugLevel
case "info":
return zapcore.InfoLevel
case "warn":
return zapcore.WarnLevel
case "error":
return zapcore.ErrorLevel
default:
return zapcore.InfoLevel
}
}
// NewZapLogger returns *zap.Logger
func NewZapLogger(conf *config.Logger) *zap.Logger {
return NewLoggerWithWriter(
os.Stdout,
LogEnv(conf.Env),
getLogLevel(conf.Level),
conf.IsStackTrace).Named(conf.Service)
}