/
logzap.go
57 lines (45 loc) · 1.11 KB
/
logzap.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
package logz
import (
"os"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
var logger = zap.NewExample()
type Option func()
func WithLogEnv(env string) Option {
return func() {
var cfg zap.Config
cfg = zap.NewProductionConfig()
switch env {
case "prod":
cfg.Level = zap.NewAtomicLevelAt(zapcore.InfoLevel)
case "debug":
cfg.Level = zap.NewAtomicLevelAt(zapcore.DebugLevel)
default:
cfg = zap.NewDevelopmentConfig()
}
cfg.EncoderConfig = zapcore.EncoderConfig{
MessageKey: "message",
LevelKey: "level",
EncodeLevel: zapcore.CapitalLevelEncoder,
TimeKey: "@timestamp",
EncodeTime: zapcore.ISO8601TimeEncoder,
CallerKey: "caller",
EncodeCaller: zapcore.ShortCallerEncoder,
}
encoder := zapcore.NewJSONEncoder(cfg.EncoderConfig)
core := zapcore.NewCore(encoder, zapcore.Lock(os.Stdout), cfg.Level)
*logger = *zap.New(core, zap.AddCaller())
}
}
func WithAppName(appName string) Option {
return func() {
*logger = *logger.With(zap.String("app", appName))
}
}
func NewLogger(opts ...Option) *zap.Logger {
for _, opt := range opts {
opt()
}
return logger
}