db.go 23 KB

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