logger/gorm_test.go

46 lines
1.3 KiB
Go
Raw Permalink Normal View History

2022-03-17 10:33:47 +08:00
package logger
import (
2022-05-18 18:53:17 +08:00
"gitee.ltd/lxh/logger/log"
2022-03-17 10:33:47 +08:00
"gorm.io/driver/mysql"
"gorm.io/gorm"
2022-05-19 09:01:05 +08:00
gl "gorm.io/gorm/logger"
2022-03-17 10:33:47 +08:00
"testing"
2022-05-19 09:01:05 +08:00
"time"
2022-03-17 10:33:47 +08:00
)
func TestGormLogger(t *testing.T) {
dsn := "saas:saas123@tcp(10.11.0.10:3307)/saas_tenant?charset=utf8mb4&parseTime=True&loc=Local"
engine, err := gorm.Open(mysql.Open(dsn), &gorm.Config{Logger: DefaultGormLogger()})
if err != nil {
2022-05-18 18:53:17 +08:00
log.Panicf("mysql connect error: %s", err.Error())
2022-03-17 10:33:47 +08:00
}
2022-05-19 09:01:05 +08:00
var count int64
if err := engine.Table("t_tenant").Count(&count).Error; err != nil {
t.Log(err)
}
t.Logf("count: %d", count)
}
func TestGormLoggerWithConfig(t *testing.T) {
dsn := "saas:saas123@tcp(10.11.0.10:3307)/saas_tenant?charset=utf8mb4&parseTime=True&loc=Local"
engine, err := gorm.Open(mysql.Open(dsn), &gorm.Config{Logger: NewGormLoggerWithConfig(gl.Config{
SlowThreshold: time.Second, // Slow SQL threshold
IgnoreRecordNotFoundError: false, // 忽略没找到结果的错误
LogLevel: gl.Warn, // Log level
Colorful: false, // Disable color
})})
if err != nil {
log.Panicf("mysql connect error: %s", err.Error())
}
2022-03-17 10:33:47 +08:00
var count int64
if err := engine.Table("t_tenant1").Count(&count).Error; err != nil {
t.Log(err)
}
t.Logf("count: %d", count)
}