db.go 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. package models
  2. import (
  3. _ "github.com/go-sql-driver/mysql"
  4. "hongze/hongze_ETA_mobile_api/models/system"
  5. "hongze/hongze_ETA_mobile_api/utils"
  6. "time"
  7. "github.com/beego/beego/v2/client/orm"
  8. )
  9. func init() {
  10. _ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
  11. orm.SetMaxIdleConns("default", 50)
  12. orm.SetMaxOpenConns("default", 100)
  13. db, _ := orm.GetDB("default")
  14. db.SetConnMaxLifetime(10 * time.Minute)
  15. _ = orm.RegisterDataBase("rddp", "mysql", utils.MYSQL_URL_RDDP)
  16. orm.SetMaxIdleConns("rddp", 50)
  17. orm.SetMaxOpenConns("rddp", 100)
  18. report_db, _ := orm.GetDB("rddp")
  19. report_db.SetConnMaxLifetime(10 * time.Minute)
  20. _ = orm.RegisterDataBase("edb", "mysql", utils.MYSQL_URL_EDB)
  21. orm.SetMaxIdleConns("edb", 50)
  22. orm.SetMaxOpenConns("edb", 100)
  23. edb_db, _ := orm.GetDB("edb")
  24. edb_db.SetConnMaxLifetime(10 * time.Minute)
  25. _ = orm.RegisterDataBase("data", "mysql", utils.MYSQL_URL_DATA)
  26. orm.SetMaxIdleConns("data", 50)
  27. orm.SetMaxOpenConns("data", 100)
  28. data_db, _ := orm.GetDB("data")
  29. data_db.SetConnMaxLifetime(10 * time.Minute)
  30. _ = orm.RegisterDataBase("gl", "mysql", utils.MYSQL_URL_GL)
  31. orm.SetMaxIdleConns("gl", 50)
  32. orm.SetMaxOpenConns("gl", 100)
  33. gl, _ := orm.GetDB("gl")
  34. gl.SetConnMaxLifetime(10 * time.Minute)
  35. _ = orm.RegisterDataBase("comein_data", "mysql", utils.MYSQL_URL_COMEIN_DATA)
  36. orm.SetMaxIdleConns("comein_data", 50)
  37. orm.SetMaxOpenConns("comein_data", 100)
  38. comein_datadb, _ := orm.GetDB("comein_data")
  39. comein_datadb.SetConnMaxLifetime(10 * time.Minute)
  40. orm.Debug = true
  41. orm.DebugLog = orm.NewLog(utils.Binlog)
  42. //注册对象
  43. orm.RegisterModel(
  44. new(Classify),
  45. new(Ppt),
  46. new(PptPages),
  47. new(PptPublishRecord),
  48. new(PptV2),
  49. new(PptV2PublishRecord),
  50. new(PptV2SaveLog),
  51. new(PptV2Group),
  52. new(PptV2GroupMapping),
  53. new(PptV2Grant), //ppt授权操作表
  54. )
  55. // 报告相关 数据表
  56. initReport()
  57. initSystem()
  58. }
  59. // initSystem 系统表 数据表
  60. func initSystem() {
  61. orm.RegisterModel(
  62. new(system.SysUserLoginRecord),
  63. new(system.SysSession),
  64. new(system.Admin),
  65. )
  66. }
  67. // initReport 报告相关 数据表
  68. func initReport() {
  69. orm.RegisterModel(
  70. new(ClassifyMenu), // 报告分类-子目录表
  71. )
  72. }