/
logger.go
83 lines (71 loc) · 2.12 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
72
73
74
75
76
77
78
79
80
81
82
83
package logging
import (
"os"
"github.com/natefinch/lumberjack"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"github.com/khighness/highness-paxos-kv/pkg/toolkit"
)
// @Author Chen Zikang
// @Email parakovo@gmail.com
// @Since 2022-09-08
// init initializes zap.Logger then we can use zap.L() to get this logger.
func init() {
var core zapcore.Core
fileCore := zapcore.NewCore(zapFileEncoder(), zapWriteSyncer(), zapLevelEnabler())
if toolkit.IsDeployLocal() {
consoleCore := zapcore.NewCore(zapConsoleEncoder(), os.Stdout, zapLevelEnabler())
core = zapcore.NewTee(fileCore, consoleCore)
} else {
core = zapcore.NewTee(fileCore)
}
logger := zap.New(core, zap.AddCaller())
zap.ReplaceGlobals(logger)
}
func zapLevelEnabler() zapcore.Level {
return zapcore.InfoLevel
}
func zapEncodeConfig() zapcore.EncoderConfig {
return zapcore.EncoderConfig{
MessageKey: "msg",
LevelKey: "level",
TimeKey: "ts",
NameKey: "logger",
CallerKey: "caller_line",
FunctionKey: zapcore.OmitKey,
StacktraceKey: "stacktrace",
LineEnding: "\n",
EncodeLevel: zapEncodeLevel,
EncodeTime: zapcore.ISO8601TimeEncoder,
EncodeDuration: zapcore.MillisDurationEncoder,
EncodeCaller: zapEncodeCaller,
ConsoleSeparator: " ",
}
}
func zapFileEncoder() zapcore.Encoder {
return zapcore.NewJSONEncoder(zapEncodeConfig())
}
func zapConsoleEncoder() zapcore.Encoder {
return zapcore.NewConsoleEncoder(zapEncodeConfig())
}
func zapEncodeLevel(level zapcore.Level, enc zapcore.PrimitiveArrayEncoder) {
enc.AppendString("[" + level.CapitalString() + "]")
}
func zapEncodeCaller(caller zapcore.EntryCaller, enc zapcore.PrimitiveArrayEncoder) {
enc.AppendString(caller.TrimmedPath())
}
func zapWriteSyncer() zapcore.WriteSyncer {
logDir := "log/"
if serviceName := os.Getenv("SERVICE_NAME"); serviceName != "" {
logDir = "/data/log/" + serviceName
}
_ = os.Mkdir(logDir, 0777)
lumberJackLogger := &lumberjack.Logger{
Filename: logDir + "/app.log",
MaxSize: 1000,
MaxBackups: 10,
MaxAge: 30,
Compress: true,
}
return zapcore.AddSync(lumberJackLogger)
}