feat: set gorm log level (#2063)
This commit is contained in:
parent
a8e05c1b90
commit
1409e37ba3
|
|
@ -24,6 +24,7 @@ import (
|
||||||
"github.com/go-sql-driver/mysql"
|
"github.com/go-sql-driver/mysql"
|
||||||
drivermysql "gorm.io/driver/mysql"
|
drivermysql "gorm.io/driver/mysql"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
|
gormlogger "gorm.io/gorm/logger"
|
||||||
"gorm.io/gorm/schema"
|
"gorm.io/gorm/schema"
|
||||||
"moul.io/zapgorm2"
|
"moul.io/zapgorm2"
|
||||||
|
|
||||||
|
|
@ -51,13 +52,20 @@ func newMyqsl(cfg *config.Config) (*gorm.DB, error) {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Initialize gorm logger.
|
||||||
|
logLevel := gormlogger.Info
|
||||||
|
if !cfg.Verbose {
|
||||||
|
logLevel = gormlogger.Warn
|
||||||
|
}
|
||||||
|
gormLogger := zapgorm2.New(logger.CoreLogger.Desugar()).LogMode(logLevel)
|
||||||
|
|
||||||
// Connect to mysql.
|
// Connect to mysql.
|
||||||
db, err := gorm.Open(drivermysql.Open(dsn), &gorm.Config{
|
db, err := gorm.Open(drivermysql.Open(dsn), &gorm.Config{
|
||||||
NamingStrategy: schema.NamingStrategy{
|
NamingStrategy: schema.NamingStrategy{
|
||||||
SingularTable: true,
|
SingularTable: true,
|
||||||
},
|
},
|
||||||
DisableForeignKeyConstraintWhenMigrating: true,
|
DisableForeignKeyConstraintWhenMigrating: true,
|
||||||
Logger: zapgorm2.New(logger.CoreLogger.Desugar()),
|
Logger: gormLogger,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue