使用logrus记录gorm sql

Posted ball球

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了使用logrus记录gorm sql相关的知识,希望对你有一定的参考价值。

gorm可以方便的输出执行的sql或慢查询。logrus是常用的日志组件,如何将gorm输出的sql通过logrus记录到日志中呢?

如果你还不知道如何让gorm输出sql,请猛击这里

1. 代码实现

我们先看代码。

思路:使用logrus,实现gorm/logger.Writer接口

var DB *gorm.DB

//定义自己的Writer
type MyWriter struct 
	mlog *logrus.Logger


//实现gorm/logger.Writer接口 
func (m *MyWriter)Printf(format string , v ...interface)
	logstr  := fmt.Sprintf(format, v...)
	//利用loggus记录日志
	m.mlog.Info(logstr)


func NewMyWriter() *MyWriter
	log := logrus.New()
	//配置logrus
	log.SetFormatter(&logrus.JSONFormatter
		TimestampFormat:"2006-01-02 15:04:05",  
	)

	return &MyWritermlog:log


//DB初始化
func Init() error 
	var err error
	dsn := "root:user237@tcp(10.160.75.237:3306)/mydb?charset=utf8&parseTime=true&loc=Local"

	slowLogger := logger.New(
		//设置Logger
		NewMyWriter(),
		logger.Config
		     //慢SQL阈值
			SlowThreshold: time.Millisecond,
			//设置日志级别,只有Warn以上才会打印sql
			LogLevel:      logger.Warn,
		,
	)

	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config
		Logger:slowLogger,
	)

	return err


... ...

func main()
	//初始化
	_ = Init()

	d := User
		Name:"ball",
		Age:0,
		IsAdmin:true,
		IsValid:false,
		LoginTime:time.Now(),
	

	DB.Create(&d)

输出

"level":"info","msg":"/home/ballqiu/go/gorm/main.go:90 SLOW SQL \\u003e= 1ms\\n[4.333ms] [rows:1] INSERT INTO `User` (`name`,`age`,`is_admin`,`is_valid`,`login_time`) VALUES ('ball',0,true,false,'2021-05-14 18:58:17.857')","time":"2021-05-14 18:58:17"

如果你将logrus配置输出至文件,日志就会写入文件中。

2. 代码解读

在作详细说明之前,先来看下上述代码使用到的interface和struct之间的关系。

  • gorm.Open()的第二参数为Option接口,因为gorm.Config实现了该接口,因此可以作为参数传入。
/*gorm.go*/

func Open(dialector Dialector, opts ...Option) (db *DB, err error)
  • gorm.Config包含Logger成员,该成员为logger.Interface类型。logger.New()返回的是一个struct logger实例,它实现了Interface接口,因此可以赋值给gorm.Config.Logger。
    slowLogger := logger.New()
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config
		Logger:slowLogger,
	)
  • logger.New的第一参数为logger.Writer接口,我们的MyWriter实现了该接口,因此可以传入。
/*gorm/logger/logger.go*/

func New(writer Writer, config Config) Interface 
    ...
    
    //
    return &logger
		//传入的writer在此处赋给了logger
		Writer:       writer,
		Config:       config,
		infoStr:      infoStr,
		warnStr:      warnStr,
		errStr:       errStr,
		traceStr:     traceStr,
		traceWarnStr: traceWarnStr,
		traceErrStr:  traceErrStr,
	


type Writer interface 
	Printf(string, ...interface)

  • logger.** 在输出时调用了Printf(Writer接口),也就是我们自己的MyWriter.Printf,因此日志得以写入logrus
/*gorm/logger/logger.go*/
func (l logger) Info(ctx context.Context, msg string, data ...interface) 
	if l.LogLevel >= Info 
		l.Printf(l.infoStr+msg, append([]interfaceutils.FileWithLineNum(), data...)...)
	

以上是关于使用logrus记录gorm sql的主要内容,如果未能解决你的问题,请参考以下文章

gorm的使用以及错误记录

gorm软删除和硬删除

记录gorm union和软删除一起使用遇到的一个奇怪问题

记录gorm union和软删除一起使用遇到的一个奇怪问题

Gorm框架学习---CRUD接口之创建

知识分享之Golang——使用gorm时进行执行自定义SQL的几种方式