gorm的日志模块源码解析

最新推荐文章于 2024-04-30 17:20:58 发布

轩脉刃

最新推荐文章于 2024-04-30 17:20:58 发布

阅读量2.6k

点赞数 2

本文链接：https://blog.csdn.net/qq_42015552/article/details/90255750

版权

如何让gorm的日志按照我的格式进行输出

这个问题是《如何为gorm日志加traceId》之后，一个群里的朋友问我的。如何让gorm的sql日志不打印到控制台，而打印到自己的日志文件中去。正好我实现了这个功能，就记录一下，并且再把gorm的logger这个线捋一下。

首先我写了一个demo来实现设置我自己的Logger。其实非常简单，只要实现print方法就行了。

 1package main
 2
 3import (
 4    "fmt"
 5    "github.com/jinzhu/gorm"
 6    _ "github.com/jinzhu/gorm/dialects/mysql"
 7    "log"
 8)
 9
10type T struct {
11    Id int `gorm:"id"`
12    A int `gorm:"a"`
13    B int `gorm:"b"`
14}
15
16func (T) TableName() string {
17    return "t"
18}
19
20type MyLogger struct {
21}
22
23func (logger *MyLogger) Print(values ...interface{}) {
24    var (
25        level           = values[0]
26        source          = values[1]
27    )
28
29    if level == "sql" {
30        sql := values[3].(string)
31        log.Println(sql, level, source)
32    } else {
33        log.Println(values)
34    }
35}
36
37
38
39func main() {
40    db, _ := gorm.Open("mysql", "root:123456@(192.168.33.10:3306)/mysqldemo?charset=utf8&parseTime=True&loc=Local")
41    defer db.Close()
42
43    logger := &MyLogger{}
44
45    db.LogMode(true)
46
47    db.SetLogger(logger)
48
49