You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
2.7 KiB
108 lines
2.7 KiB
package datas
|
|
|
|
import (
|
|
"gin-vue-admin/model"
|
|
gormadapter "github.com/casbin/gorm-adapter/v3"
|
|
"github.com/gookit/color"
|
|
"gorm.io/gorm"
|
|
"os"
|
|
)
|
|
|
|
func InitMysqlData(db *gorm.DB) {
|
|
var err error
|
|
err = InitSysApi(db)
|
|
err = InitSysUser(db)
|
|
err = InitExaCustomer(db)
|
|
err = InitCasbinModel(db)
|
|
err = InitSysAuthority(db)
|
|
err = InitSysBaseMenus(db)
|
|
err = InitAuthorityMenu(db)
|
|
err = InitSysDictionary(db)
|
|
err = InitSysAuthorityMenus(db)
|
|
err = InitSysDataAuthorityId(db)
|
|
err = InitSysDictionaryDetail(db)
|
|
err = InitExaFileUploadAndDownload(db)
|
|
if err != nil {
|
|
color.Warn.Printf("[Mysql]-->初始化数据失败,err: %v\n", err)
|
|
os.Exit(0)
|
|
}
|
|
color.Info.Println("[Mysql]-->初始化数据成功")
|
|
}
|
|
|
|
func InitMysqlTables(db *gorm.DB) {
|
|
var err error
|
|
err = db.AutoMigrate(
|
|
model.SysApi{},
|
|
model.SysUser{},
|
|
model.ExaFile{},
|
|
model.ExaCustomer{},
|
|
model.SysBaseMenu{},
|
|
model.SysWorkflow{},
|
|
model.SysAuthority{},
|
|
model.JwtBlacklist{},
|
|
model.ExaFileChunk{},
|
|
model.SysDictionary{},
|
|
model.ExaSimpleUploader{},
|
|
model.SysOperationRecord{},
|
|
model.SysWorkflowStepInfo{},
|
|
model.SysDictionaryDetail{},
|
|
model.SysBaseMenuParameter{},
|
|
model.ExaFileUploadAndDownload{},
|
|
)
|
|
if err != nil {
|
|
color.Warn.Printf("[Mysql]-->初始化数据表失败,err: %v\n", err)
|
|
os.Exit(0)
|
|
}
|
|
color.Info.Println("[Mysql]-->初始化数据表成功")
|
|
}
|
|
|
|
func InitPostgresqlData(db *gorm.DB) {
|
|
var err error
|
|
err = InitSysApi(db)
|
|
err = InitSysUser(db)
|
|
err = InitExaCustomer(db)
|
|
err = InitCasbinModel(db)
|
|
err = InitSysAuthority(db)
|
|
err = InitSysBaseMenus(db)
|
|
err = InitAuthorityMenu(db)
|
|
err = InitSysAuthorityMenus(db)
|
|
err = InitSysDataAuthorityId(db)
|
|
err = InitSysDictionaryDetail(db)
|
|
err = InitExaFileUploadAndDownload(db)
|
|
err = InitSysDictionaryToPostgresql(db)
|
|
if err != nil {
|
|
color.Error.Printf("[Postgresql]-->初始化数据失败,err: %v\n", err)
|
|
os.Exit(0)
|
|
}
|
|
color.Info.Println("[Postgresql]-->初始化数据成功")
|
|
}
|
|
|
|
func InitPostgresqlTables(db *gorm.DB) {
|
|
var err error
|
|
if !db.Migrator().HasTable("casbin_rule") {
|
|
err = db.Migrator().CreateTable(&gormadapter.CasbinRule{})
|
|
}
|
|
err = db.AutoMigrate(
|
|
model.SysApi{},
|
|
model.SysUser{},
|
|
model.ExaFile{},
|
|
model.ExaCustomer{},
|
|
model.SysBaseMenu{},
|
|
model.SysWorkflow{},
|
|
model.SysAuthority{},
|
|
model.JwtBlacklist{},
|
|
model.ExaFileChunk{},
|
|
model.ExaSimpleUploader{},
|
|
model.SysOperationRecord{},
|
|
model.SysWorkflowStepInfo{},
|
|
model.SysDictionaryDetail{},
|
|
model.SysBaseMenuParameter{},
|
|
model.ExaFileUploadAndDownload{},
|
|
SysDictionaryToPostgresql{},
|
|
)
|
|
if err != nil {
|
|
color.Error.Printf("[Postgresql]-->初始化数据表失败,err: %v\n", err)
|
|
os.Exit(0)
|
|
}
|
|
color.Info.Println("[Postgresql]-->初始化数据表成功")
|
|
}
|