db_init.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. package models
  2. import (
  3. _ "github.com/go-sql-driver/mysql"
  4. "hongze/hongze_mobile_admin/models/tables/admin"
  5. "hongze/hongze_mobile_admin/models/tables/admin_record"
  6. "hongze/hongze_mobile_admin/models/tables/company_approval"
  7. "hongze/hongze_mobile_admin/models/tables/company_approval_message"
  8. "hongze/hongze_mobile_admin/models/tables/company_contract"
  9. "hongze/hongze_mobile_admin/models/tables/company_delay_permission"
  10. "hongze/hongze_mobile_admin/models/tables/company_operation_record"
  11. "hongze/hongze_mobile_admin/models/tables/company_permission_log"
  12. "hongze/hongze_mobile_admin/models/tables/company_product"
  13. "hongze/hongze_mobile_admin/models/tables/company_product_log"
  14. "hongze/hongze_mobile_admin/models/tables/company_report_permission"
  15. "hongze/hongze_mobile_admin/models/tables/h5_admin_session"
  16. "hongze/hongze_mobile_admin/models/tables/report"
  17. "hongze/hongze_mobile_admin/models/tables/wx_token"
  18. "hongze/hongze_mobile_admin/models/tables/wx_user"
  19. "hongze/hongze_mobile_admin/utils"
  20. "rdluck_tools/orm"
  21. "time"
  22. )
  23. func InitDb() {
  24. _ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
  25. orm.SetMaxIdleConns("default", 50)
  26. orm.SetMaxOpenConns("default", 100)
  27. db, _ := orm.GetDB("default")
  28. db.SetConnMaxLifetime(10 * time.Minute)
  29. //rddp数据库
  30. _ = orm.RegisterDataBase("rddp", "mysql", utils.MYSQL_URL_RDDP)
  31. orm.SetMaxIdleConns("rddp", 50)
  32. orm.SetMaxOpenConns("rddp", 100)
  33. report_db, _ := orm.GetDB("rddp")
  34. report_db.SetConnMaxLifetime(10 * time.Minute)
  35. //注册对象
  36. orm.RegisterModel(
  37. new(admin.Admin),
  38. new(admin_record.AdminRecord),
  39. new(company_approval.CompanyApproval),
  40. new(company_approval_message.CompanyApprovalMessage),
  41. new(company_contract.CompanyContract),
  42. new(company_delay_permission.CompanyDelayPermission),
  43. new(company_operation_record.CompanyOperationRecord),
  44. new(company_permission_log.CompanyPermissionLog),
  45. new(company_product.CompanyProduct),
  46. new(company_product_log.CompanyProductLog),
  47. new(company_report_permission.CompanyReportPermission),
  48. new(h5_admin_session.H5AdminSession),
  49. new(wx_token.WxToken),
  50. new(wx_user.WxUser),
  51. new(report.Report),
  52. )
  53. }