diff --git a/db/db.go b/db/db.go index c816f82..021dde7 100644 --- a/db/db.go +++ b/db/db.go @@ -5,11 +5,17 @@ import ( "github.com/go-redis/redis/v8" "gorm.io/driver/mysql" "gorm.io/gorm" + "gorm.io/gorm/schema" ) func InitMysql(username, password, address, port, dbName string) (*gorm.DB, error) { dsn := fmt.Sprintf("%v:%v@tcp(%v:%v)/%v?charset=utf8mb4&parseTime=True&loc=Local", username, password, address, port, dbName) - db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{Logger: &dbLogger{}}) + db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{ + Logger: &dbLogger{}, + NamingStrategy: schema.NamingStrategy{ + SingularTable: true, // 使用单数表名 + }, + }) return db, err }