1. 程式人生 > 實用技巧 >[GO] gin 框架gorm下使用logrus記錄sql語句

[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