134 lines
3.8 KiB
Go
134 lines
3.8 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"git.echol.cn/loser/st/server/global"
|
|
"git.echol.cn/loser/st/server/model/system"
|
|
"git.echol.cn/loser/st/server/service"
|
|
astutil "git.echol.cn/loser/st/server/utils/ast"
|
|
"git.echol.cn/loser/st/server/utils/stacktrace"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
"os"
|
|
"strings"
|
|
"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(global.GVA_CONFIG.Zap.Encoder(), syncer, levelEnabler)
|
|
return entity
|
|
}
|
|
|
|
func (z *ZapCore) WriteSyncer(formats ...string) zapcore.WriteSyncer {
|
|
cutter := NewCutter(
|
|
global.GVA_CONFIG.Zap.Director,
|
|
z.level.String(),
|
|
global.GVA_CONFIG.Zap.RetentionDay,
|
|
CutterWithLayout(time.DateOnly),
|
|
CutterWithFormats(formats...),
|
|
)
|
|
if global.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(global.GVA_CONFIG.Zap.Encoder(), syncer, z.level)
|
|
}
|
|
}
|
|
// 先写入原日志目标
|
|
err := z.Core.Write(entry, fields)
|
|
|
|
// 捕捉 Error 及以上级别日志并入库，且可提取 zap.Error(err) 的错误内容
|
|
if entry.Level >= zapcore.ErrorLevel {
|
|
// 避免与 GORM zap 写入互相递归：跳过由 gorm logger writer 触发的日志
|
|
if strings.Contains(entry.Caller.File, "gorm_logger_writer.go") {
|
|
return err
|
|
}
|
|
|
|
form := "后端"
|
|
level := entry.Level.String()
|
|
// 生成基础信息
|
|
info := entry.Message
|
|
|
|
// 提取 zap.Error(err) 内容
|
|
var errStr string
|
|
for i := 0; i < len(fields); i++ {
|
|
f := fields[i]
|
|
if f.Type == zapcore.ErrorType || f.Key == "error" || f.Key == "err" {
|
|
if f.Interface != nil {
|
|
errStr = fmt.Sprintf("%v", f.Interface)
|
|
} else if f.String != "" {
|
|
errStr = f.String
|
|
}
|
|
break
|
|
}
|
|
}
|
|
if errStr != "" {
|
|
info = fmt.Sprintf("%s | 错误: %s", info, errStr)
|
|
}
|
|
|
|
// 附加来源与堆栈信息
|
|
if entry.Caller.File != "" {
|
|
info = fmt.Sprintf("%s \n 源文件:%s:%d", info, entry.Caller.File, entry.Caller.Line)
|
|
}
|
|
stack := entry.Stack
|
|
if stack != "" {
|
|
info = fmt.Sprintf("%s \n 调用栈：%s", info, stack)
|
|
// 解析最终业务调用方，并提取其方法源码
|
|
if frame, ok := stacktrace.FindFinalCaller(stack); ok {
|
|
fnName, fnSrc, sLine, eLine, exErr := astutil.ExtractFuncSourceByPosition(frame.File, frame.Line)
|
|
if exErr == nil {
|
|
info = fmt.Sprintf("%s \n 最终调用方法:%s:%d (%s lines %d-%d)\n----- 产生日志的方法代码如下 -----\n%s", info, frame.File, frame.Line, fnName, sLine, eLine, fnSrc)
|
|
} else {
|
|
info = fmt.Sprintf("%s \n 最终调用方法:%s:%d (%s) | extract_err=%v", info, frame.File, frame.Line, fnName, exErr)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 使用后台上下文，避免依赖 gin.Context
|
|
ctx := context.Background()
|
|
_ = service.ServiceGroupApp.SystemServiceGroup.SysErrorService.CreateSysError(ctx, &system.SysError{
|
|
Form: &form,
|
|
Info: &info,
|
|
Level: level,
|
|
})
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (z *ZapCore) Sync() error {
|
|
return z.Core.Sync()
|
|
}
|