db.go 24 KB

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