You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
1.9 KiB
75 lines
1.9 KiB
package internal
|
|
|
|
import (
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
"os"
|
|
"tgk-touch/internal/global"
|
|
"time"
|
|
)
|
|
|
|
type ZapCore struct {
|
|
level zapcore.Level
|
|
zapcore.Core
|
|
}
|
|
|
|
func NewZapCore(level zapcore.Level) *ZapCore {
|
|
entity := &ZapCore{level: level}
|
|
syncer := entity.WriteSyncer()
|
|
levelEnabler := zap.LevelEnablerFunc(func(l zapcore.Level) bool {
|
|
return l == level
|
|
})
|
|
entity.Core = zapcore.NewCore(g.GVA_Config.Zap.Encoder(), syncer, levelEnabler)
|
|
return entity
|
|
}
|
|
|
|
func (z *ZapCore) WriteSyncer(formats ...string) zapcore.WriteSyncer {
|
|
|
|
cutter := NewCutter(
|
|
g.GVA_Config.Zap.Director,
|
|
z.level.String(),
|
|
g.GVA_Config.Zap.RetentionDay,
|
|
CutterWithLayout(time.DateOnly),
|
|
CutterWithFormats(formats...),
|
|
CutterWithMaxSize(g.GVA_Config.Zap.MaxSize), // 传入最大大小(MB)
|
|
CutterWithCompress(g.GVA_Config.Zap.Compress), // 传入是否压缩
|
|
CutterWithCompressLevel(g.GVA_Config.Zap.CompressLevel),
|
|
CutterWithCheckInterval(g.GVA_Config.Zap.CheckInterval),
|
|
)
|
|
if g.GVA_Config.Zap.LogInConsole {
|
|
multiSyncer := zapcore.NewMultiWriteSyncer(os.Stdout, cutter)
|
|
return zapcore.AddSync(multiSyncer)
|
|
}
|
|
return zapcore.AddSync(cutter)
|
|
}
|
|
|
|
func (z *ZapCore) Enabled(level zapcore.Level) bool {
|
|
return z.level == level
|
|
}
|
|
|
|
func (z *ZapCore) With(fields []zapcore.Field) zapcore.Core {
|
|
return z.Core.With(fields)
|
|
}
|
|
|
|
func (z *ZapCore) Check(entry zapcore.Entry, check *zapcore.CheckedEntry) *zapcore.CheckedEntry {
|
|
if z.Enabled(entry.Level) {
|
|
return check.AddCore(entry, z)
|
|
}
|
|
return check
|
|
}
|
|
|
|
func (z *ZapCore) Write(entry zapcore.Entry, fields []zapcore.Field) error {
|
|
for i := 0; i < len(fields); i++ {
|
|
if fields[i].Key == "business" || fields[i].Key == "folder" || fields[i].Key == "directory" {
|
|
syncer := z.WriteSyncer(fields[i].String)
|
|
z.Core = zapcore.NewCore(g.GVA_Config.Zap.Encoder(), syncer, z.level)
|
|
}
|
|
}
|
|
|
|
return z.Core.Write(entry, fields)
|
|
}
|
|
|
|
func (z *ZapCore) Sync() error {
|
|
return z.Core.Sync()
|
|
}
|