fix bug
This commit is contained in:
parent
3d6871138a
commit
404957f16a
|
@ -15,7 +15,7 @@ import (
|
||||||
|
|
||||||
var (
|
var (
|
||||||
// MigrateTables 存储需要在数据库初始化时自动迁移的表
|
// MigrateTables 存储需要在数据库初始化时自动迁移的表
|
||||||
MigrateTables []any = make([]any, 0)
|
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)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue