db_init.go 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637
  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_delay_permission"
  8. "hongze/hongze_mobile_admin/models/tables/company_permission_log"
  9. "hongze/hongze_mobile_admin/models/tables/company_report_permission"
  10. "hongze/hongze_mobile_admin/models/tables/h5_admin_session"
  11. "hongze/hongze_mobile_admin/models/tables/wx_token"
  12. "hongze/hongze_mobile_admin/utils"
  13. "rdluck_tools/orm"
  14. "time"
  15. )
  16. func InitDb() {
  17. _ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
  18. orm.SetMaxIdleConns("default", 50)
  19. orm.SetMaxOpenConns("default", 100)
  20. db, _ := orm.GetDB("default")
  21. db.SetConnMaxLifetime(10 * time.Minute)
  22. //注册对象
  23. orm.RegisterModel(
  24. new(admin.Admin),
  25. new(admin_record.AdminRecord),
  26. new(company_approval.CompanyApproval),
  27. new(company_delay_permission.CompanyDelayPermission),
  28. new(company_permission_log.CompanyPermissionLog),
  29. new(company_report_permission.CompanyReportPermission),
  30. new(h5_admin_session.H5AdminSession),
  31. new(wx_token.WxToken),
  32. )
  33. }