mirror of
https://github.com/flipped-aurora/gin-vue-admin.git
synced 2026-09-21 20:35:16 +00:00
PiexlMax(奇淼andAzir
0ce5a63c1d
* feat: 优化菜单管理的路径选择框 * fixed: 修复错误的文件命名 * feat: 增加参数管理功能，调整菜单选择组件。 * feat: 恢复config.yaml * feat: 更新AI功能 * feat: 增加API自动填充功能 * feat: 增加AI自动填充表格导出模板 * feat: 增加AI自动填充表格导出模板和AI自动模板填充 * feat: 新增方法支持选择是否鉴权 --------- Co-authored-by: Azir <2075125282@qq.com>
107 lines
2.6 KiB
Go
107 lines
2.6 KiB
Go
package initialize
|
|
|
|
import (
|
|
"context"
|
|
adapter "github.com/casbin/gorm-adapter/v3"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/example"
|
|
sysModel "github.com/flipped-aurora/gin-vue-admin/server/model/system"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/service/system"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const initOrderEnsureTables = system.InitOrderExternal - 1
|
|
|
|
type ensureTables struct{}
|
|
|
|
// auto run
|
|
func init() {
|
|
system.RegisterInit(initOrderEnsureTables, &ensureTables{})
|
|
}
|
|
|
|
func (ensureTables) InitializerName() string {
|
|
return "ensure_tables_created"
|
|
}
|
|
func (e *ensureTables) InitializeData(ctx context.Context) (next context.Context, err error) {
|
|
return ctx, nil
|
|
}
|
|
|
|
func (e *ensureTables) DataInserted(ctx context.Context) bool {
|
|
return true
|
|
}
|
|
|
|
func (e *ensureTables) MigrateTable(ctx context.Context) (context.Context, error) {
|
|
db, ok := ctx.Value("db").(*gorm.DB)
|
|
if !ok {
|
|
return ctx, system.ErrMissingDBContext
|
|
}
|
|
tables := []interface{}{
|
|
sysModel.SysApi{},
|
|
sysModel.SysUser{},
|
|
sysModel.SysBaseMenu{},
|
|
sysModel.SysAuthority{},
|
|
sysModel.JwtBlacklist{},
|
|
sysModel.SysDictionary{},
|
|
sysModel.SysAutoCodeHistory{},
|
|
sysModel.SysOperationRecord{},
|
|
sysModel.SysDictionaryDetail{},
|
|
sysModel.SysBaseMenuParameter{},
|
|
sysModel.SysBaseMenuBtn{},
|
|
sysModel.SysAuthorityBtn{},
|
|
sysModel.SysAutoCodePackage{},
|
|
sysModel.SysExportTemplate{},
|
|
sysModel.Condition{},
|
|
sysModel.JoinTemplate{},
|
|
sysModel.SysParams{},
|
|
|
|
adapter.CasbinRule{},
|
|
|
|
example.ExaFile{},
|
|
example.ExaCustomer{},
|
|
example.ExaFileChunk{},
|
|
example.ExaFileUploadAndDownload{},
|
|
}
|
|
for _, t := range tables {
|
|
_ = db.AutoMigrate(&t)
|
|
// 视图 authority_menu 会被当成表来创建，引发冲突错误（更新版本的gorm似乎不会）
|
|
// 由于 AutoMigrate() 基本无需考虑错误，因此显式忽略
|
|
}
|
|
return ctx, nil
|
|
}
|
|
|
|
func (e *ensureTables) TableCreated(ctx context.Context) bool {
|
|
db, ok := ctx.Value("db").(*gorm.DB)
|
|
if !ok {
|
|
return false
|
|
}
|
|
tables := []interface{}{
|
|
sysModel.SysApi{},
|
|
sysModel.SysUser{},
|
|
sysModel.SysBaseMenu{},
|
|
sysModel.SysAuthority{},
|
|
sysModel.JwtBlacklist{},
|
|
sysModel.SysDictionary{},
|
|
sysModel.SysAutoCodeHistory{},
|
|
sysModel.SysOperationRecord{},
|
|
sysModel.SysDictionaryDetail{},
|
|
sysModel.SysBaseMenuParameter{},
|
|
sysModel.SysBaseMenuBtn{},
|
|
sysModel.SysAuthorityBtn{},
|
|
sysModel.SysAutoCodePackage{},
|
|
sysModel.SysExportTemplate{},
|
|
sysModel.Condition{},
|
|
sysModel.JoinTemplate{},
|
|
|
|
adapter.CasbinRule{},
|
|
|
|
example.ExaFile{},
|
|
example.ExaCustomer{},
|
|
example.ExaFileChunk{},
|
|
example.ExaFileUploadAndDownload{},
|
|
}
|
|
yes := true
|
|
for _, t := range tables {
|
|
yes = yes && db.Migrator().HasTable(t)
|
|
}
|
|
return yes
|
|
}
|