/
runelog.go
executable file
·108 lines (90 loc) · 2.35 KB
/
runelog.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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
// Copyright (c) 2022 Runetale Inc & AUTHORS All rights reserved.
// Use of this source code is governed by a BSD 3-Clause License
// license that can be found in the LICENSE file.
package runelog
import (
"fmt"
"net/url"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
lumberjack "gopkg.in/natefinch/lumberjack.v2"
)
const (
DebugLevelStr string = "debug"
InfoLevelStr string = "info"
WarningLevelStr string = "warning"
ErrorLevelStr string = "error"
)
// var (
// globalLogger *zap.Logger
// )
type Runelog struct {
Logger *zap.SugaredLogger
}
func NewRunelog(name string, logLevel string, logFile string, dev bool) (*Runelog, error) {
l, err := initRunelog(logLevel, logFile, dev)
if err != nil {
return nil, err
}
return &Runelog{
Logger: l.Named(name).Sugar(),
}, nil
}
func initRunelog(logLevel string, logFile string, dev bool) (*zap.Logger, error) {
var level zapcore.Level
switch logLevel {
case DebugLevelStr:
level = zap.DebugLevel
case InfoLevelStr:
level = zap.InfoLevel
case WarningLevelStr:
level = zap.WarnLevel
case ErrorLevelStr:
level = zap.ErrorLevel
default:
return nil, fmt.Errorf("unknown log level %s", logLevel)
}
encoderConfig := zapcore.EncoderConfig{
TimeKey: "ts",
LevelKey: "level",
NameKey: "logger",
CallerKey: "caller",
MessageKey: "msg",
StacktraceKey: "stacktrace",
LineEnding: zapcore.DefaultLineEnding,
EncodeLevel: zapcore.LowercaseLevelEncoder,
EncodeTime: zapcore.ISO8601TimeEncoder,
EncodeDuration: zapcore.SecondsDurationEncoder,
}
ll := lumberjack.Logger{
Filename: logFile,
MaxSize: 1024, //MB
MaxBackups: 30, // days
MaxAge: 90, //days
Compress: true,
}
zap.RegisterSink("lumberjack", func(*url.URL) (zap.Sink, error) {
return lumberjackSink{
Logger: &ll,
}, nil
})
loggerConfig := zap.Config{
Level: zap.NewAtomicLevelAt(level),
Development: dev,
Encoding: "console",
EncoderConfig: encoderConfig,
OutputPaths: []string{fmt.Sprintf("lumberjack:%s", logFile), "stderr"},
}
buildedLogger, err := loggerConfig.Build()
if err != nil {
panic(fmt.Sprintf("build zap logger from config error: %v", err))
}
zap.ReplaceGlobals(buildedLogger)
return buildedLogger, nil
}
type lumberjackSink struct {
*lumberjack.Logger
}
func (lumberjackSink) Sync() error {
return nil
}