🎨 优化gorm日志配置

gorm.go

@@ -1,21 +1,85 @@
 package logger
 
-import "strings"
+import (
+	"context"
+	"errors"
+	"fmt"
+	gl "gorm.io/gorm/logger"
+	"time"
+)
 
 // 基于Gorm的日志实现
-type gormLogger struct{}
-
-// 打印
-func (gormLogger) Write(p []byte) (n int, err error) {
-	str := string(p)
-	// 去掉第一行
-	//str = strings.Split(str, "\n")[1]
-	str = strings.Join(strings.Split(str, "\n")[1:], " ")
-	Say.Debug(str)
-	return 0, nil
+type gormLogger struct {
+	gl.Config
 }
 
-// NewGormLogger ...
-func NewGormLogger() *gormLogger {
-	return &gormLogger{}
+// LogMode 实现LogMode接口
+func (l *gormLogger) LogMode(level gl.LogLevel) gl.Interface {
+	nl := *l
+	nl.LogLevel = level
+	return &nl
+}
+
+// Info 实现Info接口
+func (l gormLogger) Info(ctx context.Context, msg string, data ...interface{}) {
+	if l.LogLevel >= gl.Info {
+		//	// 去掉第一行
+		//	msg = strings.Join(strings.Split(msg, "\n")[1:], " ")
+		//	Say.Info(msg)
+		//
+		//	l.Printf(msg, append([]interface{}{utils.FileWithLineNum()}, data...)...)
+	}
+}
+
+// Warn 实现Warn接口
+func (l gormLogger) Warn(ctx context.Context, msg string, data ...interface{}) {
+	if l.LogLevel >= gl.Warn {
+		//
+	}
+}
+
+// 实现Error接口
+func (l gormLogger) Error(ctx context.Context, msg string, data ...interface{}) {
+	if l.LogLevel >= gl.Error {
+		//
+	}
+}
+
+// Trace 实现Trace接口
+func (l gormLogger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error) {
+	if l.LogLevel <= gl.Silent {
+		return
+	}
+
+	elapsed := time.Since(begin)
+	sql, rows := fc()
+	msg := fmt.Sprintf("[%v] [rows:%v] %s", elapsed.String(), rows, sql)
+	if rows == -1 {
+		msg = fmt.Sprintf("%s [-] %s", elapsed.String(), sql)
+	}
+
+	switch {
+	case err != nil && l.LogLevel >= gl.Error && (!errors.Is(err, gl.ErrRecordNotFound) || !l.IgnoreRecordNotFoundError):
+		Say.Errorf("%s -> %s", err.Error(), sql)
+	case elapsed > l.SlowThreshold && l.SlowThreshold != 0 && l.LogLevel >= gl.Warn:
+		slowLog := fmt.Sprintf("SLOW SQL >= %v", l.SlowThreshold)
+		Say.Warnf("%v -> %v", slowLog, sql)
+	case l.LogLevel == gl.Info:
+		Say.Info(msg)
+	}
+}
+
+// NewGormLoggerWithConfig ...
+func NewGormLoggerWithConfig(config gl.Config) gl.Interface {
+	return &gormLogger{config}
+}
+
+// DefaultGormLogger 默认的日志实现
+func DefaultGormLogger() gl.Interface {
+	return &gormLogger{gl.Config{
+		SlowThreshold:             time.Second, // Slow SQL threshold
+		IgnoreRecordNotFoundError: false,       // 忽略没找到结果的错误
+		LogLevel:                  gl.Info,     // Log level
+		Colorful:                  false,       // Disable color
+	}}
 }