:truck:将gorm日志移动到plugins下

plugins/gorm.go

@@ -0,0 +1,76 @@
+package plugins
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"code.mrx.ltd/pkg/zap_logger/log"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+var DefaultGormLogger = NewGormLogger(logger.Config{
+	SlowThreshold:             2 * time.Second,
+	Colorful:                  false,
+	IgnoreRecordNotFoundError: false,
+	ParameterizedQueries:      false,
+	LogLevel:                  logger.Info,
+})
+
+type gormLoggerPlugin struct{ logger.Config }
+
+func NewGormLogger(gl logger.Config) *gormLoggerPlugin {
+	return &gormLoggerPlugin{gl}
+}
+
+func (g *gormLoggerPlugin) LogMode(level logger.LogLevel) logger.Interface {
+	g.LogLevel = level
+	return g
+}
+
+func (g *gormLoggerPlugin) Info(ctx context.Context, msg string, args ...interface{}) {
+	if g.LogLevel >= logger.Info {
+		log.Infof(msg, args...)
+	}
+}
+
+func (g *gormLoggerPlugin) Warn(ctx context.Context, msg string, args ...interface{}) {
+	if g.LogLevel >= logger.Warn {
+		log.Warnf(msg, args...)
+	}
+}
+
+func (g *gormLoggerPlugin) Error(ctx context.Context, msg string, args ...interface{}) {
+	if g.LogLevel >= logger.Error {
+		log.Errorf(msg, args...)
+	}
+}
+
+func (g *gormLoggerPlugin) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
+	if g.LogLevel <= logger.Silent {
+		return
+	}
+
+	sql, rows := fc()
+	elapsed := time.Since(begin)
+	msg := fmt.Sprintf("[%v] [rows:%v] %s", elapsed, rows, sql)
+	if rows == -1 {
+		msg = fmt.Sprintf("[%v] [rows:%v] %s", elapsed, "-", sql)
+	}
+	if elapsed > g.SlowThreshold && g.SlowThreshold != 0 && g.LogLevel >= logger.Warn {
+		msg = fmt.Sprintf("[SLOW SQL] [%v] [rows:%v] %s", elapsed, rows, sql)
+	}
+
+	switch {
+	case err != nil && g.LogLevel >= logger.Error && (!errors.Is(err, gorm.ErrRecordNotFound) || !g.IgnoreRecordNotFoundError):
+		msg = fmt.Sprintf("%s -> %s", err.Error(), msg)
+		g.Error(ctx, msg)
+	case elapsed > g.SlowThreshold && g.SlowThreshold != 0 && g.LogLevel >= logger.Warn:
+		msg = fmt.Sprintf("SLOW SQL -> %s", msg)
+		g.Warn(ctx, msg)
+	case g.LogLevel == logger.Info:
+		g.Info(ctx, msg)
+	}
+}