chart_common.go 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "eta/eta_chart_lib/models"
  5. aiPredictModel "eta/eta_chart_lib/models/ai_predict_model"
  6. "eta/eta_chart_lib/models/data_manage"
  7. cross_varietyReq "eta/eta_chart_lib/models/data_manage/cross_variety/request"
  8. "eta/eta_chart_lib/models/data_manage/future_good"
  9. "eta/eta_chart_lib/models/data_manage/future_good/request"
  10. "eta/eta_chart_lib/models/data_manage/future_good/response"
  11. line_equationReq "eta/eta_chart_lib/models/data_manage/line_equation/request"
  12. line_featureReq "eta/eta_chart_lib/models/data_manage/line_feature/request"
  13. "eta/eta_chart_lib/services/data"
  14. correlationServ "eta/eta_chart_lib/services/data/correlation"
  15. "eta/eta_chart_lib/services/data/cross_variety"
  16. "eta/eta_chart_lib/services/data/excel"
  17. future_goodServ "eta/eta_chart_lib/services/data/future_good"
  18. "eta/eta_chart_lib/services/data/line_equation"
  19. lineFeatureServ "eta/eta_chart_lib/services/data/line_feature"
  20. "eta/eta_chart_lib/services/data/range_analysis"
  21. dwmini "eta/eta_chart_lib/services/dw_mini"
  22. "eta/eta_chart_lib/utils"
  23. "fmt"
  24. "sort"
  25. "strconv"
  26. "strings"
  27. "time"
  28. )
  29. // CommonChartInfoDetailFromUniqueCode
  30. // @Title 根据编码获取图表详情
  31. // @Description 根据编码获取图表详情接口
  32. // @Param UniqueCode query int true "图表唯一编码,如果是管理后台访问,传固定字符串:7c69b590249049942070ae9dcd5bf6dc"
  33. // @Param IsCache query bool true "是否走缓存,默认false"
  34. // @Param Token query string true "东吴小程序token"
  35. // @Param Source query int true "查询来源 1:东吴"
  36. // @Success 200 {object} data_manage.ChartInfoDetailFromUniqueCodeResp
  37. // @router /common/detail [get]
  38. func (this *ChartController) CommonChartInfoDetailFromUniqueCode() {
  39. br := new(models.BaseResponse).Init()
  40. defer func() {
  41. this.Data["json"] = br
  42. this.ServeJSON()
  43. }()
  44. uniqueCode := this.GetString("UniqueCode")
  45. token := this.GetString("Token")
  46. source, _ := this.GetInt("Source")
  47. if uniqueCode == "" {
  48. br.Msg = "参数错误"
  49. br.ErrMsg = "参数错误,uniqueCode is empty"
  50. return
  51. }
  52. key := utils.HZ_CHART_LIB_DETAIL + uniqueCode
  53. resp := new(models.ChartInfoDetailResp)
  54. // 图表水印
  55. conf, e := models.GetBusinessConf()
  56. if e != nil {
  57. br.Msg = "获取失败"
  58. br.ErrMsg = "获取配置信息失败, Err: " + e.Error()
  59. return
  60. }
  61. var isCollect bool
  62. if source == utils.CHART_SOURCE_DW && token != "" {
  63. tmpIsCollect, err := dwmini.GetMyChartIsCollect(token, uniqueCode)
  64. if err != nil {
  65. br.Msg = "获取失败"
  66. br.ErrMsg = "获取收藏状态失败,Err:" + err.Error()
  67. return
  68. }
  69. isCollect = tmpIsCollect
  70. }
  71. //判断是否有缓存
  72. if utils.Re == nil {
  73. if utils.Re == nil && utils.Rc.IsExist(key) {
  74. if data, err1 := utils.Rc.RedisBytes(key); err1 == nil {
  75. err := json.Unmarshal(data, &resp)
  76. if err == nil && resp != nil {
  77. if conf[models.BusinessConfWatermarkChart] == "true" && conf[models.BusinessConfCompanyWatermark] != "" {
  78. resp.WaterMark = conf[models.BusinessConfCompanyWatermark]
  79. }
  80. if isCollect {
  81. resp.IsCollect = isCollect
  82. }
  83. br.Ret = 200
  84. br.Success = true
  85. br.Msg = "获取成功"
  86. br.Data = resp
  87. fmt.Println("source redis")
  88. return
  89. }
  90. }
  91. }
  92. }
  93. chartInfo, err := models.GetChartInfoByUniqueCode(uniqueCode)
  94. if err != nil {
  95. if utils.IsErrNoRow(err) {
  96. br.Msg = "该图已被删除,请刷新页面"
  97. br.ErrMsg = "该图已被删除,请刷新页面,Err:" + err.Error()
  98. return
  99. }
  100. br.Msg = "获取失败"
  101. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  102. return
  103. }
  104. //var resp interface{}
  105. var isOk bool
  106. var msg, errMsg string
  107. switch chartInfo.Source {
  108. case utils.CHART_SOURCE_DEFAULT:
  109. resp, isOk, msg, errMsg = GetChartInfoDetailFromUniqueCode(chartInfo, key)
  110. case utils.CHART_SOURCE_FUTURE_GOOD:
  111. resp, isOk, msg, errMsg = GetFutureGoodChartInfoDetailFromUniqueCode(chartInfo, key)
  112. case utils.CHART_SOURCE_FUTURE_GOOD_PROFIT:
  113. resp, isOk, msg, errMsg = GetFutureGoodProfitChartInfoDetailFromUniqueCode(chartInfo, key)
  114. case utils.CHART_SOURCE_CORRELATION, utils.CHART_SOURCE_ROLLING_CORRELATION:
  115. resp, isOk, msg, errMsg = GetCorrelationChartInfoDetailFromUniqueCode(chartInfo, key)
  116. case utils.CHART_SOURCE_LINE_EQUATION:
  117. resp, isOk, msg, errMsg = GetLineEquationChartInfoDetailFromUniqueCode(chartInfo, key)
  118. case utils.CHART_SOURCE_LINE_FEATURE_STANDARD_DEVIATION, utils.CHART_SOURCE_LINE_FEATURE_PERCENTILE, utils.CHART_SOURCE_LINE_FEATURE_FREQUENCY:
  119. resp, isOk, msg, errMsg = GetLineFeatureChartInfoDetailFromUniqueCode(chartInfo, key)
  120. case utils.CHART_SOURCE_CROSS_HEDGING:
  121. resp, isOk, msg, errMsg = GetCrossVarietyChartInfoDetailFromUniqueCode(chartInfo, key)
  122. case utils.CHART_SOURCE_BALANCE_EXCEL:
  123. resp, isOk, msg, errMsg = GetBalanceChartInfoDetailFromUniqueCode(chartInfo, key, this.Lang)
  124. if !isOk {
  125. br.Msg = msg
  126. br.ErrMsg = errMsg
  127. return
  128. }
  129. case utils.CHART_SOURCE_RANGE_ANALYSIS:
  130. resp, isOk, msg, errMsg = GetRangeAnalysisChartInfoDetailFromUniqueCode(chartInfo, key, this.Lang)
  131. if !isOk {
  132. br.Msg = msg
  133. br.ErrMsg = errMsg
  134. return
  135. }
  136. default:
  137. br.Msg = "错误的图表"
  138. br.ErrMsg = "错误的图表"
  139. return
  140. }
  141. if !isOk {
  142. br.Msg = msg
  143. br.ErrMsg = errMsg
  144. return
  145. }
  146. if isCollect {
  147. resp.IsCollect = isCollect
  148. }
  149. if conf[models.BusinessConfWatermarkChart] == "true" && conf[models.BusinessConfCompanyWatermark] != "" {
  150. resp.WaterMark = conf[models.BusinessConfCompanyWatermark]
  151. }
  152. br.Ret = 200
  153. br.Success = true
  154. br.Msg = "获取成功"
  155. br.Data = resp
  156. }
  157. // GetFutureGoodChartInfoDetailFromUniqueCode 根据编码获取图表详情
  158. func GetFutureGoodChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  159. resp = new(models.ChartInfoDetailResp)
  160. // 获取主题样式
  161. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  162. if err != nil {
  163. msg = "获取失败"
  164. errMsg = "获取主题信息失败,Err:" + err.Error()
  165. return
  166. }
  167. chartInfo.ChartThemeStyle = chartTheme.Config
  168. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  169. chartInfoId := chartInfo.ChartInfoId
  170. startDate := chartInfo.StartDate
  171. endDate := chartInfo.EndDate
  172. // 兼容日期错误
  173. {
  174. if strings.Count(startDate, "-") == 1 {
  175. startDate = startDate + "-01"
  176. }
  177. if strings.Count(endDate, "-") == 1 {
  178. endDate = endDate + "-01"
  179. }
  180. }
  181. edbInfoMappingList, err := models.GetEtaEdbChartEdbMappingList(chartInfoId)
  182. if err != nil {
  183. msg = "获取失败"
  184. errMsg = "获取图表,现货指标信息失败,Err:" + err.Error()
  185. return
  186. }
  187. futureGoodEdbInfoMapping, err := models.GetFutureGoodEdbChartEdbMapping(chartInfoId)
  188. if err != nil {
  189. msg = "获取失败"
  190. errMsg = "获取图表的期货商品指标信息失败,Err:" + err.Error()
  191. return
  192. }
  193. // 商品价格曲线图的一些配置
  194. var barConfig data_manage.FutureGoodBarChartInfoReq
  195. //barChartInfoSort := data_manage.BarChartInfoSortReq{}
  196. if chartInfo.BarConfig == `` {
  197. msg = "商品价格曲线图未配置"
  198. errMsg = "商品价格曲线图未配置"
  199. return
  200. }
  201. err = json.Unmarshal([]byte(chartInfo.BarConfig), &barConfig)
  202. if err != nil {
  203. msg = "商品价格曲线图配置异常"
  204. errMsg = "商品价格曲线图配置异常"
  205. return
  206. }
  207. baseEdbInfoId := barConfig.BaseEdbInfoId
  208. var baseEdbInfoMapping *models.ChartEdbInfoMapping
  209. // todo 兼容历史数据,
  210. if baseEdbInfoId == 0 {
  211. // 默认取第一个现货指标
  212. baseEdbInfoId = edbInfoMappingList[0].EdbInfoId
  213. baseEdbInfoMapping = edbInfoMappingList[0]
  214. barConfig.BaseEdbInfoId = baseEdbInfoId
  215. } else {
  216. baseEdbInfoMapping, err = models.GetChartEdbMappingByEdbInfoId(baseEdbInfoId)
  217. if err != nil {
  218. msg = "获取失败"
  219. errMsg = "获取图表,指标信息失败,Err:" + err.Error()
  220. return
  221. }
  222. }
  223. // 获取图表中的指标数据
  224. barConfigEdbInfoIdList, edbList, xEdbIdValue, xDataList, yDataList, err := future_goodServ.GetChartEdbData(chartInfoId, startDate, endDate, baseEdbInfoMapping, edbInfoMappingList, futureGoodEdbInfoMapping, barConfig, true)
  225. if err != nil {
  226. msg = "获取失败"
  227. errMsg = "获取图表,指标信息失败,Err:" + err.Error()
  228. return
  229. }
  230. warnEdbList := make([]string, 0)
  231. if len(edbList) <= 0 {
  232. msg = "商品价格曲线图表指标异常"
  233. errMsg = "商品价格曲线图表异常"
  234. return
  235. }
  236. baseEdbInfo := edbList[0] //现货指标
  237. for _, v := range edbList {
  238. if v.IsNullData {
  239. warnEdbList = append(warnEdbList, v.EdbName+"("+v.EdbCode+")")
  240. }
  241. // 指标别名
  242. if barConfigEdbInfoIdList != nil && len(barConfigEdbInfoIdList) > 0 {
  243. for _, reqEdb := range barConfigEdbInfoIdList {
  244. if v.EdbInfoId == reqEdb.EdbInfoId {
  245. v.EdbAliasName = reqEdb.Name
  246. v.EdbAliasNameEn = reqEdb.NameEn
  247. }
  248. }
  249. }
  250. }
  251. chartInfo.UnitEn = baseEdbInfo.UnitEn
  252. // 图表的指标来源
  253. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  254. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  255. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  256. xDataListFinal := make([]models.XData, 0)
  257. for _, v := range xDataList {
  258. tmp := models.XData{
  259. Name: v.Name,
  260. NameEn: v.NameEn,
  261. IsHide: v.IsHide,
  262. }
  263. xDataListFinal = append(xDataListFinal, tmp)
  264. }
  265. resp.ChartInfo = chartInfo
  266. resp.EdbInfoList = edbList
  267. resp.XEdbIdValue = xEdbIdValue
  268. resp.YDataList = yDataList
  269. resp.XDataList = xDataListFinal
  270. //resp.BarChartInfo = barConfig
  271. //resp.Status = true
  272. // 将数据加入缓存
  273. if utils.Re == nil {
  274. data, _ := json.Marshal(resp)
  275. utils.Rc.Put(key, data, 2*time.Hour)
  276. }
  277. isOk = true
  278. return
  279. }
  280. // GetFutureGoodProfitChartInfoDetailFromUniqueCode 根据编码获取商品利润图表详情
  281. func GetFutureGoodProfitChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  282. resp = new(models.ChartInfoDetailResp)
  283. chartInfoId := chartInfo.ChartInfoId
  284. // 获取主题样式
  285. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  286. if err != nil {
  287. msg = "获取失败"
  288. errMsg = "获取主题信息失败,Err:" + err.Error()
  289. return
  290. }
  291. chartInfo.ChartThemeStyle = chartTheme.Config
  292. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  293. // 商品利润曲线图的一些配置
  294. var extraConf request.ChartInfoReq
  295. err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &extraConf)
  296. if err != nil {
  297. msg = "商品利润曲线图配置异常"
  298. errMsg = "商品利润曲线图配置异常,Err:" + err.Error()
  299. return
  300. }
  301. if len(extraConf.EdbInfoIdList) == 0 {
  302. extraConf.EdbInfoIdList = append(extraConf.EdbInfoIdList, extraConf.BaseEdbInfoId)
  303. }
  304. edbList := make([]*models.ChartEdbInfoMapping, 0)
  305. edbInfoMappingList, err := models.GetEtaEdbChartEdbMappingList(chartInfoId)
  306. if err != nil {
  307. msg = "获取失败"
  308. errMsg = "获取商品利润图表,基础指标信息失败,Err:" + err.Error()
  309. return
  310. }
  311. baseEdbInfo := new(models.ChartEdbInfoMapping)
  312. for _, v := range edbInfoMappingList {
  313. if v.EdbInfoId == extraConf.BaseEdbInfoId {
  314. baseEdbInfo = v
  315. }
  316. }
  317. edbList = edbInfoMappingList
  318. futureGoodEdbInfoMappingList, err := models.GetFutureGoodEdbChartEdbMappingList(chartInfoId)
  319. if err != nil {
  320. msg = "获取失败"
  321. errMsg = "获取商品利润图表,商品指标信息失败,Err:" + err.Error()
  322. return
  323. }
  324. edbList = append(edbList, futureGoodEdbInfoMappingList...)
  325. if len(edbList) <= 0 {
  326. msg = "商品利润曲线图表指标异常"
  327. errMsg = "商品利润曲线图表指标异常"
  328. return
  329. }
  330. xDataList := make([]models.XData, 0)
  331. yDataList := make([]models.YData, 0)
  332. // 查找商品利润图表的扩展信息
  333. chartInfoFutureGoodProfit := new(future_good.ChartInfoFutureGoodProfit)
  334. if err = chartInfoFutureGoodProfit.GetItemById(chartInfo.ChartInfoId); err != nil {
  335. msg = "获取失败"
  336. errMsg = "获取基础相关性图表信息失败, Err: " + err.Error()
  337. return
  338. }
  339. err = json.Unmarshal([]byte(chartInfoFutureGoodProfit.XValue), &xDataList)
  340. if err != nil {
  341. msg = "获取失败"
  342. errMsg = "转换X轴数据失败, Err: " + err.Error()
  343. return
  344. }
  345. err = json.Unmarshal([]byte(chartInfoFutureGoodProfit.YValue), &yDataList)
  346. if err != nil {
  347. msg = "获取失败"
  348. errMsg = "转换Y轴数据失败, Err: " + err.Error()
  349. return
  350. }
  351. warnEdbList := make([]string, 0)
  352. for _, v := range edbList {
  353. if v.IsNullData {
  354. warnEdbList = append(warnEdbList, v.EdbName+"("+v.EdbCode+")")
  355. }
  356. }
  357. chartInfo.UnitEn = baseEdbInfo.UnitEn
  358. resp.ChartInfo = chartInfo
  359. resp.EdbInfoList = edbList
  360. resp.DataResp = response.ProfitFutureGoodChartResp{
  361. YDataList: yDataList,
  362. XDataList: xDataList,
  363. ProfitName: chartInfoFutureGoodProfit.ProfitName,
  364. ProfitNameEn: chartInfoFutureGoodProfit.ProfitNameEn,
  365. Extra: extraConf,
  366. }
  367. // 将数据加入缓存
  368. if utils.Re == nil {
  369. data, _ := json.Marshal(resp)
  370. utils.Rc.Put(key, data, 2*time.Hour)
  371. }
  372. isOk = true
  373. return
  374. }
  375. // GetLineEquationChartInfoDetailFromUniqueCode 根据编码获取拟合方程图表详情
  376. func GetLineEquationChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  377. resp = new(models.ChartInfoDetailResp)
  378. // 获取主题样式
  379. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  380. if err != nil {
  381. msg = "获取失败"
  382. errMsg = "获取主题信息失败,Err:" + err.Error()
  383. return
  384. }
  385. chartInfo.ChartThemeStyle = chartTheme.Config
  386. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  387. //chartInfoId := chartInfo.ChartInfoId
  388. if chartInfo.ExtraConfig == `` {
  389. msg = "获取失败"
  390. errMsg = "获取配置信息失败"
  391. return
  392. }
  393. var lineChartInfoConfig line_equationReq.LineChartInfoReq
  394. err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &lineChartInfoConfig)
  395. if err != nil {
  396. msg = "获取失败"
  397. errMsg = "获取图表配置信息失败, Err:" + err.Error()
  398. return
  399. }
  400. var getAData, getBData, getR2Data bool
  401. switch lineChartInfoConfig.Source {
  402. case utils.CHART_MULTIPLE_GRAPH_LINE_EQUATION_ONE:
  403. getAData = true
  404. case utils.CHART_MULTIPLE_GRAPH_LINE_EQUATION_TWO:
  405. getBData = true
  406. case utils.CHART_MULTIPLE_GRAPH_LINE_EQUATION_THREE:
  407. getR2Data = true
  408. }
  409. edbList, dataResp, err, errMsg := line_equation.GetChartEdbData(chartInfo.ChartInfoId, lineChartInfoConfig, getAData, getBData, getR2Data)
  410. if err != nil {
  411. if errMsg == `` {
  412. errMsg = "获取失败"
  413. }
  414. return
  415. }
  416. var resultResp interface{}
  417. switch lineChartInfoConfig.Source {
  418. case utils.CHART_MULTIPLE_GRAPH_LINE_EQUATION_ONE:
  419. resultResp = dataResp.AData
  420. case utils.CHART_MULTIPLE_GRAPH_LINE_EQUATION_TWO:
  421. resultResp = dataResp.BData
  422. case utils.CHART_MULTIPLE_GRAPH_LINE_EQUATION_THREE:
  423. resultResp = dataResp.R2Data
  424. }
  425. //chartInfo.UnitEn = baseEdbInfo.UnitEn
  426. // 图表的指标来源
  427. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  428. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  429. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  430. resp.ChartInfo = chartInfo
  431. resp.DataResp = resultResp
  432. resp.EdbInfoList = edbList
  433. // 将数据加入缓存
  434. if utils.Re == nil {
  435. data, _ := json.Marshal(resp)
  436. utils.Rc.Put(key, data, 2*time.Hour)
  437. }
  438. isOk = true
  439. return
  440. }
  441. // GetLineFeatureChartInfoDetailFromUniqueCode 根据编码获取统计特征图表详情
  442. func GetLineFeatureChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  443. resp = new(models.ChartInfoDetailResp)
  444. // 获取主题样式
  445. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  446. if err != nil {
  447. msg = "获取失败"
  448. errMsg = "获取主题信息失败,Err:" + err.Error()
  449. return
  450. }
  451. chartInfo.ChartThemeStyle = chartTheme.Config
  452. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  453. // 获取图表关联指标
  454. edbMappingList, err := models.GetChartEdbMappingList(chartInfo.ChartInfoId)
  455. if err != nil {
  456. msg = "获取失败"
  457. errMsg = "获取图表关联指标信息失败,Err:" + err.Error()
  458. return
  459. }
  460. if len(edbMappingList) != 1 {
  461. msg = "获取失败"
  462. errMsg = fmt.Sprint("获取图表关联指标信息异常,数量:", len(edbMappingList))
  463. return
  464. }
  465. edbMapping := edbMappingList[0]
  466. var edbList []*models.ChartEdbInfoMapping
  467. var resultResp interface{}
  468. switch chartInfo.Source {
  469. case utils.CHART_SOURCE_LINE_FEATURE_STANDARD_DEVIATION:
  470. calculateValue, tmpErr := strconv.Atoi(chartInfo.ExtraConfig)
  471. if tmpErr != nil {
  472. msg = "获取失败"
  473. errMsg = "格式化配置项失败,Err:" + tmpErr.Error()
  474. return
  475. }
  476. startDate, endDate := utils.GetDateByDateType(chartInfo.DateType, chartInfo.StartDate, chartInfo.EndDate)
  477. edbList, resultResp, err, msg = lineFeatureServ.GetStandardDeviationData(chartInfo.ChartInfoId, startDate, endDate, edbMapping, calculateValue)
  478. case utils.CHART_SOURCE_LINE_FEATURE_PERCENTILE:
  479. var percentileConfig line_featureReq.Percentile
  480. err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &percentileConfig)
  481. if err != nil {
  482. msg = "获取失败"
  483. errMsg = "格式化配置项失败,Err:" + err.Error()
  484. return
  485. }
  486. startDate, endDate := utils.GetDateByDateType(chartInfo.DateType, chartInfo.StartDate, chartInfo.EndDate)
  487. edbList, resultResp, err, msg = lineFeatureServ.GetPercentileData(chartInfo.ChartInfoId, startDate, endDate, edbMapping, percentileConfig.CalculateValue, percentileConfig.CalculateUnit, percentileConfig.PercentType)
  488. case utils.CHART_SOURCE_LINE_FEATURE_FREQUENCY:
  489. var frequencyDistributionConfig line_featureReq.FrequencyDistribution
  490. err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &frequencyDistributionConfig)
  491. if err != nil {
  492. msg = "获取失败"
  493. errMsg = "格式化配置项失败,Err:" + err.Error()
  494. return
  495. }
  496. // 获取图表中的指标数据
  497. edbList, resultResp, err, errMsg = lineFeatureServ.GetFrequencyDistributionData(0, edbMapping, frequencyDistributionConfig.DateType, frequencyDistributionConfig.FrequencyValue, frequencyDistributionConfig.StartDate, frequencyDistributionConfig.EndDate)
  498. default:
  499. msg = `错误的图表`
  500. errMsg = fmt.Sprint("错误的图表来源,source", chartInfo.Source)
  501. return
  502. }
  503. if err != nil {
  504. if msg == `` {
  505. msg = "获取失败"
  506. }
  507. errMsg = "获取图表,指标信息失败,Err:" + err.Error()
  508. return
  509. }
  510. //chartInfo.UnitEn = baseEdbInfo.UnitEn
  511. // 图表的指标来源
  512. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  513. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  514. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  515. resp.ChartInfo = chartInfo
  516. resp.DataResp = resultResp
  517. resp.EdbInfoList = edbList
  518. // 将数据加入缓存
  519. if utils.Re == nil {
  520. data, _ := json.Marshal(resp)
  521. utils.Rc.Put(key, data, 2*time.Hour)
  522. }
  523. isOk = true
  524. return
  525. }
  526. // GetCrossVarietyChartInfoDetailFromUniqueCode 根据编码获取图表详情
  527. func GetCrossVarietyChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  528. resp = new(models.ChartInfoDetailResp)
  529. // 获取主题样式
  530. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 5)
  531. if err != nil {
  532. msg = "获取失败"
  533. errMsg = "获取主题信息失败,Err:" + err.Error()
  534. return
  535. }
  536. chartInfo.ChartThemeStyle = chartTheme.Config
  537. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  538. if chartInfo.ExtraConfig == `` {
  539. msg = "图表配置信息异常"
  540. errMsg = "图表配置信息异常"
  541. return
  542. }
  543. var config cross_varietyReq.ChartConfigReq
  544. err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &config)
  545. if err != nil {
  546. msg = "解析跨品种分析配置失败"
  547. errMsg = "解析跨品种分析配置失败,Err:" + err.Error()
  548. return
  549. }
  550. // 获取图表x轴y轴
  551. edbList, dataResp, err, msg, _ := cross_variety.GetChartData(0, config)
  552. if err != nil {
  553. errMsg = "获取图表,指标信息失败,Err:" + err.Error()
  554. return
  555. }
  556. //判断是否需要展示英文标识
  557. //chartInfo.IsEnChart = data.CheckIsEnChart(chartInfo.ChartNameEn, edbList[0:1], chartInfo.Source, chartInfo.ChartType)
  558. //chartInfo.UnitEn = edbInfoMappingA.UnitEn
  559. // 另存为
  560. resp.ChartInfo = chartInfo
  561. resp.DataResp = dataResp
  562. resp.EdbInfoList = edbList
  563. // 图表的指标来源
  564. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  565. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  566. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  567. // 将数据加入缓存
  568. if utils.Re == nil {
  569. d, _ := json.Marshal(resp)
  570. _ = utils.Rc.Put(key, d, 2*time.Hour)
  571. }
  572. isOk = true
  573. return
  574. }
  575. func GetBalanceChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key, lang string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  576. resp = new(models.ChartInfoDetailResp)
  577. msg = `获取失败`
  578. var err error
  579. defer func() {
  580. if err != nil {
  581. if errMsg != "" {
  582. msg = errMsg
  583. }
  584. errMsg = err.Error()
  585. }
  586. }()
  587. // 相关联指标
  588. mappingListTmp, dataListMap, err, errMsg := excel.GetBalanceExcelChartSingle(chartInfo.ChartInfoId, 0, lang)
  589. if err != nil {
  590. errMsg = "获取失败"
  591. err = fmt.Errorf(" 获取图表,指标信息失败 Err:%s", err.Error())
  592. return
  593. }
  594. var chartInfoResp *models.ChartInfoDetailResp
  595. chartInfoResp, err, errMsg = data.GetBalanceExcelChartDetail(chartInfo, mappingListTmp, dataListMap)
  596. if err != nil {
  597. msg = "查询图表详情失败"
  598. errMsg = "查询图表详情失败,Err:" + err.Error()
  599. return
  600. }
  601. resp = &models.ChartInfoDetailResp{
  602. ChartInfo: chartInfoResp.ChartInfo,
  603. EdbInfoList: chartInfoResp.EdbInfoList,
  604. XEdbIdValue: chartInfoResp.XEdbIdValue,
  605. YDataList: chartInfoResp.YDataList,
  606. XDataList: chartInfoResp.XDataList,
  607. CorrelationChartInfo: chartInfoResp.CorrelationChartInfo,
  608. DataResp: chartInfoResp.DataResp,
  609. }
  610. if utils.Re == nil {
  611. jsonData, _ := json.Marshal(resp)
  612. utils.Rc.Put(key, jsonData, 10*time.Minute)
  613. }
  614. isOk = true
  615. return
  616. }
  617. // FutureGoodChartInfoRefresh
  618. // @Title 商品价格图表刷新接口
  619. // @Description 商品价格图表刷新接口
  620. // @Param UniqueCode query string true "图表唯一编码,如果是管理后台访问,传固定字符串:7c69b590249049942070ae9dcd5bf6dc"
  621. // @Success Ret=200 刷新成功
  622. // @router /future_good/refresh [get]
  623. func (this *ChartController) FutureGoodChartInfoRefresh() {
  624. br := new(models.BaseResponse).Init()
  625. chartId := 0
  626. defer func() {
  627. // 添加日志
  628. if chartId > 0 {
  629. shareChartRefreshLogInfo := &models.ShareChartRefreshLog{
  630. Ip: this.Ctx.Input.IP(),
  631. ChartId: chartId,
  632. CreateTime: time.Now(),
  633. }
  634. models.AddShareChartRefreshLog(shareChartRefreshLogInfo)
  635. }
  636. this.Data["json"] = br
  637. this.ServeJSON()
  638. }()
  639. uniqueCode := this.GetString("UniqueCode")
  640. if uniqueCode == "" {
  641. br.Msg = "参数错误"
  642. br.ErrMsg = "参数错误,uniqueCode is empty"
  643. return
  644. }
  645. chartInfo, err := models.GetChartInfoByUniqueCode(uniqueCode)
  646. if err != nil {
  647. if utils.IsErrNoRow(err) {
  648. br.Msg = "该图已被删除,请刷新页面"
  649. br.ErrMsg = "该图已被删除,请刷新页面,Err:" + err.Error()
  650. return
  651. }
  652. br.Msg = "获取失败"
  653. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  654. return
  655. }
  656. chartId = chartInfo.ChartInfoId
  657. err = future_goodServ.FutureGoodChartInfoRefresh(chartInfo.ChartInfoId)
  658. if err != nil {
  659. br.Msg = "刷新失败"
  660. br.ErrMsg = "刷新图表关联指标信息失败,Err:" + err.Error()
  661. return
  662. }
  663. //err = data.ChartInfoRefresh(chartInfo.ChartInfoId)
  664. //if err != nil {
  665. // br.Msg = "刷新失败"
  666. // br.ErrMsg = "刷新图表关联指标信息失败,Err:" + err.Error()
  667. // return
  668. //}
  669. //err = data.ChartInfoRefreshV2(chartInfo.ChartInfoId)
  670. //if err != nil {
  671. // br.Msg = "刷新失败"
  672. // br.ErrMsg = "刷新图表关联指标信息失败,Err:" + err.Error()
  673. // return
  674. //}
  675. //清除数据缓存
  676. key := utils.HZ_CHART_LIB_DETAIL + uniqueCode
  677. if utils.Re == nil {
  678. utils.Rc.Delete(key)
  679. }
  680. br.Ret = 200
  681. br.Success = true
  682. br.Msg = "刷新成功"
  683. }
  684. // GetCorrelationChartInfoDetailFromUniqueCode 根据编码获取相关性图表详情
  685. func GetCorrelationChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  686. resp = new(models.ChartInfoDetailResp)
  687. // 获取主题样式
  688. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  689. if err != nil {
  690. msg = "获取失败"
  691. errMsg = "获取主题信息失败,Err:" + err.Error()
  692. return
  693. }
  694. chartInfo.ChartThemeStyle = chartTheme.Config
  695. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  696. chartInfoId := chartInfo.ChartInfoId
  697. startDate := chartInfo.StartDate
  698. endDate := chartInfo.EndDate
  699. // 兼容日期错误
  700. {
  701. if strings.Count(startDate, "-") == 1 {
  702. startDate = startDate + "-01"
  703. }
  704. if strings.Count(endDate, "-") == 1 {
  705. endDate = endDate + "-01"
  706. }
  707. }
  708. // 相关性图表信息
  709. correlationChart := new(data_manage.ChartInfoCorrelation)
  710. if e := correlationChart.GetItemById(chartInfoId); e != nil {
  711. msg = "获取失败"
  712. errMsg = "获取图表相关性信息失败, Err:" + e.Error()
  713. return
  714. }
  715. // 获取指标信息
  716. edbInfoMappingA, e := models.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdFirst)
  717. if e != nil {
  718. msg = "获取失败"
  719. errMsg = "获取相关性图表, A指标mapping信息失败, Err:" + e.Error()
  720. return
  721. }
  722. edbInfoMappingB := new(models.ChartEdbInfoMapping)
  723. if correlationChart.AnalysisMode != 1 {
  724. edbInfoMappingB, e = models.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdSecond)
  725. if e != nil {
  726. msg = "获取失败"
  727. errMsg = "获取相关性图表, B指标mapping信息失败, Err:" + e.Error()
  728. return
  729. }
  730. }
  731. var dataResp interface{} // 绘图数据返回(目前是滚动相关性的图)
  732. var xEdbIdValue []int
  733. var yDataList []models.YData
  734. if correlationChart.AnalysisMode != 1 {
  735. switch chartInfo.Source {
  736. case utils.CHART_SOURCE_CORRELATION: // 相关性图
  737. moveUnitDays, ok := utils.FrequencyDaysMap[correlationChart.CalculateUnit]
  738. if !ok {
  739. msg = "错误的分析周期"
  740. errMsg = "相关性图表数据有误"
  741. return
  742. }
  743. st := time.Now().AddDate(0, 0, -correlationChart.CalculateValue*moveUnitDays).Format(utils.FormatDate)
  744. ed := time.Now().Format(utils.FormatDate)
  745. xEdbIdValue, yDataList, e = correlationServ.GetChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, st, ed, chartInfo.ExtraConfig)
  746. if e != nil {
  747. msg = "获取失败"
  748. errMsg = "获取相关性图表, 图表计算值失败, Err:" + e.Error()
  749. return
  750. }
  751. case utils.CHART_SOURCE_ROLLING_CORRELATION: // 滚动相关性图
  752. st, ed := utils.GetDateByDateType(correlationChart.DateType, correlationChart.StartDate.Format(utils.FormatDate), correlationChart.EndDate.Format(utils.FormatDate))
  753. dataResp, e = correlationServ.GetRollingCorrelationChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, correlationChart.CalculateValue, correlationChart.CalculateUnit, st, ed, chartInfo.ChartName, chartInfo.ChartNameEn)
  754. }
  755. } else {
  756. xEdbIdValue, yDataList, e = correlationServ.GetFactorChartDataByChartId(chartInfoId, chartInfo.ExtraConfig)
  757. if e != nil {
  758. msg = "获取失败"
  759. errMsg = "获取相关性图表, 图表计算值失败, Err:" + e.Error()
  760. return
  761. }
  762. }
  763. // 完善指标信息
  764. edbList, e := correlationServ.GetChartEdbInfoFormat(chartInfo.ChartInfoId, edbInfoMappingA, edbInfoMappingB)
  765. if e != nil {
  766. msg = "获取失败"
  767. errMsg = "获取相关性图表, 完善指标信息失败, Err:" + e.Error()
  768. return
  769. }
  770. // 图表的指标来源
  771. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  772. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  773. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  774. correlationInfo := new(models.CorrelationInfo)
  775. correlationInfo.LeadValue = correlationChart.LeadValue
  776. correlationInfo.LeadUnit = correlationChart.LeadUnit
  777. correlationInfo.StartDate = correlationChart.StartDate.Format(utils.FormatDate)
  778. correlationInfo.EndDate = correlationChart.EndDate.Format(utils.FormatDate)
  779. correlationInfo.LeadValue = correlationChart.LeadValue
  780. correlationInfo.EdbInfoIdFirst = correlationChart.EdbInfoIdFirst
  781. correlationInfo.EdbInfoIdSecond = correlationChart.EdbInfoIdSecond
  782. correlationInfo.AnalysisMode = correlationChart.AnalysisMode
  783. resp.ChartInfo = chartInfo
  784. resp.EdbInfoList = edbList
  785. resp.XEdbIdValue = xEdbIdValue
  786. resp.YDataList = yDataList
  787. resp.CorrelationChartInfo = correlationInfo
  788. resp.DataResp = dataResp
  789. // 将数据加入缓存
  790. if utils.Re == nil {
  791. d, _ := json.Marshal(resp)
  792. _ = utils.Rc.Put(key, d, 2*time.Hour)
  793. }
  794. isOk = true
  795. return
  796. }
  797. // CorrelationChartInfoRefresh
  798. // @Title 商品价格图表刷新接口
  799. // @Description 商品价格图表刷新接口
  800. // @Param UniqueCode query string true "图表唯一编码,如果是管理后台访问,传固定字符串:7c69b590249049942070ae9dcd5bf6dc"
  801. // @Success Ret=200 刷新成功
  802. // @router /correlation/refresh [get]
  803. func (this *ChartController) CorrelationChartInfoRefresh() {
  804. br := new(models.BaseResponse).Init()
  805. chartId := 0
  806. defer func() {
  807. // 添加日志
  808. if chartId > 0 {
  809. shareChartRefreshLogInfo := &models.ShareChartRefreshLog{
  810. Ip: this.Ctx.Input.IP(),
  811. ChartId: chartId,
  812. CreateTime: time.Now(),
  813. }
  814. models.AddShareChartRefreshLog(shareChartRefreshLogInfo)
  815. }
  816. this.Data["json"] = br
  817. this.ServeJSON()
  818. }()
  819. uniqueCode := this.GetString("UniqueCode")
  820. if uniqueCode == "" {
  821. br.Msg = "参数错误"
  822. br.ErrMsg = "参数错误,uniqueCode is empty"
  823. return
  824. }
  825. chartInfo, err := models.GetChartInfoByUniqueCode(uniqueCode)
  826. if err != nil {
  827. if utils.IsErrNoRow(err) {
  828. br.Msg = "该图已被删除,请刷新页面"
  829. br.ErrMsg = "该图已被删除,请刷新页面,Err:" + err.Error()
  830. return
  831. }
  832. br.Msg = "获取失败"
  833. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  834. return
  835. }
  836. chartId = chartInfo.ChartInfoId
  837. err = data.ChartInfoRefreshV2(chartInfo.ChartInfoId)
  838. if err != nil {
  839. br.Msg = "刷新失败"
  840. br.ErrMsg = "刷新图表关联指标信息失败,Err:" + err.Error()
  841. return
  842. }
  843. //// 刷新相关性图表
  844. //if e := correlationServ.ChartInfoRefresh(chartInfo.ChartInfoId); e != nil {
  845. // br.Msg = "刷新失败"
  846. // br.ErrMsg = "刷新相关性图表失败, Err:" + e.Error()
  847. // return
  848. //}
  849. //清除数据缓存
  850. key := utils.HZ_CHART_LIB_DETAIL + uniqueCode
  851. if utils.Re == nil {
  852. _ = utils.Rc.Delete(key)
  853. }
  854. br.Ret = 200
  855. br.Success = true
  856. br.Msg = "刷新成功"
  857. }
  858. // GetRangeAnalysisChartInfoDetailFromUniqueCode 根据编码获取区间计算图表详情
  859. func GetRangeAnalysisChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key, lang string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  860. resp = new(models.ChartInfoDetailResp)
  861. // 获取主题样式
  862. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  863. if err != nil {
  864. msg = "获取失败"
  865. errMsg = "获取主题信息失败,Err:" + err.Error()
  866. return
  867. }
  868. chartInfo.ChartThemeStyle = chartTheme.Config
  869. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  870. chartInfoId := chartInfo.ChartInfoId
  871. // 获取指标信息
  872. //chartInfo.CorrelationLeadUnit = req.LeadUnit
  873. edbInfoMappingList, err := models.GetChartEdbMappingList(chartInfoId)
  874. if err != nil {
  875. msg = "获取失败"
  876. errMsg = "获取图表,指标信息失败,Err:" + err.Error()
  877. return
  878. }
  879. dateType := chartInfo.DateType
  880. // 开始/结束日期
  881. startYear := chartInfo.StartYear
  882. startDate := chartInfo.StartDate
  883. endDate := chartInfo.EndDate
  884. // 区间计算图表配置校验
  885. var extraConfig models.ChartRangeAnalysisExtraConf
  886. if chartInfo.ExtraConfig == `` {
  887. msg = "配置信息错误"
  888. return
  889. }
  890. err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &extraConfig)
  891. if err != nil {
  892. msg = "配置信息错误"
  893. errMsg = "图表配置信息错误,Err:" + err.Error()
  894. return
  895. }
  896. // 获取图表数据
  897. if len(edbInfoMappingList) == 0 {
  898. msg = "获取失败"
  899. errMsg = "图表没有指标,无法计算"
  900. return
  901. }
  902. // 获取图表x轴y轴
  903. edbList, xEdbIdValue, dataResp, e := range_analysis.GetChartDataByEdbInfoList(chartInfoId, dateType, startYear, startDate, endDate, edbInfoMappingList, &extraConfig)
  904. if e != nil {
  905. msg = "获取失败"
  906. errMsg = "获取区间计算图表, 图表计算值失败, Err:" + e.Error()
  907. return
  908. }
  909. // 图表的指标来源
  910. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  911. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  912. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  913. resp.ChartInfo = chartInfo
  914. resp.EdbInfoList = edbList
  915. resp.XEdbIdValue = xEdbIdValue
  916. resp.DataResp = dataResp
  917. if utils.Re == nil {
  918. jsonData, _ := json.Marshal(resp)
  919. utils.Rc.Put(key, jsonData, 10*time.Minute)
  920. }
  921. isOk = true
  922. return
  923. }
  924. // GetAiPredictChartInfoDetailFromUniqueCode 根据编码获取AI预测模型图表详情
  925. func GetAiPredictChartInfoDetailFromUniqueCode(chartInfo *models.ChartInfo, key string) (resp *models.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  926. var err error
  927. msg = "获取成功"
  928. defer func() {
  929. if err != nil {
  930. tips := fmt.Sprintf("UniqueCode获取图表详情失败, %v", err)
  931. msg = "获取失败"
  932. errMsg = fmt.Sprintf(tips)
  933. utils.FileLog.Info(tips)
  934. }
  935. }()
  936. if chartInfo == nil {
  937. err = fmt.Errorf("图表信息不存在")
  938. return
  939. }
  940. if chartInfo.Source != utils.CHART_SOURCE_AI_PREDICT_MODEL_DAILY && chartInfo.Source != utils.CHART_SOURCE_AI_PREDICT_MODEL_MONTHLY {
  941. err = fmt.Errorf("图表来源有误, Source: %d", chartInfo.Source)
  942. return
  943. }
  944. resp = new(models.ChartInfoDetailResp)
  945. // 获取图表标的
  946. edbMappings, e := models.GetChartEdbMappingsByChartInfoId(chartInfo.ChartInfoId)
  947. if e != nil {
  948. err = fmt.Errorf("获取图表指标关联失败, %v", e)
  949. return
  950. }
  951. if len(edbMappings) == 0 {
  952. err = fmt.Errorf("图表指标关联不存在, %v", e)
  953. return
  954. }
  955. indexId := edbMappings[0].EdbInfoId
  956. if indexId <= 0 {
  957. err = fmt.Errorf("图表标的有误")
  958. return
  959. }
  960. indexOb := new(aiPredictModel.AiPredictModelIndex)
  961. indexItem, e := indexOb.GetItemById(indexId)
  962. if e != nil {
  963. err = fmt.Errorf("获取图表标的失败, %v", e)
  964. return
  965. }
  966. if indexItem != nil && indexItem.AiPredictModelIndexId <= 0 {
  967. err = fmt.Errorf("图表标的不存在, IndexId: %d", indexId)
  968. return
  969. }
  970. // 获取标的数据
  971. indexData := make([]*aiPredictModel.AiPredictModelData, 0)
  972. dataSource := aiPredictModel.ModelDataSourceDaily
  973. if chartInfo.Source == utils.CHART_SOURCE_AI_PREDICT_MODEL_MONTHLY {
  974. dataSource = aiPredictModel.ModelDataSourceMonthly
  975. }
  976. dataOb := new(aiPredictModel.AiPredictModelData)
  977. dataCond := fmt.Sprintf(` AND %s = ?`, dataOb.Cols().IndexCode)
  978. dataPars := make([]interface{}, 0)
  979. dataPars = append(dataPars, indexItem.IndexCode)
  980. list, e := dataOb.GetItemsByCondition(dataCond, dataPars, []string{}, fmt.Sprintf("%s DESC", dataOb.Cols().DataTime))
  981. if e != nil {
  982. err = fmt.Errorf("获取标的数据失败, %v", e)
  983. return
  984. }
  985. for _, v := range list {
  986. if v.Source == dataSource {
  987. indexData = append(indexData, v)
  988. continue
  989. }
  990. }
  991. // 图表详情
  992. resp, e = GetAiPredictChartDetailByData(indexItem, indexData, dataSource)
  993. if e != nil {
  994. err = fmt.Errorf("获取图表详情失败, %v", e)
  995. return
  996. }
  997. if utils.Re == nil {
  998. jsonData, _ := json.Marshal(resp)
  999. _ = utils.Rc.Put(key, jsonData, 10*time.Minute)
  1000. }
  1001. isOk = true
  1002. return
  1003. }
  1004. func GetAiPredictChartDetailByData(indexItem *aiPredictModel.AiPredictModelIndex, indexData []*aiPredictModel.AiPredictModelData, source int) (resp *models.ChartInfoDetailResp, err error) {
  1005. resp = new(models.ChartInfoDetailResp)
  1006. // 标的配置
  1007. var extraConfig aiPredictModel.AiPredictModelIndexExtraConfig
  1008. if indexItem.ExtraConfig != "" {
  1009. if e := json.Unmarshal([]byte(indexItem.ExtraConfig), &extraConfig); e != nil {
  1010. err = fmt.Errorf("标的额外配置解析失败, Config: %s, Err: %v", indexItem.ExtraConfig, e)
  1011. return
  1012. }
  1013. }
  1014. // 图表信息
  1015. var predictLegendName, confLeftMin, confLeftMax, unit string
  1016. if source == aiPredictModel.ModelDataSourceDaily {
  1017. predictLegendName = extraConfig.DailyChart.PredictLegendName
  1018. if predictLegendName == "" {
  1019. predictLegendName = "Predicted"
  1020. }
  1021. unit = extraConfig.DailyChart.Unit
  1022. confLeftMin = extraConfig.DailyChart.LeftMin
  1023. confLeftMax = extraConfig.DailyChart.LeftMax
  1024. }
  1025. if source == aiPredictModel.ModelDataSourceMonthly {
  1026. predictLegendName = "预测值"
  1027. unit = extraConfig.MonthlyChart.Unit
  1028. confLeftMin = extraConfig.MonthlyChart.LeftMin
  1029. confLeftMax = extraConfig.MonthlyChart.LeftMax
  1030. }
  1031. // 这里简单兼容下吧,暂时就不修数据了
  1032. if confLeftMin == "" {
  1033. confLeftMin = indexItem.LeftMin
  1034. }
  1035. if confLeftMax == "" {
  1036. confLeftMax = indexItem.LeftMax
  1037. }
  1038. // 获取指标对应的图表
  1039. chartSourceMapping := map[int]int{
  1040. aiPredictModel.ModelDataSourceMonthly: utils.CHART_SOURCE_AI_PREDICT_MODEL_MONTHLY,
  1041. aiPredictModel.ModelDataSourceDaily: utils.CHART_SOURCE_AI_PREDICT_MODEL_DAILY,
  1042. }
  1043. chartInfo, e := data_manage.GetAiPredictChartInfoByIndexId(chartSourceMapping[source], indexItem.AiPredictModelIndexId)
  1044. if e != nil && !utils.IsErrNoRow(e) {
  1045. err = fmt.Errorf("获取标的图表失败, %v", e)
  1046. return
  1047. }
  1048. // 获取曲线图主题样式
  1049. chartView := new(models.ChartInfo)
  1050. if chartInfo != nil && chartInfo.ChartInfoId > 0 {
  1051. chartView.ChartInfoId = chartInfo.ChartInfoId
  1052. chartView.ChartName = chartInfo.ChartName
  1053. chartView.ChartNameEn = chartInfo.ChartNameEn
  1054. chartView.Source = chartInfo.Source
  1055. chartView.ChartImage = chartInfo.ChartImage
  1056. } else {
  1057. chartView.ChartName = indexItem.IndexName
  1058. chartView.ChartNameEn = indexItem.IndexName
  1059. }
  1060. chartView.ChartType = utils.CHART_SOURCE_DEFAULT
  1061. chartTheme, e := data.GetChartThemeConfig(0, chartView.ChartType, utils.CHART_TYPE_CURVE)
  1062. if e != nil {
  1063. err = fmt.Errorf("获取图表主题样式失败, %v", e)
  1064. return
  1065. }
  1066. chartView.ChartThemeStyle = chartTheme.Config
  1067. chartView.ChartThemeId = chartTheme.ChartThemeId
  1068. chartView.DateType = 3
  1069. chartView.Calendar = "公历"
  1070. chartView.ChartSource = "AI预测模型"
  1071. chartView.ChartSourceEn = "AI预测模型"
  1072. chartView.Unit = unit
  1073. chartView.UnitEn = unit
  1074. // EdbList-固定一条为标的实际值、一条为预测值
  1075. edbList := make([]*models.ChartEdbInfoMapping, 0)
  1076. edbActual, edbPredict := new(models.ChartEdbInfoMapping), new(models.ChartEdbInfoMapping)
  1077. edbActual.EdbName = indexItem.IndexName
  1078. edbActual.EdbNameEn = indexItem.IndexName
  1079. edbActual.IsAxis = 1
  1080. edbActual.Unit = unit
  1081. edbActual.UnitEn = unit
  1082. edbPredict.EdbName = predictLegendName
  1083. edbPredict.EdbNameEn = predictLegendName
  1084. edbPredict.IsAxis = 1
  1085. edbPredict.Unit = unit
  1086. edbPredict.UnitEn = unit
  1087. actualData, predictData := make([]*models.EdbDataList, 0), make([]*models.EdbDataList, 0)
  1088. var startDate, endDate time.Time
  1089. var actualValues, predictValues []float64
  1090. var actualNewest, predictNewest bool
  1091. var actualLatestTimestamp int64 // 实际值最后一天的时间戳,作为日度图表的分割线
  1092. for k, v := range indexData {
  1093. // 如果实际值和预测值都是null那么该日期无效直接忽略
  1094. if !v.Value.Valid && !v.PredictValue.Valid {
  1095. continue
  1096. }
  1097. // 将有效值加入[]float64,最后取极值
  1098. if v.Value.Valid {
  1099. actualValues = append(actualValues, v.Value.Float64)
  1100. }
  1101. if v.PredictValue.Valid {
  1102. predictValues = append(predictValues, v.PredictValue.Float64)
  1103. }
  1104. // 开始结束时间
  1105. if k == 0 {
  1106. startDate = v.DataTime
  1107. endDate = v.CreateTime
  1108. }
  1109. if v.DataTime.Before(startDate) {
  1110. startDate = v.DataTime
  1111. }
  1112. if v.DataTime.After(endDate) {
  1113. endDate = v.DataTime
  1114. }
  1115. // 指标数据
  1116. if v.Value.Valid {
  1117. if !actualNewest {
  1118. edbActual.LatestDate = v.DataTime.Format(utils.FormatDate)
  1119. edbActual.LatestValue = v.Value.Float64
  1120. actualLatestTimestamp = v.DataTime.UnixNano() / 1e6
  1121. actualNewest = true
  1122. }
  1123. actualData = append(actualData, &models.EdbDataList{
  1124. DataTime: v.DataTime.Format(utils.FormatDate),
  1125. Value: v.Value.Float64,
  1126. DataTimestamp: v.DataTimestamp,
  1127. })
  1128. }
  1129. if v.PredictValue.Valid {
  1130. if !predictNewest {
  1131. edbPredict.LatestDate = v.DataTime.Format(utils.FormatDate)
  1132. edbPredict.LatestValue = v.Value.Float64
  1133. predictNewest = true
  1134. }
  1135. predictData = append(predictData, &models.EdbDataList{
  1136. DataTime: v.DataTime.Format(utils.FormatDate),
  1137. Value: v.PredictValue.Float64,
  1138. DataTimestamp: v.DataTimestamp,
  1139. })
  1140. }
  1141. }
  1142. // 图表数据这里均做一个升序排序
  1143. sort.Slice(actualData, func(i, j int) bool {
  1144. return actualData[i].DataTimestamp < actualData[j].DataTimestamp
  1145. })
  1146. sort.Slice(predictData, func(i, j int) bool {
  1147. return predictData[i].DataTimestamp < predictData[j].DataTimestamp
  1148. })
  1149. // 极值
  1150. actualMin, actualMax := utils.FindMinMax(actualValues)
  1151. predictMin, predictMax := utils.FindMinMax(predictValues)
  1152. edbActual.MinData = actualMin
  1153. edbActual.MaxData = actualMax
  1154. edbPredict.MinData = predictMin
  1155. edbPredict.MaxData = predictMax
  1156. edbActual.DataList = actualData
  1157. edbPredict.DataList = predictData
  1158. edbList = append(edbList, edbActual, edbPredict)
  1159. // 上下限
  1160. if confLeftMin != "" {
  1161. chartView.LeftMin = confLeftMin
  1162. } else {
  1163. leftMin := actualMin
  1164. if leftMin > predictMin {
  1165. leftMin = predictMin
  1166. }
  1167. chartView.LeftMin = fmt.Sprint(leftMin)
  1168. }
  1169. if confLeftMax != "" {
  1170. chartView.LeftMax = confLeftMax
  1171. } else {
  1172. leftMax := actualMax
  1173. if leftMax < predictMax {
  1174. leftMax = predictMax
  1175. }
  1176. chartView.LeftMax = fmt.Sprint(leftMax)
  1177. }
  1178. chartView.StartDate = startDate.Format(utils.FormatDate)
  1179. chartView.EndDate = endDate.Format(utils.FormatDate)
  1180. // 日度图表的分割线日期
  1181. if source == aiPredictModel.ModelDataSourceDaily {
  1182. var dataResp struct {
  1183. ActualLatestTimestamp int64
  1184. }
  1185. dataResp.ActualLatestTimestamp = actualLatestTimestamp
  1186. resp.DataResp = dataResp
  1187. }
  1188. resp.ChartInfo = chartView
  1189. resp.EdbInfoList = edbList
  1190. return
  1191. }