/
logger.go
68 lines (59 loc) · 1.45 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
package logging
import (
"fmt"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
var Config *zap.Config
var Log *zap.Logger
var SugaredLog *zap.SugaredLogger
func InitGlobalLogger() error {
fmt.Println("Initialize global logger")
cfg, cfgErr := loadConfig()
if cfgErr != nil {
return cfgErr
}
level, levelErr := getZapLevel(cfg.level)
if levelErr != nil {
return levelErr
}
Config = &zap.Config{
Encoding: cfg.encoding,
Level: zap.NewAtomicLevelAt(level),
OutputPaths: []string{"stdout"},
ErrorOutputPaths: []string{"stderr"},
EncoderConfig: buildEncoderConfig(level),
}
Log, _ = Config.Build()
SugaredLog = Log.Sugar()
return nil
}
func getZapLevel(levelString string) (zapcore.Level, error) {
level := zapcore.InfoLevel
err := level.Set(levelString)
if err != nil {
return zapcore.InfoLevel, err
}
return level, nil
}
func buildEncoderConfig(level zapcore.Level) zapcore.EncoderConfig {
if level == zapcore.DebugLevel {
return zapcore.EncoderConfig{
TimeKey: "time",
EncodeTime: zapcore.ISO8601TimeEncoder,
LevelKey: "level",
EncodeLevel: zapcore.CapitalLevelEncoder,
CallerKey: "caller",
EncodeCaller: zapcore.ShortCallerEncoder,
MessageKey: "message",
}
} else {
return zapcore.EncoderConfig{
TimeKey: "time",
EncodeTime: zapcore.ISO8601TimeEncoder,
LevelKey: "level",
EncodeLevel: zapcore.CapitalLevelEncoder,
MessageKey: "message",
}
}
}