实现 gorm.io/gorm/logger 下的函数⬇️

// gorm 源码
type Interface interface {  LogMode(LogLevel) Interface  Info(context.Context, string, ...interface{})  Warn(context.Context, string, ...interface{})  Error(context.Context, string, ...interface{})  Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error)  
}

以下为自定义的重写实现，主要函数是Trace

package log  import (  "context"  "time"  "github.com/go-kratos/kratos/v2/log""gorm.io/gorm/logger"
)  type GormLogger struct {  SlowThreshold time.Duration  
}  func NewGormLogger() *GormLogger {  return &GormLogger{  SlowThreshold: 200 * time.Millisecond, // 一般超过200毫秒就算慢查所以不使用配置进行更改  }  
}  var _ logger.Interface = (*GormLogger)(nil)  func (l *GormLogger) LogMode(lev logger.LogLevel) logger.Interface {  return &GormLogger{}  
}  
func (l *GormLogger) Info(ctx context.Context, msg string, data ...interface{}) {  log.Context(ctx).Infof(msg, data)  
}  
func (l *GormLogger) Warn(ctx context.Context, msg string, data ...interface{}) {  log.Context(ctx).Errorf(msg, data)  
}  
func (l *GormLogger) Error(ctx context.Context, msg string, data ...interface{}) {  log.Context(ctx).Errorf(msg, data)  
}  
func (l *GormLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {  // 获取运行时间  elapsed := time.Since(begin)  // 获取 SQL 语句和返回条数  sql, rows := fc()  // Gorm 错误时打印if err != nil {  log.Context(ctx).Errorf("SQL ERROR, | sql=%v, rows=%v, elapsed=%v", sql, rows, elapsed)  }  // 慢查询日志  if l.SlowThreshold != 0 && elapsed > l.SlowThreshold {  log.Context(ctx).Warn("Database Slow Log, | sql=%v, rows=%v, elapsed=%v", sql, rows, elapsed)  }  }

这里我使用的log组件是kratos框架的log组件，设置好zap后注入为全局log

在GORM创建连接的地方注入我们重写后的自定义Logger

db, err := gorm.Open(mysql.Open(cfg.Dsn()), &gorm.Config{  QueryFields: true,    Logger: logdef.NewGormLogger(),  // 注入
})

最后，在查询的地方，带上withContext即可

func (ud *userDao) AddOne(c context.Context, user *model.User) (userId int64, err error) {  err = db.GetConn().WithContext(c).Create(user).Error  if err != nil {  log.Context(c).Errorf("AddOne fail|err=%+v", err)  return  }  return user.Id, nil  
}