45 lines
725 B
Go
45 lines
725 B
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
// 类型迁移
|
|
func migration() {
|
|
|
|
err := DB.AutoMigrate()
|
|
//err = DBOrder.AutoMigrate(
|
|
// &model.Pay{},
|
|
//)
|
|
|
|
if err != nil {
|
|
fmt.Println("register table fail")
|
|
panic(1)
|
|
}
|
|
|
|
//自动迁移模式
|
|
AddTable(&LoginLog{})
|
|
AddTable(&RealName{})
|
|
AddTable(&User{})
|
|
AddTable(&RefreshToken{})
|
|
}
|
|
|
|
// 数据迁移
|
|
func AddColumn(dst interface{}, column string) {
|
|
if DB.Migrator().HasColumn(dst, column) == false {
|
|
if err := DB.Migrator().AddColumn(dst, column); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func AddTable(dst interface{}) {
|
|
if DB.Migrator().HasTable(dst) == false {
|
|
if err := DB.Migrator().CreateTable(dst); err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|