chart_common.go 43 KB

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