db.go 22 KB

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