chart_common.go 43 KB

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