feat: set gorm log level (#2063)

This commit is contained in:
Gaius 2023-02-10 14:56:10 +08:00
parent a8e05c1b90
commit 1409e37ba3
No known key found for this signature in database
GPG Key ID: 8B4E5D1290FA2FFB
1 changed files with 9 additions and 1 deletions

View File

@ -24,6 +24,7 @@ import (
"github.com/go-sql-driver/mysql"
drivermysql "gorm.io/driver/mysql"
"gorm.io/gorm"
gormlogger "gorm.io/gorm/logger"
"gorm.io/gorm/schema"
"moul.io/zapgorm2"
@ -51,13 +52,20 @@ func newMyqsl(cfg *config.Config) (*gorm.DB, error) {
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.
db, err := gorm.Open(drivermysql.Open(dsn), &gorm.Config{
NamingStrategy: schema.NamingStrategy{
SingularTable: true,
},
DisableForeignKeyConstraintWhenMigrating: true,
Logger: zapgorm2.New(logger.CoreLogger.Desugar()),
Logger: gormLogger,
})
if err != nil {
return nil, err