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 logger
import (
"errors"
"os"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"gopkg.in/natefinch/lumberjack.v2"
)
var logger *zap.Logger = nil
// Set up a new Zap logger. If `onlyFile` is true, set up the logger to work
// only on file, else prints on stdout
func InitLogger(logFile string, onlyFile bool) *zap.Logger {
cfg := zap.NewProductionConfig()
cfg.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
var core zapcore.Core
lumberjackLogger := &lumberjack.Logger{
Filename: logFile,
MaxSize: 100,
MaxBackups: 5,
MaxAge: 30,
Compress: true,
}
fileCore := zapcore.NewCore(
zapcore.NewJSONEncoder(cfg.EncoderConfig),
zapcore.AddSync(lumberjackLogger),
cfg.Level,
)
if onlyFile {
core = fileCore
} else {
consoleCore := zapcore.NewCore(
zapcore.NewConsoleEncoder(cfg.EncoderConfig),
zapcore.Lock(os.Stdout),
cfg.Level,
)
core = zapcore.NewTee(fileCore, consoleCore)
}
logger = zap.New(core)
return logger
}
// Return the global Zap logger after calling `InitLogger` method
func GetLogger() (*zap.Logger, error) {
if logger == nil {
return nil, errors.New("You must call `InitLogger()` first.")
}
return logger, nil
}
|
