/
zap.go
93 lines (82 loc) · 2.44 KB
/
zap.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
package logger
import (
app "github.com/ZYallers/zgin/application"
"github.com/natefinch/lumberjack"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"sync"
"sync/atomic"
"time"
)
const (
fileMaxSize = 100 // Unit: MB
maxBackups = 20
fileSuffix = `.log`
cacheMaxSize = 100
)
var (
logDir = app.LogDir
loggerCache, lumberCache sync.Map
loggerCounter int32
levelEnabler zap.LevelEnablerFunc = func(lvl zapcore.Level) bool {
return lvl >= zapcore.DebugLevel
}
jsonEncoder = zapcore.NewJSONEncoder(zapcore.EncoderConfig{
TimeKey: "ts",
LevelKey: "level",
NameKey: "logger",
CallerKey: "caller",
MessageKey: "msg",
StacktraceKey: "stacktrace",
LineEnding: zapcore.DefaultLineEnding,
EncodeLevel: zapcore.LowercaseLevelEncoder,
EncodeTime: func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
enc.AppendString(t.Format("2006/01/02 15:04:05.000"))
},
EncodeDuration: zapcore.SecondsDurationEncoder,
EncodeCaller: zapcore.ShortCallerEncoder,
})
)
func AppLogger() *zap.Logger {
file := logDir + "/" + app.Name + fileSuffix
logger, _ := newLogger(file)
return logger
}
func Use(filename string) *zap.Logger {
if filename == "" {
return nil
}
file := logDir + "/" + app.Name + "-" + filename + fileSuffix
// 判断是否已存在缓存中
if logger, ok := loggerCache.Load(file); ok {
return logger.(*zap.Logger)
}
// 判断容器是否达到最大数量
if atomic.LoadInt32(&loggerCounter) >= cacheMaxSize {
// 超出最大容量,随机删除一半
var counter, clean int32 = 0, cacheMaxSize / 2
loggerCache.Range(func(key, value interface{}) bool {
if counter++; counter > clean {
return false
}
if lumber, ok := lumberCache.Load(key); ok {
_ = lumber.(*lumberjack.Logger).Close()
}
lumberCache.Delete(key)
loggerCache.Delete(key)
return true
})
atomic.AddInt32(&loggerCounter, -clean)
}
logger, hook := newLogger(file)
loggerCache.Store(file, logger)
lumberCache.Store(file, hook)
atomic.AddInt32(&loggerCounter, 1)
return logger
}
func newLogger(filename string) (*zap.Logger, *lumberjack.Logger) {
hook := &lumberjack.Logger{MaxSize: fileMaxSize, MaxBackups: maxBackups, LocalTime: true, Compress: false, Filename: filename}
logger := zap.New(zapcore.NewCore(jsonEncoder, zapcore.AddSync(hook), levelEnabler))
logger.Info("new logger succeed", zap.String("filename", hook.Filename))
return logger, hook
}