gorm.go 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. package initialize
  2. import (
  3. "os"
  4. "wails-app/internal/dao"
  5. "go.uber.org/zap"
  6. "gorm.io/gorm"
  7. "wails-app/internal/global"
  8. "wails-app/internal/model/example"
  9. )
  10. func Gorm() *gorm.DB {
  11. switch global.GVA_CONFIG.System.DbType {
  12. case "mysql":
  13. return GormMysql()
  14. case "pgsql":
  15. return GormPgSql()
  16. case "oracle":
  17. return GormOracle()
  18. case "mssql":
  19. return GormMssql()
  20. case "sqlite":
  21. return GormSqlite()
  22. default:
  23. return GormMysql()
  24. }
  25. }
  26. func RegisterTables() {
  27. db := global.GVA_DB
  28. err := db.AutoMigrate(
  29. dao.SysApi{},
  30. dao.SysUser{},
  31. dao.SysBaseMenu{},
  32. dao.JwtBlacklist{},
  33. dao.SysAuthority{},
  34. dao.SysDictionary{},
  35. dao.SysOperationRecord{},
  36. dao.SysDictionaryDetail{},
  37. dao.SysBaseMenuParameter{},
  38. dao.SysBaseMenuBtn{},
  39. dao.SysAuthorityBtn{},
  40. example.ExaFile{},
  41. example.ExaFileChunk{},
  42. dao.ExaFileUploadAndDownload{},
  43. dao.Vehicle{},
  44. dao.VehicleType{},
  45. dao.Owner{},
  46. dao.FeeConfig{},
  47. dao.Booth{},
  48. dao.Channel{},
  49. dao.ParkingLot{},
  50. dao.VehicleRecord{},
  51. dao.PaymentRecord{},
  52. dao.PaymentMethodConfig{},
  53. dao.PaymentEntryConfig{},
  54. dao.PaymentEntryMethod{},
  55. dao.UHFReader{},
  56. dao.Camera{},
  57. dao.Shortlist{},
  58. dao.MonthlyCard{},
  59. dao.DigitalTicket{},
  60. dao.ShiftRecord{},
  61. dao.Printer{},
  62. )
  63. if err != nil {
  64. global.GVA_LOG.Error("register table failed", zap.Error(err))
  65. os.Exit(0)
  66. }
  67. // 操作日志复合索引:created_at 位于 GVA_MODEL 内嵌字段(多表共用,不能加 tag),故显式创建
  68. // 覆盖高频查询:按用户 + 时间筛选操作日志
  69. if err := db.Exec("CREATE INDEX IF NOT EXISTS idx_op_user_time ON sys_operation_records (user_id, created_at)").Error; err != nil {
  70. global.GVA_LOG.Error("create idx_op_user_time failed", zap.Error(err))
  71. }
  72. global.GVA_LOG.Info("register table success")
  73. }