micro-account/pkg/model/migration.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
}