/
logger.go
71 lines (54 loc) · 1.24 KB
/
logger.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
package cmd
import (
"io"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
var (
log *zap.SugaredLogger
zapLog *zap.Logger
logLevel zap.AtomicLevel
)
const (
logInfoLevel = "info"
logDebugLevel = "debug"
)
type nopSync struct {
io.Writer
}
func (nopSync) Sync() error { return nil }
func getLogger(stdout zapcore.WriteSyncer, levelName string) *zap.Logger {
logLevel = zap.NewAtomicLevel()
logLevel.SetLevel(zap.InfoLevel)
encoderCfg := zap.NewProductionEncoderConfig()
encoderCfg.LevelKey = ""
encoderCfg.TimeKey = ""
switch levelName {
case logInfoLevel:
case logDebugLevel:
encoderCfg.TimeKey = "timestamp"
encoderCfg.EncodeTime = zapcore.TimeEncoderOfLayout("2006-01-02 15:04:05")
logLevel.SetLevel(zap.DebugLevel)
}
zapLog := zap.New(zapcore.NewCore(
zapcore.NewConsoleEncoder(encoderCfg),
zapcore.Lock(stdout),
logLevel,
))
return zapLog
}
func getStdLogger(out io.Writer, levelName string) *zap.Logger {
var x zapcore.WriteSyncer
x, ok := out.(zapcore.WriteSyncer)
if !ok {
x = nopSync{out}
}
return getLogger(x, levelName)
}
func setLogger(out io.Writer, levelName string, named string) {
zapLog = getStdLogger(out, levelName)
log = zapLog.Sugar()
if named != "" {
log = log.Named(named)
}
}