Compare commits

..

2 Commits

Author SHA1 Message Date
404957f16a fix bug 2025-10-05 15:00:07 +08:00
3d6871138a fix bug 2025-10-05 14:59:01 +08:00

View File

@@ -15,7 +15,7 @@ import (
var ( var (
// MigrateTables 存储需要在数据库初始化时自动迁移的表 // MigrateTables 存储需要在数据库初始化时自动迁移的表
MigrateTables []any = []any{} MigrateTables []any
) )
// NewDatabase 根据提供的驱动类型创建新的数据库连接 // NewDatabase 根据提供的驱动类型创建新的数据库连接
@@ -25,6 +25,7 @@ var (
// options: 数据库连接选项 // options: 数据库连接选项
// 返回: GORM数据库实例 // 返回: GORM数据库实例
func NewDatabase(driver string, dsn []string, options *types.SqlOptions) (db *gorm.DB, err error) { func NewDatabase(driver string, dsn []string, options *types.SqlOptions) (db *gorm.DB, err error) {
MigrateTables = make([]any, 0)
driver = strings.ToLower(driver) driver = strings.ToLower(driver)
switch driver { switch driver {
case "mysql": case "mysql":
@@ -138,5 +139,8 @@ func NewPostgres(dsn []string, options *types.SqlOptions) (gormDb *gorm.DB, err
// //
// - table: 需要自动迁移的表 // - table: 需要自动迁移的表
func AppendMigrate(table any) { func AppendMigrate(table any) {
if MigrateTables == nil {
MigrateTables = make([]any, 0)
}
MigrateTables = append(MigrateTables, table) MigrateTables = append(MigrateTables, table)
} }