db.go 22 KB

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