init.go 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. package datas
  2. import (
  3. "gin-vue-admin/model"
  4. gormadapter "github.com/casbin/gorm-adapter/v3"
  5. "github.com/gookit/color"
  6. "gorm.io/gorm"
  7. "os"
  8. )
  9. func InitMysqlData(db *gorm.DB) {
  10. InitSysApi(db)
  11. InitSysUser(db)
  12. InitExaCustomer(db)
  13. InitCasbinModel(db)
  14. InitSysAuthority(db)
  15. InitSysBaseMenus(db)
  16. InitAuthorityMenu(db)
  17. InitSysDictionary(db)
  18. InitSysAuthorityMenus(db)
  19. InitSysDataAuthorityId(db)
  20. InitSysDictionaryDetail(db)
  21. InitExaFileUploadAndDownload(db)
  22. InitWkProcess(db)
  23. }
  24. func InitMysqlTables(db *gorm.DB) {
  25. var err error
  26. if !db.Migrator().HasTable("casbin_rule") {
  27. err = db.Migrator().CreateTable(&gormadapter.CasbinRule{})
  28. }
  29. err = db.AutoMigrate(
  30. model.SysApi{},
  31. model.SysUser{},
  32. model.ExaFile{},
  33. model.ExaCustomer{},
  34. model.SysBaseMenu{},
  35. model.SysAuthority{},
  36. model.JwtBlacklist{},
  37. model.ExaFileChunk{},
  38. model.SysDictionary{},
  39. model.ExaSimpleUploader{},
  40. model.SysOperationRecord{},
  41. model.SysDictionaryDetail{},
  42. model.SysBaseMenuParameter{},
  43. model.ExaFileUploadAndDownload{},
  44. )
  45. if err != nil {
  46. color.Warn.Printf("[Mysql]-->初始化数据表失败,err: %v\n", err)
  47. os.Exit(0)
  48. }
  49. color.Info.Println("[Mysql]-->初始化数据表成功")
  50. }