db.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675
  1. package models
  2. import (
  3. "eta/eta_api/models/ai_summary"
  4. "eta/eta_api/models/aimod"
  5. binlogDao "eta/eta_api/models/binlog"
  6. "eta/eta_api/models/company"
  7. "eta/eta_api/models/data_manage"
  8. "eta/eta_api/models/data_manage/chart_theme"
  9. "eta/eta_api/models/data_manage/cross_variety"
  10. "eta/eta_api/models/data_manage/data_manage_permission"
  11. "eta/eta_api/models/data_manage/edb_refresh"
  12. "eta/eta_api/models/data_manage/excel"
  13. future_good2 "eta/eta_api/models/data_manage/future_good"
  14. "eta/eta_api/models/data_manage/supply_analysis"
  15. "eta/eta_api/models/data_stat"
  16. edbmonitor "eta/eta_api/models/edb_monitor"
  17. "eta/eta_api/models/eta_trial"
  18. "eta/eta_api/models/fe_calendar"
  19. "eta/eta_api/models/ppt_english"
  20. "eta/eta_api/models/report"
  21. "eta/eta_api/models/report_approve"
  22. "eta/eta_api/models/sandbox"
  23. "eta/eta_api/models/semantic_analysis"
  24. "eta/eta_api/models/smart_report"
  25. "eta/eta_api/models/speech_recognition"
  26. "eta/eta_api/models/system"
  27. "eta/eta_api/models/yb"
  28. binlogSvr "eta/eta_api/services/binlog"
  29. "eta/eta_api/utils"
  30. "time"
  31. _ "github.com/go-sql-driver/mysql"
  32. "github.com/beego/beego/v2/client/orm"
  33. )
  34. func init() {
  35. _ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
  36. orm.SetMaxIdleConns("default", 50)
  37. orm.SetMaxOpenConns("default", 100)
  38. db, _ := orm.GetDB("default")
  39. db.SetConnMaxLifetime(10 * time.Minute)
  40. _ = orm.RegisterDataBase("rddp", "mysql", utils.MYSQL_URL_RDDP)
  41. orm.SetMaxIdleConns("rddp", 50)
  42. orm.SetMaxOpenConns("rddp", 100)
  43. report_db, _ := orm.GetDB("rddp")
  44. report_db.SetConnMaxLifetime(10 * time.Minute)
  45. _ = orm.RegisterDataBase("edb", "mysql", utils.MYSQL_URL_EDB)
  46. orm.SetMaxIdleConns("edb", 50)
  47. orm.SetMaxOpenConns("edb", 100)
  48. edb_db, _ := orm.GetDB("edb")
  49. edb_db.SetConnMaxLifetime(10 * time.Minute)
  50. _ = orm.RegisterDataBase("data", "mysql", utils.MYSQL_URL_DATA)
  51. orm.SetMaxIdleConns("data", 50)
  52. orm.SetMaxOpenConns("data", 100)
  53. data_db, _ := orm.GetDB("data")
  54. data_db.SetConnMaxLifetime(10 * time.Minute)
  55. _ = orm.RegisterDataBase("gl", "mysql", utils.MYSQL_URL_GL)
  56. orm.SetMaxIdleConns("gl", 50)
  57. orm.SetMaxOpenConns("gl", 100)
  58. gl, _ := orm.GetDB("gl")
  59. gl.SetConnMaxLifetime(10 * time.Minute)
  60. // 用户主库
  61. if utils.MYSQL_WEEKLY_URL != `` && (utils.BusinessCode == utils.BusinessCodeRelease || utils.BusinessCode == utils.BusinessCodeSandbox || utils.BusinessCode == utils.BusinessCodeDebug) {
  62. _ = orm.RegisterDataBase("weekly", "mysql", utils.MYSQL_WEEKLY_URL)
  63. orm.SetMaxIdleConns("weekly", 50)
  64. orm.SetMaxOpenConns("weekly", 100)
  65. weeklyDb, _ := orm.GetDB("weekly")
  66. weeklyDb.SetConnMaxLifetime(10 * time.Minute)
  67. }
  68. if utils.MYSQL_AI_URL != "" {
  69. _ = orm.RegisterDataBase("ai", "mysql", utils.MYSQL_AI_URL)
  70. orm.SetMaxIdleConns("ai", 50)
  71. orm.SetMaxOpenConns("ai", 100)
  72. weeklyDb, _ := orm.GetDB("ai")
  73. weeklyDb.SetConnMaxLifetime(10 * time.Minute)
  74. }
  75. orm.Debug = true
  76. orm.DebugLog = orm.NewLog(utils.Binlog)
  77. //注册对象
  78. orm.RegisterModel(
  79. new(Classify),
  80. new(Resource),
  81. new(PptV2),
  82. new(PptV2PublishRecord),
  83. new(PptV2SaveLog),
  84. new(PptV2Group),
  85. new(PptV2GroupMapping),
  86. new(PptV2Grant), //ppt授权操作表
  87. new(VarietyClassify), // 标签分类表
  88. new(VarietyTag), // 标签表
  89. new(ResearchVarietyTagRelation), // 研究员标签关系表
  90. new(ReportPptImg), //Ppt转报告的图片记录表
  91. )
  92. // 系统表 数据表
  93. initSystem()
  94. // 客户相关
  95. initCompany()
  96. // 联系人相关
  97. initWxUser()
  98. // 报告相关 数据表
  99. initReport()
  100. // 研报相关
  101. initYb()
  102. // 注册Edb沙盘 数据表
  103. initSandbox()
  104. // 上海指标服务 数据表
  105. initShEdbData()
  106. // 指标服务 数据表
  107. initEdbData()
  108. // 图表 数据表
  109. initChart()
  110. // 英文报告
  111. initEnglishReport()
  112. // 英文ppt
  113. initPptEnglish()
  114. // 期货数据库
  115. initFutureGood()
  116. // 共享网盘
  117. initCloudDisk()
  118. // 语义分析
  119. initSemanticAnalysis()
  120. // 供应分析
  121. initSupplyAnalysis()
  122. //多图配置
  123. initMultipleGraphConfig()
  124. // 商家配置
  125. initBusinessConf()
  126. // 图库框架
  127. initChartFramework()
  128. // 外部链接
  129. initOutLink()
  130. // 智能研报
  131. initSmartReport()
  132. // ETA试用相关表
  133. if utils.BusinessCode == utils.BusinessCodeSandbox {
  134. initEtaTrial()
  135. }
  136. // initDataStat 数据源统计管理相关表
  137. initDataStat()
  138. // 初始化EXCEL的表
  139. initExcel()
  140. // 初始化跨品种分析表
  141. initCrossVariety()
  142. //初始化AI
  143. initAi()
  144. // 报告审批
  145. initReportApprove()
  146. // 初始化图表主题
  147. initChartTheme()
  148. // 初始化指标刷新
  149. initEdbRefresh()
  150. // 语音识别
  151. initSpeechRecognition()
  152. // 初始化数据资产权限的一些表
  153. initDataMangePerMission()
  154. // 初始化外汇日历
  155. initFeCalendar()
  156. // 初始化因子指标系列
  157. initFactorEdbSeries()
  158. // 初始化指标监控
  159. initEdbMonitor()
  160. // 开启mysql binlog监听
  161. if utils.MYSQL_DATA_BINLOG_URL != "" {
  162. initBinlog()
  163. go binlogSvr.ListenMysql()
  164. }
  165. // 初始化部分数据表变量(直接init会有顺序问题=_=!)
  166. afterInitTable()
  167. }
  168. // initSystem 系统表 数据表
  169. func initSystem() {
  170. orm.RegisterModel(
  171. new(system.SysUserLoginRecord),
  172. new(system.SysSession),
  173. new(system.SysDepartment),
  174. new(system.SysGroup),
  175. new(system.SysRoleMenu),
  176. new(system.Admin),
  177. new(system.SysRole),
  178. new(system.SysMenuButton),
  179. new(system.SysRoleButton),
  180. new(system.ResearchGroup), // 研究员研究方向分组表
  181. new(system.ResearchGroupRelation), // 研究员研究方向分组关系表
  182. new(system.SysRoleAdmin), //管理员账号和角色映射表
  183. new(system.AdminConfig), //系统用户配置表
  184. new(system.AdminOperateRecord),
  185. new(system.AdminVerifyCodeRecord), // 用户短信邮箱验证码记录表
  186. )
  187. }
  188. // initCompany 客户相关 数据表
  189. func initCompany() {
  190. orm.RegisterModel(
  191. new(company.Company),
  192. )
  193. }
  194. // initWxUser 联系人相关 数据表
  195. func initWxUser() {
  196. orm.RegisterModel(
  197. new(WxUser),
  198. new(UserSellerRelation),
  199. )
  200. }
  201. // initReport 报告相关 数据表
  202. func initReport() {
  203. orm.RegisterModel(
  204. new(Report),
  205. new(ReportViewRecord),
  206. new(ChartPermissionSearchKeyWordMapping), //报告分类权限表
  207. new(ReportChapter), // 报告章节表
  208. new(ReportChapterTicker), // 晨报章节ticker
  209. new(ReportChapterTypePermission), // 晨周报章节类型权限表
  210. new(ChartPermission), // 权限表
  211. new(YbPcSuncode),
  212. new(YbSuncodePars),
  213. new(ReportAuthor), //报告作者
  214. new(ClassifyMenu), // 报告分类-子目录表
  215. new(ClassifyMenuRelation), // 报告分类-子目录关联表
  216. new(ChartPermissionChapterMapping), // 权限mapping表
  217. new(ReportChapterType), // 报告章节类型表
  218. new(ReportStateRecord), // 研报状态修改记录表
  219. new(report.ReportGrant), // 报告授权用户表
  220. new(report.ReportChapterGrant), // 报告章节授权用户表
  221. new(report.ReportChapterPermissionMapping), // 报告章节的权限关系表
  222. )
  223. }
  224. // initYb 研报 数据表
  225. func initYb() {
  226. orm.RegisterModel(
  227. new(yb.PriceDriven), // 价格驱动表
  228. new(yb.PriceDrivenSaveLog), // 价格驱动保存记录表
  229. new(yb.PriceDrivenTag), // 价格驱动标签表
  230. new(yb.CommunityQuestionProcess), // 研报问答社区流程表
  231. )
  232. }
  233. // initSandbox 注册Edb沙盘 数据表
  234. func initSandbox() {
  235. //注册对象
  236. orm.RegisterModel(
  237. new(sandbox.Sandbox), //沙盘主表
  238. new(sandbox.SandboxVersion), //沙盘版本表
  239. new(sandbox.SandboxDraft), //沙盘草稿表
  240. new(sandbox.SandboxClassify), //沙盘分类表
  241. )
  242. }
  243. // initShEdbData 上海服务器指标 数据表
  244. func initShEdbData() {
  245. orm.RegisterModel(
  246. new(Edbdata),
  247. new(EdbdataDeleteRecord),
  248. new(EdbdataImportFail),
  249. new(EdbImportFail), //指标导入失败数据表
  250. new(Edbinfo), //edb库的edbinfo表
  251. new(data_manage.ExcelStyle), //在线excel样式表
  252. new(data_manage.ExcelEdbdataMapping), //excel样式和指标映射表
  253. new(EdbinfoOpRecord), // 手工数据的操作日志
  254. )
  255. }
  256. // initEdbData 指标服务 数据表
  257. func initEdbData() {
  258. orm.RegisterModel(
  259. new(data_manage.EdbClassify),
  260. new(data_manage.EdbInfo), //hz_data库的edb_info表
  261. new(data_manage.EdbDataThs),
  262. new(data_manage.EdbDataWind),
  263. new(data_manage.EdbDataPb),
  264. new(data_manage.ManualUserClassify),
  265. new(data_manage.EdbdataClassify),
  266. new(data_manage.ManualUser),
  267. new(data_manage.EdbInfoCalculate),
  268. new(data_manage.SearchKeyword),
  269. new(data_manage.BaseFromSmm),
  270. new(data_manage.BaseFromSmmIndex),
  271. new(data_manage.BaseFromSmmData),
  272. new(data_manage.BaseFromSmmClassify),
  273. new(data_manage.EdbInfoLog),
  274. new(data_manage.EdbInfoRecord),
  275. new(data_manage.EdbInfoCalculateMapping),
  276. new(data_manage.PredictEdbConf), //预测指标配置
  277. new(data_manage.BaseFromMysteelChemicalClassify), //预测指标配置
  278. new(data_manage.BaseFromMysteelChemicalIndex), //钢联化工
  279. new(data_manage.BaseFromEiaSteoClassify), // Eia steo 报告指标
  280. new(data_manage.BaseFromEiaSteoIndex), // Eia steo 报告指标分类
  281. new(data_manage.PredictEdbRuleData), //预测指标,动态规则的计算数据
  282. new(data_manage.ComTradeCountry), //联合国商品贸易数据库的国家编码表
  283. new(data_manage.ComTradeData), //联合国商品贸易数据表
  284. new(data_manage.BaseFromSci),
  285. new(data_manage.BaseFromSciIndex),
  286. new(data_manage.BaseFromSciData),
  287. new(data_manage.BaseFromSciClassify),
  288. new(data_manage.TradePositionDalianTop),
  289. new(data_manage.TradePositionZhengzhouTop),
  290. new(data_manage.TradePositionCffexTop),
  291. new(data_manage.TradePositionShanghaiTop),
  292. new(data_manage.TradePositionIneTop),
  293. new(data_manage.BaseFromBaiinfo),
  294. new(data_manage.BaseFromBaiinfoIndex),
  295. new(data_manage.BaseFromBaiinfoData),
  296. new(data_manage.BaseFromBaiinfoClassify),
  297. new(data_manage.EdbDataInsertConfig), // 指标数据插入配置表
  298. new(data_manage.EdbInfoNoPermissionAdmin), //指标不可见用户配置表
  299. new(data_manage.EdbTerminal), //指标终端
  300. new(data_manage.BaseFromThsHfIndex),
  301. new(data_manage.BaseFromThsHfData),
  302. new(data_manage.BaseFromThsHfClassify),
  303. new(data_manage.BaseFromEdbMapping),
  304. new(data_manage.EdbInfoRelation), //指标关系表
  305. new(data_manage.BaseFromSciHqClassify),
  306. new(data_manage.BaseFromSciHqIndex),
  307. new(data_manage.BaseFromSciHqData),
  308. )
  309. }
  310. // initChart 图表 数据表
  311. func initChart() {
  312. orm.RegisterModel(
  313. new(data_manage.ChartClassify),
  314. new(data_manage.ChartInfo),
  315. new(data_manage.ChartEdbMapping),
  316. new(data_manage.MyChartClassify),
  317. new(data_manage.MyChart),
  318. new(data_manage.MyChartLog),
  319. new(data_manage.MyChartClassifyMapping),
  320. new(data_manage.ChartInfoLog),
  321. new(data_manage.ChartInfoCorrelation),
  322. new(data_manage.ChartSeries),
  323. new(data_manage.ChartSeriesEdbMapping),
  324. )
  325. }
  326. // initEnglishReport 英文报告
  327. func initEnglishReport() {
  328. orm.RegisterModel(
  329. new(EnglishReport),
  330. new(EnglishPolicyReport),
  331. new(EnglishReportEmail),
  332. new(EnglishReportEmailPV),
  333. new(EnglishReportEmailLog),
  334. new(EnglishClassify),
  335. new(EnglishCompany), // 英文客户
  336. new(EnglishCompanyLog), // 英文客户操作日志
  337. new(EnglishReportEmailOpLog), // 英文联系人操作日志
  338. new(EnglishCompanyTodo), // 英文客户TODO任务
  339. new(EnglishCompanyTodoPublic), // 英文客户公共TODO任务
  340. new(EnglishVideoCover), // 英文研报线上路演视频封面
  341. new(EnglishVideo), // 英文研报线上路演
  342. new(EnPermission), // 英文品种权限表
  343. new(EnCompanyPermission), // 英文客户品种权限关联表
  344. new(EnClassifyPermission), // 英文分类品种权限关联表
  345. )
  346. }
  347. // initPptEnglish ppt英文
  348. func initPptEnglish() {
  349. orm.RegisterModel(
  350. new(ppt_english.PptEnglish),
  351. new(ppt_english.PptEnglishPublishRecord),
  352. new(ppt_english.PptEnglishSaveLog),
  353. new(ppt_english.PptEnglishGroup),
  354. new(ppt_english.PptEnglishGroupMapping),
  355. new(ppt_english.PptEnglishGrant), //ppt授权操作表
  356. new(ppt_english.ReportPptEnglishImg), //Ppt转报告的图片记录表
  357. )
  358. }
  359. // initFutureGood 注册期货数据 数据表
  360. func initFutureGood() {
  361. //注册对象
  362. orm.RegisterModel(
  363. new(future_good2.FutureGoodEdbInfo), //期货指标表
  364. new(future_good2.FutureGoodEdbData), //期货指标数据表
  365. //new(future_good2.FutureGoodChartClassify), //期货指标分类表
  366. new(future_good2.ChartInfoFutureGoodProfit), //期货利润图的扩展表
  367. )
  368. }
  369. // initCloudDisk 共享云盘
  370. func initCloudDisk() {
  371. //注册对象
  372. orm.RegisterModel(
  373. new(CloudDiskMenu), // 云盘目录表
  374. new(CloudDiskResource), // 云盘资源表
  375. )
  376. }
  377. // initSemanticAnalysis 语义分析
  378. func initSemanticAnalysis() {
  379. orm.RegisterModel(
  380. new(semantic_analysis.SaLabel), // 语义分析-标签表
  381. new(semantic_analysis.SaDocClassify), // 语义分析-文档分类表
  382. new(semantic_analysis.SaDoc), // 语义分析-文档表
  383. new(semantic_analysis.SaDocSection), // 语义分析-文档段落表
  384. new(semantic_analysis.SaCompareClassify), // 语义分析-比对分类表
  385. new(semantic_analysis.SaCompare), // 语义分析-比对表
  386. new(semantic_analysis.SaCompareDoc), // 语义分析-比对关联文档表
  387. new(semantic_analysis.SaCompareLabel), // 语义分析-比对段落表
  388. new(semantic_analysis.SaCompareSearchKeyword), // 语义分析-比对搜索关键词
  389. )
  390. }
  391. // initSupplyAnalysis 供应分析
  392. func initSupplyAnalysis() {
  393. orm.RegisterModel(
  394. new(supply_analysis.Variety),
  395. new(supply_analysis.VarietyAdminPermission),
  396. new(supply_analysis.VarietyEdbInfo),
  397. new(supply_analysis.VarietyPlant),
  398. )
  399. }
  400. // initMultipleGraphConfig 多图配置 数据表
  401. func initMultipleGraphConfig() {
  402. //注册对象
  403. orm.RegisterModel(
  404. new(data_manage.MultipleGraphConfig), //多图配置表
  405. new(data_manage.MultipleGraphConfigChartMapping), //图表与多图配置的关系表
  406. new(data_manage.MultipleGraphConfigEdbMapping), //指标与多图配置的关系表
  407. )
  408. }
  409. // initBusinessConf 商家配置
  410. func initBusinessConf() {
  411. orm.RegisterModel(
  412. new(BusinessConf), // 商家配置表
  413. new(BusinessConfOperationRecord), // 商家配置操作记录表
  414. )
  415. }
  416. func initOutLink() {
  417. orm.RegisterModel(
  418. new(OutLink)) // 外部链接表
  419. }
  420. // initEtaTrial 试用平台相关表
  421. func initEtaTrial() {
  422. orm.RegisterModel(
  423. new(eta_trial.QuestionnaireFillRecord), // 问卷填写记录表
  424. )
  425. }
  426. // initDataStat 数据源统计管理相关表
  427. func initDataStat() {
  428. orm.RegisterModel(
  429. new(data_stat.EdbInfoDeleteLog), // 指标删除日志表
  430. new(data_stat.EdbInfoUpdateStat), // 指标更新/刷新 汇总表(数据源明细表)
  431. new(data_stat.EdbInfoUpdateLog), // 指标更新/刷新日志列表
  432. new(data_stat.EdbSourceStat), // 数据源统计表
  433. )
  434. }
  435. // initChartFramework 图库框架相关表
  436. func initChartFramework() {
  437. orm.RegisterModel(
  438. new(data_manage.ChartFramework), // 图库框架主表
  439. new(data_manage.ChartFrameworkNode), // 图库框架节点表
  440. )
  441. }
  442. // initExcel 初始化EXCEL
  443. func initExcel() {
  444. orm.RegisterModel(
  445. new(excel.ExcelClassify), //ETA excel表格分类
  446. new(excel.ExcelInfo), //ETA excel表格
  447. new(excel.ExcelDraft), //ETA excel表格草稿
  448. new(excel.ExcelSheet), //ETA excel sheet
  449. new(excel.ExcelSheetData), //ETA excel sheet data
  450. new(excel.ExcelEdbMapping), //ETA excel 与 指标 的关系表
  451. new(excel.ExcelWorker), // 平衡表协作人表格
  452. new(excel.ExcelChartEdb), // 平衡表做图指标
  453. new(excel.ExcelChartData), // 平衡表作图数据
  454. new(excel.ExcelInfoRuleMapping), //表格的管理规则
  455. )
  456. }
  457. // initSmartReport 智能研报相关表
  458. func initSmartReport() {
  459. orm.RegisterModel(
  460. new(smart_report.SmartReport), // 智能研报主表
  461. new(smart_report.SmartReportSaveLog), // 智能研报-保存记录表
  462. new(smart_report.SmartReportResource), // 智能研报-资源表
  463. )
  464. }
  465. // initCrossVariety 跨品种分析
  466. func initCrossVariety() {
  467. orm.RegisterModel(
  468. new(cross_variety.ChartVariety), // 品种表
  469. new(cross_variety.ChartTag), // 标签表
  470. new(cross_variety.ChartTagVariety), // 标签、品种、指标关系表
  471. new(cross_variety.ChartVarietyMapping), // 图表与品种的关系表
  472. new(cross_variety.ChartInfoCrossVariety), // 跨品种分析配置表
  473. )
  474. }
  475. func initAi() {
  476. orm.RegisterModel(
  477. new(aimod.AiChatTopic),
  478. new(aimod.AiChat),
  479. new(aimod.FileUploadRecord),
  480. new(ai_summary.AiSummaryClassify),
  481. new(ai_summary.AiSummary),
  482. new(ai_summary.AiPrompt),
  483. )
  484. }
  485. // initReportApprove 报告审批相关表
  486. func initReportApprove() {
  487. orm.RegisterModel(
  488. new(report_approve.ReportApprove), // 审批表
  489. new(report_approve.ReportApproveFlow), // 审批流表
  490. new(report_approve.ReportApproveNode), // 审批节点表
  491. new(report_approve.ReportApproveRecord), // 审批记录表
  492. new(report_approve.ReportApproveMessage), // 审批消息表
  493. )
  494. }
  495. // initChartTheme 初始化图表主题
  496. func initChartTheme() {
  497. orm.RegisterModel(
  498. new(chart_theme.ChartTheme), // 图表主题
  499. new(chart_theme.ChartThemeType), // 图表主题类型
  500. new(chart_theme.ChartThemeDefaultData), //默认数据
  501. )
  502. }
  503. // initEdbRefresh 初始化指标刷新
  504. func initEdbRefresh() {
  505. orm.RegisterModel(
  506. new(edb_refresh.EdbRefreshSource), // 刷新的数据源表
  507. new(edb_refresh.EdbRefreshDefaultConfig), // 指标的默认刷新时间配置表
  508. new(edb_refresh.EdbRefreshConfig), // 指标的刷新时间配置表
  509. new(edb_refresh.EdbRefreshMapping), // 指标刷新时间配置关系表
  510. )
  511. }
  512. // initSpeechRecognition 语音识别
  513. func initSpeechRecognition() {
  514. orm.RegisterModel(
  515. new(speech_recognition.SpeechRecognition), // 语音识别表
  516. new(speech_recognition.SpeechRecognitionApiLog), // 语音识别-API请求日志
  517. new(speech_recognition.SpeechRecognitionContent), // 语音识别-转换内容表
  518. new(speech_recognition.SpeechRecognitionMenu), // 语音识别-目录表
  519. new(speech_recognition.SpeechRecognitionTag), // 语音识别-标签表
  520. new(speech_recognition.SpeechRecognitionTagMenu), // 语音识别-标签目录表
  521. new(speech_recognition.SpeechRecognitionTagMapping), // 语音识别-标签关联表
  522. )
  523. }
  524. // initDataMangePerMission
  525. // @Description: 初始化数据资产权限的一些表
  526. // @author: Roc
  527. // @datetime 2024-03-27 14:03:11
  528. func initDataMangePerMission() {
  529. orm.RegisterModel(
  530. new(data_manage_permission.EdbInfoPermission), // 指标权限表
  531. new(data_manage_permission.EdbClassifyPermission), // 指标分类权限表
  532. new(data_manage_permission.ChartInfoPermission), // 图表权限表
  533. new(data_manage_permission.ChartClassifyPermission), // 图表分类权限表
  534. new(data_manage_permission.ExcelInfoPermission), // ETA表格权限表
  535. new(data_manage_permission.ExcelClassifyPermission), // ETA表格分类权限表
  536. new(data_manage_permission.DataPermissionMessage), // 数据权限变更消息表
  537. new(data_manage_permission.DataPermissionMoveRecord), // 数据资产转移记录表
  538. new(data_manage_permission.EdbInfoPermissionNoAuthRecord), // 指标资产数据未授权分类记录表
  539. new(data_manage_permission.ChartInfoPermissionNoAuthRecord), // 图表资产数据未授权分类记录表
  540. new(data_manage_permission.ExcelInfoPermissionNoAuthRecord), // ETA表格资产数据未授权分类记录表
  541. new(data_manage_permission.EdbInfoClassifyPermissionNoAuthRecord), // 指标资产分类数据未授权分类记录表
  542. new(data_manage_permission.ChartInfoClassifyPermissionNoAuthRecord), // 图表资产分类数据未授权分类记录表
  543. new(data_manage_permission.ExcelInfoClassifyPermissionNoAuthRecord), // ETA表格资产分类数据未授权分类记录表
  544. new(data_manage_permission.DataPermissionClassifyNoAuthRecord), // 资产分类数据权限未授权记录表
  545. new(data_manage_permission.DataPermissionNoAuthRecord), // 资产数据权限设置记录表
  546. )
  547. }
  548. // initFeCalendar 初始化外汇日历
  549. func initFeCalendar() {
  550. orm.RegisterModel(
  551. new(fe_calendar.FeCalendarMatter), // 事项表
  552. )
  553. }
  554. // initFactorEdbSeries 因子指标系列数据表
  555. func initFactorEdbSeries() {
  556. orm.RegisterModel(
  557. new(data_manage.FactorEdbSeries), // 因子指标系列
  558. new(data_manage.FactorEdbSeriesChartMapping), // 因子指标系列-图表关联
  559. new(data_manage.FactorEdbSeriesMapping), // 因子指标系列-指标计算数据
  560. new(data_manage.FactorEdbSeriesCalculateData), // 因子指标系列-指标关联
  561. new(data_manage.FactorEdbSeriesCalculateDataQjjs), // 因子指标系列-区间计算数据
  562. )
  563. }
  564. // 预警管理
  565. func initEdbMonitor() {
  566. orm.RegisterModel(
  567. new(edbmonitor.EdbMonitorInfo), // 预警管理表
  568. new(edbmonitor.EdbMonitorClassify), // 预警管理分类表
  569. new(edbmonitor.EdbMonitorMessage), // 预警管理消息表
  570. )
  571. }
  572. func initBinlog() {
  573. orm.RegisterModel(
  574. new(binlogDao.BusinessSysInteractionLog), // binlog表
  575. )
  576. }
  577. // afterInitTable
  578. // @Description: 初始化表结构的的后置操作
  579. // @author: Roc
  580. // @datetime 2024-07-01 13:31:09
  581. func afterInitTable() {
  582. // 初始化指标来源配置
  583. data_manage.InitEdbSourceVar()
  584. // 初始化是否启用mongo配置
  585. InitUseMongoConf()
  586. }