diff --git a/database/new.go b/database/new.go index a949e0d..e78e685 100644 --- a/database/new.go +++ b/database/new.go @@ -15,7 +15,7 @@ import ( var ( // MigrateTables 存储需要在数据库初始化时自动迁移的表 - MigrateTables []any + MigrateTables []any = []any{} ) // NewDatabase 根据提供的驱动类型创建新的数据库连接 @@ -133,3 +133,10 @@ func NewPostgres(dsn []string, options *types.SqlOptions) (gormDb *gorm.DB, err return gormDb, nil } + +// AppendMigrate 调用此函数后,会在数据库初始化时自动迁移表结构 +// +// - table: 需要自动迁移的表 +func AppendMigrate(table any) { + MigrateTables = append(MigrateTables, table) +}