chart_common.go 43 KB

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