45 lines
799 B
Go
45 lines
799 B
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
//类型迁移
|
|
func migration() {
|
|
//自动迁移模式
|
|
addTable(&ApprovalTypeGroup{})
|
|
|
|
//增加字段
|
|
addColumn(&ApprovalType{}, "is_normal")
|
|
addColumn(&ApprovalType{}, "key_json")
|
|
addColumn(&ApprovalType{}, "group_id")
|
|
|
|
addColumn(&Exhibition{}, "apply_id")
|
|
addColumn(&ApprovalExhibition{}, "package_id")
|
|
addColumn(&ApprovalExhibition{}, "show_date")
|
|
|
|
addColumn(&Approval{}, "value_json")
|
|
}
|
|
|
|
//数据迁移
|
|
func addColumn(dst interface{}, column string) {
|
|
exist := DB.Migrator().HasColumn(dst, column)
|
|
if !exist {
|
|
err := DB.Migrator().AddColumn(dst, column)
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func addTable(dst interface{}) {
|
|
|
|
if DB.Migrator().HasTable(dst) == false {
|
|
_ = DB.Migrator().CreateTable(dst)
|
|
}
|
|
|
|
return
|
|
}
|