init.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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. var err error
  11. err = InitSysApi(db)
  12. err = InitSysUser(db)
  13. err = InitExaCustomer(db)
  14. err = InitCasbinModel(db)
  15. err = InitSysAuthority(db)
  16. err = InitSysBaseMenus(db)
  17. err = InitAuthorityMenu(db)
  18. err = InitSysDictionary(db)
  19. err = InitSysAuthorityMenus(db)
  20. err = InitSysDataAuthorityId(db)
  21. err = InitSysDictionaryDetail(db)
  22. err = InitExaFileUploadAndDownload(db)
  23. err = InitWkProcess(db)
  24. if err != nil {
  25. color.Warn.Printf("[Mysql]-->初始化数据失败,err: %v\n", err)
  26. os.Exit(0)
  27. }
  28. color.Info.Println("[Mysql]-->初始化数据成功")
  29. }
  30. func InitMysqlTables(db *gorm.DB) {
  31. var err error
  32. if !db.Migrator().HasTable("casbin_rule") {
  33. err = db.Migrator().CreateTable(&gormadapter.CasbinRule{})
  34. }
  35. err = db.AutoMigrate(
  36. model.SysApi{},
  37. model.SysUser{},
  38. model.ExaFile{},
  39. model.ExaCustomer{},
  40. model.SysBaseMenu{},
  41. model.SysAuthority{},
  42. model.JwtBlacklist{},
  43. model.ExaFileChunk{},
  44. model.SysDictionary{},
  45. model.ExaSimpleUploader{},
  46. model.SysOperationRecord{},
  47. model.SysDictionaryDetail{},
  48. model.SysBaseMenuParameter{},
  49. model.ExaFileUploadAndDownload{},
  50. )
  51. if err != nil {
  52. color.Warn.Printf("[Mysql]-->初始化数据表失败,err: %v\n", err)
  53. os.Exit(0)
  54. }
  55. color.Info.Println("[Mysql]-->初始化数据表成功")
  56. }