先看看gorm自己带的日志
原文 https://gorm.io/zh_CN/docs/logger.html
newLogger := logger.New(
log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer(日志输出的目标,前缀和日志包含的内容——译者注)
logger.Config{
SlowThreshold: time.Second, // 慢 SQL 阈值
LogLevel: logger.Silent, // 日志级别
IgnoreRecordNotFoundError: true, // 忽略ErrRecordNotFound(记录未找到)错误
Colorful: false, // 禁用彩色打印
},
)
// 全局模式
db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{
Logger: newLogger,
})
// 新建会话模式
tx := db.Session(&Session{Logger: newLogger})
tx.First(&user)
tx.Model(&user).Update("Age", 18)
自定义 Logger
参考 GORM 的 默认 logger 来定义您自己的 logger
Logger 需要实现以下接口,它接受 context,所以你可以用它来追踪日志
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)
}
我们实现一个
我这里选了go-zero 来处理orm的日志
package db
import (
"context"
"fmt"
"github.com/zeromicro/go-zero/core/logx"
"gorm.io/gorm/logger"
"time"
)
//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)
//}
type dbLog struct {
LogLevel logger.LogLevel
}
func New() *dbLog {
return new(dbLog)
}
func (l *dbLog) LogMode(level logger.LogLevel) logger.Interface {
l.LogLevel = level
return l
}
func (l *dbLog) Info(ctx context.Context, msg string, data ...interface{}) {
if l.LogLevel < logger.Info {
return
}
logx.WithContext(ctx).Debugf(msg, data)
}
func (l *dbLog) Warn(ctx context.Context, msg string, data ...interface{}) {
if l.LogLevel < logger.Warn {
return
}
logx.WithContext(ctx).Infof(msg, data)
}
func (l *dbLog) Error(ctx context.Context, msg string, data ...interface{}) {
if l.LogLevel < logger.Error {
return
}
logx.WithContext(ctx).Errorf(msg, data)
}
func (l *dbLog) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
//这块的逻辑可以自己根据业务情况修改
fmt.Println(l.LogLevel)
elapsed := time.Since(begin)
sql, rows := fc()
logx.WithContext(ctx).WithDuration(elapsed).Slowf("Trace sql: %v row: %v err: %v", sql, rows, err)
}