[GO] gin 框架gorm下使用logrus記錄sql語句

使用gin gorm時 , 默認sql語句是打印在終端的 ,想要記錄到文件中 , 能夠使用logrus來記錄到文件

先把logrus的一些配置作好

package tools

import (
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/sirupsen/logrus"
    "os"
    "path"
    "time"
)

func Logger() *logrus.Logger {
    now := time.Now()
    logFilePath := ""
    if dir, err := os.Getwd(); err == nil {
        logFilePath = dir + "/logs/"
    }
    if err := os.MkdirAll(logFilePath, 0777); err != nil {
        fmt.Println(err.Error())
    }
    logFileName := now.Format("2006-01-02") + ".log"
    //日誌文件
    fileName := path.Join(logFilePath, logFileName)
    if _, err := os.Stat(fileName); err != nil {
        if _, err := os.Create(fileName); err != nil {
            fmt.Println(err.Error())
        }
    }
    //寫入文件
    src, err := os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
    if err != nil {
        fmt.Println("err", err)
    }

    //實例化
    logger := logrus.New()

    //設置輸出
    logger.Out = src

    //設置日誌級別
    logger.SetLevel(logrus.DebugLevel)

    //設置日誌格式
    logger.SetFormatter(&logrus.TextFormatter{
        TimestampFormat: "2006-01-02 15:04:05",
    })
    return logger
}

在初始化gorm的地方設置logger , 把上面的logger設置進去就能夠了

    DB, err = gorm.Open("mysql", dsn)
    if err != nil {
        panic("數據庫鏈接失敗!")
    }
    DB.SingularTable(true)
    DB.LogMode(true)
    DB.SetLogger(tools.Logger())
    DB.DB().SetMaxIdleConns(10)
    DB.DB().SetMaxOpenConns(100)

文件中記錄的sql