修改数据表迁移用法

This commit is contained in:
linwenxiang
2020-09-12 00:15:23 +08:00
parent 317d417824
commit 3a6a06f52a
3 changed files with 33 additions and 34 deletions
+2 -2
View File
@@ -41,7 +41,7 @@ func (e *Migration) Migrate() {
var err error
var count int64
for _, v := range versions {
err = e.db.Table("sys_migration").Where("version = ?", v).Count(&count).Error
err = e.db.Debug().Table("sys_migration").Where("version = ?", v).Count(&count).Error
if err != nil {
log.Fatalln(err)
}
@@ -50,7 +50,7 @@ func (e *Migration) Migrate() {
count = 0
continue
}
err = (e.version[v])(e.db, v)
err = (e.version[v])(e.db.Debug(), v)
if err != nil {
log.Fatalln(err)
}
@@ -18,35 +18,33 @@ func init() {
}
func _1599190683659Tables(db *gorm.DB, version string) error {
return db.Transaction(func(tx *gorm.DB) error {
//
err := tx.Migrator().AutoMigrate(
new(models.CasbinRule),
new(models.SysDept),
new(models.SysConfig),
new(tools.SysTables),
new(tools.SysColumns),
new(models.Menu),
new(models.LoginLog),
new(models.SysOperLog),
new(models.RoleMenu),
new(models.SysRoleDept),
new(models.SysUser),
new(models.SysRole),
new(models.Post),
new(models.DictData),
new(models.DictType),
new(models.SysJob),
new(models.SysConfig),
new(models.SysSetting),
new(models.SysFileDir),
new(models.SysCategory),
new(models.SysContent))
if err != nil {
return err
}
return tx.Create(&common.Migration{
Version: version,
}).Error
})
err := db.Migrator().AutoMigrate(
new(models.CasbinRule),
new(models.SysDept),
new(models.SysConfig),
new(tools.SysTables),
new(tools.SysColumns),
new(models.Menu),
new(models.LoginLog),
new(models.SysOperLog),
new(models.RoleMenu),
new(models.SysRoleDept),
new(models.SysUser),
new(models.SysRole),
new(models.Post),
new(models.DictData),
new(models.DictType),
new(models.SysJob),
new(models.SysConfig),
new(models.SysSetting),
new(models.SysFileDir),
new(models.SysCategory),
new(models.SysContent),
)
if err != nil {
return err
}
return db.Create(&common.Migration{
Version: version,
}).Error
}