🎉 初始化项目

server/core/internal/zap_core.go

@@ -0,0 +1,133 @@
+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()
+}