chart_info.go 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066
  1. package future_good
  2. import (
  3. "errors"
  4. "eta/eta_chart_lib/models"
  5. "eta/eta_chart_lib/models/data_manage"
  6. future_good2 "eta/eta_chart_lib/models/data_manage/future_good"
  7. "eta/eta_chart_lib/services/alarm_msg"
  8. "eta/eta_chart_lib/services/data"
  9. "eta/eta_chart_lib/utils"
  10. "fmt"
  11. "github.com/shopspring/decimal"
  12. "sort"
  13. "strconv"
  14. "time"
  15. )
  16. // GetChartEdbData 获取图表的指标数据
  17. func GetChartEdbData(chartInfoId int, startDate, endDate string, baseEdbInfoMapping *models.ChartEdbInfoMapping, edbInfoMappingList []*models.ChartEdbInfoMapping, futureGoodEdbInfoMapping *models.ChartEdbInfoMapping, barChartInfoConf data_manage.FutureGoodBarChartInfoReq, needData bool) (barConfigEdbInfoIdList []data_manage.BarChartInfoEdbItemReq, edbList []*models.ChartEdbInfoMapping, xEdbIdValue []int, xDataList []data_manage.XData, yDataList []models.YData, err error) {
  18. edbList = make([]*models.ChartEdbInfoMapping, 0)
  19. barChartInfoDateList := barChartInfoConf.DateList
  20. baseEdbInfoId := barChartInfoConf.BaseEdbInfoId
  21. if futureGoodEdbInfoMapping == nil {
  22. err = errors.New("商品指标未选取")
  23. return
  24. }
  25. if len(edbInfoMappingList) == 0 {
  26. err = errors.New("ETA指标未选取")
  27. return
  28. }
  29. // 指标对应的所有数据
  30. edbDataListMap := make(map[int][]*models.EdbDataList)
  31. // todo item
  32. //item := new(models.ChartEdbInfoMapping)
  33. if futureGoodEdbInfoMapping.Unit == `无` {
  34. futureGoodEdbInfoMapping.Unit = ``
  35. }
  36. if chartInfoId <= 0 {
  37. for k, edbInfoMapping := range edbInfoMappingList {
  38. edbInfoMapping.FrequencyEn = data.GetFrequencyEn(edbInfoMapping.Frequency)
  39. if edbInfoMapping.Unit == `无` {
  40. edbInfoMapping.Unit = ``
  41. }
  42. edbInfoMapping.IsAxis = 1
  43. edbInfoMapping.LeadValue = 0
  44. edbInfoMapping.LeadUnit = ""
  45. edbInfoMapping.ChartEdbMappingId = 0
  46. edbInfoMapping.ChartInfoId = 0
  47. edbInfoMapping.IsOrder = false
  48. edbInfoMapping.EdbInfoType = 1
  49. edbInfoMapping.ChartStyle = ""
  50. edbInfoMapping.ChartColor = ""
  51. edbInfoMapping.ChartWidth = 0
  52. edbInfoMappingList[k] = edbInfoMapping
  53. }
  54. futureGoodEdbInfoMapping.IsAxis = 1
  55. futureGoodEdbInfoMapping.LeadValue = 0
  56. futureGoodEdbInfoMapping.LeadUnit = ""
  57. futureGoodEdbInfoMapping.ChartEdbMappingId = 0
  58. futureGoodEdbInfoMapping.ChartInfoId = 0
  59. futureGoodEdbInfoMapping.IsOrder = false
  60. futureGoodEdbInfoMapping.EdbInfoType = 1
  61. futureGoodEdbInfoMapping.ChartStyle = ""
  62. futureGoodEdbInfoMapping.ChartColor = ""
  63. futureGoodEdbInfoMapping.ChartWidth = 0
  64. } else {
  65. for k, edbInfoMapping := range edbInfoMappingList {
  66. edbInfoMapping.FrequencyEn = data.GetFrequencyEn(edbInfoMapping.Frequency)
  67. if edbInfoMapping.Unit == `无` {
  68. edbInfoMapping.Unit = ``
  69. }
  70. edbInfoMapping.LeadUnitEn = data.GetLeadUnitEn(edbInfoMapping.LeadUnit)
  71. edbInfoMappingList[k] = edbInfoMapping
  72. }
  73. futureGoodEdbInfoMapping.LeadUnitEn = data.GetLeadUnitEn(futureGoodEdbInfoMapping.LeadUnit)
  74. }
  75. //查询横轴配置项
  76. xDataItemMap := make(map[int]data_manage.XData)
  77. for k, v := range barChartInfoConf.XDataList {
  78. xDataItemMap[k] = v
  79. }
  80. // 普通的指标数据
  81. {
  82. edbList = edbInfoMappingList
  83. for k, edbInfoMapping := range edbInfoMappingList {
  84. tmp := data_manage.BarChartInfoEdbItemReq{
  85. EdbInfoId: edbInfoMapping.EdbInfoId,
  86. Name: edbInfoMapping.EdbName,
  87. NameEn: edbInfoMapping.EdbNameEn,
  88. Source: edbInfoMapping.Source,
  89. }
  90. // 如果有配置,则用配置中的指标名称替换
  91. xItem, ok := xDataItemMap[k]
  92. if ok && xItem.Name != "" {
  93. tmp.Name = xItem.Name
  94. tmp.NameEn = xItem.NameEn
  95. tmp.IsHide = xItem.IsHide
  96. }
  97. barConfigEdbInfoIdList = append(barConfigEdbInfoIdList, tmp)
  98. }
  99. }
  100. // 获取主力合约和最新日期
  101. var zlFutureGoodEdbInfo *future_good2.FutureGoodEdbInfo
  102. var latestDate string // 最新日期是这个
  103. var regionType string // 交易所来源:“国内”,“海外”
  104. var futureGoodEdbType int
  105. {
  106. // 寻找主力合约
  107. zlFutureGoodEdbInfo, err = future_good2.GetFutureGoodEdbInfo(futureGoodEdbInfoMapping.EdbInfoId)
  108. if err != nil {
  109. return
  110. }
  111. regionType = zlFutureGoodEdbInfo.RegionType
  112. futureGoodEdbType = zlFutureGoodEdbInfo.FutureGoodEdbType
  113. // 如果当前合约配置的 "画图时,日期来源的指标id" 并不是自己的话,那么就去查找对应配置的合约
  114. if zlFutureGoodEdbInfo.DateSourceId != zlFutureGoodEdbInfo.FutureGoodEdbInfoId {
  115. sourceDateFutureGoodEdbInfo, tmpErr := future_good2.GetFutureGoodEdbInfo(zlFutureGoodEdbInfo.DateSourceId)
  116. if tmpErr != nil {
  117. err = tmpErr
  118. return
  119. }
  120. latestDate = sourceDateFutureGoodEdbInfo.EndDate // 最新日期是这个
  121. } else {
  122. latestDate = zlFutureGoodEdbInfo.EndDate // 最新日期是这个
  123. }
  124. if latestDate == `0000-00-00` {
  125. err = errors.New("日期异常")
  126. return
  127. }
  128. }
  129. // 获取期货指标以及期货数据
  130. tmpFutureGoodEdbInfoList, err := future_good2.GetFutureGoodEdbInfoListByParentId(futureGoodEdbInfoMapping.EdbInfoId)
  131. if err != nil {
  132. return
  133. }
  134. latestDateTime, _ := time.ParseInLocation(utils.FormatDate, latestDate, time.Local)
  135. _, futureGoodEdbInfoList, err := getFutureGoodEdbInfoList(latestDateTime, tmpFutureGoodEdbInfoList, barChartInfoDateList)
  136. if err != nil {
  137. return
  138. }
  139. futureGoodMappingList := make([]*models.ChartEdbInfoMapping, 0)
  140. for k, v := range futureGoodEdbInfoList {
  141. newMappingInfo := &models.ChartEdbInfoMapping{
  142. EdbInfoId: v.FutureGoodEdbInfoId,
  143. SourceName: v.Exchange,
  144. Source: 0,
  145. EdbCode: v.FutureGoodEdbCode,
  146. EdbName: v.FutureGoodEdbName,
  147. EdbAliasName: v.FutureGoodEdbName,
  148. EdbNameEn: v.FutureGoodEdbNameEn,
  149. EdbType: baseEdbInfoMapping.EdbType, //todo baseEdbInfoMapping
  150. Frequency: baseEdbInfoMapping.Frequency,
  151. FrequencyEn: baseEdbInfoMapping.FrequencyEn,
  152. Unit: baseEdbInfoMapping.Unit,
  153. UnitEn: baseEdbInfoMapping.UnitEn,
  154. StartDate: v.StartDate,
  155. EndDate: v.EndDate,
  156. ModifyTime: v.ModifyTime.Format(utils.FormatDateTime),
  157. ChartEdbMappingId: v.FutureGoodEdbInfoId,
  158. ChartInfoId: baseEdbInfoMapping.ChartInfoId,
  159. MaxData: v.MaxValue,
  160. MinData: v.MinValue,
  161. IsOrder: baseEdbInfoMapping.IsOrder,
  162. IsAxis: baseEdbInfoMapping.IsAxis,
  163. EdbInfoType: baseEdbInfoMapping.EdbInfoType,
  164. EdbInfoCategoryType: baseEdbInfoMapping.EdbInfoCategoryType,
  165. LeadValue: baseEdbInfoMapping.LeadValue,
  166. LeadUnit: baseEdbInfoMapping.LeadUnit,
  167. LeadUnitEn: baseEdbInfoMapping.LeadUnitEn,
  168. ChartStyle: baseEdbInfoMapping.ChartStyle,
  169. ChartColor: baseEdbInfoMapping.ChartColor,
  170. PredictChartColor: baseEdbInfoMapping.PredictChartColor,
  171. ChartWidth: baseEdbInfoMapping.ChartWidth,
  172. ChartType: baseEdbInfoMapping.ChartType,
  173. LatestDate: v.LatestDate.Format(utils.FormatDateTime),
  174. LatestValue: v.LatestValue,
  175. UniqueCode: futureGoodEdbInfoMapping.UniqueCode + strconv.Itoa(k),
  176. MinValue: v.MinValue,
  177. MaxValue: v.MaxValue,
  178. DataList: nil,
  179. IsNullData: false,
  180. }
  181. futureGoodMappingList = append(futureGoodMappingList, newMappingInfo)
  182. tmp := data_manage.BarChartInfoEdbItemReq{
  183. EdbInfoId: newMappingInfo.EdbInfoId,
  184. Name: fmt.Sprint("M+", v.Month),
  185. NameEn: fmt.Sprint("M+", v.Month),
  186. Source: newMappingInfo.Source,
  187. }
  188. // 判断如果有配置,需要替换为配置内容
  189. xItem, ok := xDataItemMap[k]
  190. if ok && xItem.Name != "" {
  191. tmp.Name = xItem.Name
  192. tmp.NameEn = xItem.NameEn
  193. tmp.IsHide = xItem.IsHide
  194. }
  195. barConfigEdbInfoIdList = append(barConfigEdbInfoIdList, tmp)
  196. }
  197. if regionType == `海外` {
  198. zlFutureGoodEdbMapping := &models.ChartEdbInfoMapping{
  199. EdbInfoId: zlFutureGoodEdbInfo.FutureGoodEdbInfoId,
  200. SourceName: zlFutureGoodEdbInfo.Exchange,
  201. Source: 0,
  202. EdbCode: zlFutureGoodEdbInfo.FutureGoodEdbCode,
  203. EdbName: zlFutureGoodEdbInfo.FutureGoodEdbName,
  204. EdbAliasName: zlFutureGoodEdbInfo.FutureGoodEdbName,
  205. EdbNameEn: zlFutureGoodEdbInfo.FutureGoodEdbNameEn,
  206. EdbType: baseEdbInfoMapping.EdbType,
  207. Frequency: baseEdbInfoMapping.Frequency,
  208. FrequencyEn: baseEdbInfoMapping.FrequencyEn,
  209. Unit: baseEdbInfoMapping.Unit,
  210. UnitEn: baseEdbInfoMapping.UnitEn,
  211. StartDate: zlFutureGoodEdbInfo.StartDate,
  212. EndDate: zlFutureGoodEdbInfo.EndDate,
  213. ModifyTime: zlFutureGoodEdbInfo.ModifyTime.Format(utils.FormatDateTime),
  214. ChartEdbMappingId: zlFutureGoodEdbInfo.FutureGoodEdbInfoId,
  215. ChartInfoId: baseEdbInfoMapping.ChartInfoId,
  216. MaxData: zlFutureGoodEdbInfo.MaxValue,
  217. MinData: zlFutureGoodEdbInfo.MinValue,
  218. IsOrder: baseEdbInfoMapping.IsOrder,
  219. IsAxis: baseEdbInfoMapping.IsAxis,
  220. EdbInfoType: baseEdbInfoMapping.EdbInfoType,
  221. EdbInfoCategoryType: baseEdbInfoMapping.EdbInfoCategoryType,
  222. LeadValue: baseEdbInfoMapping.LeadValue,
  223. LeadUnit: baseEdbInfoMapping.LeadUnit,
  224. LeadUnitEn: baseEdbInfoMapping.LeadUnitEn,
  225. ChartStyle: baseEdbInfoMapping.ChartStyle,
  226. ChartColor: baseEdbInfoMapping.ChartColor,
  227. PredictChartColor: baseEdbInfoMapping.PredictChartColor,
  228. ChartWidth: baseEdbInfoMapping.ChartWidth,
  229. ChartType: baseEdbInfoMapping.ChartType,
  230. LatestDate: zlFutureGoodEdbInfo.LatestDate.Format(utils.FormatDateTime),
  231. LatestValue: zlFutureGoodEdbInfo.LatestValue,
  232. UniqueCode: futureGoodEdbInfoMapping.UniqueCode + strconv.Itoa(0),
  233. MinValue: zlFutureGoodEdbInfo.MinValue,
  234. MaxValue: zlFutureGoodEdbInfo.MaxValue,
  235. DataList: nil,
  236. IsNullData: false,
  237. }
  238. edbList = append(edbList, zlFutureGoodEdbMapping)
  239. edbList = append(edbList, futureGoodMappingList...)
  240. } else {
  241. edbList = append(edbList, futureGoodMappingList...)
  242. }
  243. if !needData {
  244. return
  245. }
  246. // 获取数据
  247. // 现货数据
  248. for _, edbInfoMapping := range edbInfoMappingList {
  249. dataList := make([]*models.EdbDataList, 0)
  250. dataList, err = models.GetEdbDataList(edbInfoMapping.Source, edbInfoMapping.SubSource, edbInfoMapping.EdbInfoId, startDate, endDate)
  251. if err != nil {
  252. return
  253. }
  254. edbDataListMap[edbInfoMapping.EdbInfoId] = dataList
  255. // todo item
  256. //item.DataList = dataList
  257. }
  258. // 期货数据
  259. for _, v := range futureGoodMappingList {
  260. dataList := make([]*models.EdbDataList, 0)
  261. tmpDataList, tmpErr := future_good2.GetFutureGoodEdbDataListByDate(v.EdbInfoId, startDate, endDate)
  262. if tmpErr != nil {
  263. return
  264. }
  265. for _, tmpData := range tmpDataList {
  266. dataList = append(dataList, &models.EdbDataList{
  267. EdbDataId: tmpData.FutureGoodEdbDataId,
  268. EdbInfoId: tmpData.FutureGoodEdbInfoId,
  269. DataTime: tmpData.DataTime.Format(utils.FormatDate),
  270. DataTimestamp: tmpData.DataTimestamp,
  271. Value: tmpData.Close,
  272. })
  273. }
  274. edbDataListMap[v.EdbInfoId] = dataList
  275. v.DataList = dataList
  276. }
  277. xEdbIdValue, yDataList, err = BarChartData(baseEdbInfoMapping, edbInfoMappingList, futureGoodEdbInfoList, edbDataListMap, barChartInfoDateList, regionType, baseEdbInfoMapping.EndDate)
  278. if len(barChartInfoConf.XDataList) > 0 {
  279. xDataList = barChartInfoConf.XDataList
  280. } else {
  281. for _, v := range edbInfoMappingList {
  282. xDataList = append(xDataList, data_manage.XData{
  283. Name: v.EdbName,
  284. NameEn: v.EdbNameEn,
  285. IsHide: 0,
  286. })
  287. }
  288. }
  289. //yDataList =make([]models.YData,0)
  290. //models.YData{
  291. // Date: "",
  292. // Color: "",
  293. // Name: "",
  294. // NameEn: "",
  295. // Value: nil,
  296. // NoDataEdbList: nil,
  297. // XEdbInfoIdList: nil,
  298. //}
  299. futureGoodEdbInfoIndexMap := make(map[int]int)
  300. for index, v := range futureGoodEdbInfoList {
  301. futureGoodEdbInfoIndexMap[v.FutureGoodEdbInfoId] = index
  302. }
  303. nMap := make(map[int]int)
  304. maxIndex := 0 // 最大x轴的下标
  305. for k, tmpYData := range yDataList {
  306. yDataMap := tmpYData.EdbValMap
  307. edbInfoIdList := make([]int, 0)
  308. noDataEdbInfoIdList := make([]int, 0)
  309. valueList := make([]float64, 0)
  310. noDataEdbIdMap := make(map[int]int)
  311. // 所有的N值
  312. for _, n := range tmpYData.M {
  313. nMap[n] = n
  314. }
  315. // 基础指标
  316. {
  317. edbInfoIdList = append(edbInfoIdList, baseEdbInfoId)
  318. tmpVal, ok := yDataMap[baseEdbInfoId]
  319. valueList = append(valueList, tmpVal)
  320. if !ok || tmpVal == 0 {
  321. noDataEdbInfoIdList = append(noDataEdbInfoIdList, baseEdbInfoId)
  322. noDataEdbIdMap[baseEdbInfoId] = baseEdbInfoId
  323. }
  324. }
  325. for _, futureGoodEdbInfo := range futureGoodEdbInfoList {
  326. tmpEdbInfId := futureGoodEdbInfo.FutureGoodEdbInfoId
  327. edbInfoIdList = append(edbInfoIdList, tmpEdbInfId)
  328. tmpVal, ok := yDataMap[tmpEdbInfId]
  329. valueList = append(valueList, tmpVal)
  330. if !ok || tmpVal == 0 {
  331. noDataEdbInfoIdList = append(noDataEdbInfoIdList, tmpEdbInfId)
  332. noDataEdbIdMap[tmpEdbInfId] = tmpEdbInfId
  333. }
  334. }
  335. //tmpYData.Value = valueList
  336. tmpYData.NoDataEdbList = noDataEdbInfoIdList
  337. yDataList[k] = tmpYData
  338. lenEdbId := len(edbInfoIdList)
  339. tmpMaxIndex := lenEdbId - 1 // 当前数据的最大x轴的下标
  340. for i := lenEdbId - 1; i >= 0; i-- {
  341. // 如果没有在无数据的指标列表中找到,那么就找到了最大x轴的下标
  342. if _, ok := noDataEdbIdMap[edbInfoIdList[i]]; !ok {
  343. // 如果最大x轴的下标 小于 当前下标,那么就重新赋值
  344. if maxIndex < i-1 {
  345. maxIndex = i - 1
  346. }
  347. break
  348. }
  349. tmpMaxIndex = i - 1
  350. }
  351. // 如果最大x轴的下标 小于 当前下标,那么就重新赋值
  352. if maxIndex < tmpMaxIndex {
  353. maxIndex = tmpMaxIndex
  354. }
  355. }
  356. //xEdbIdValue = xEdbIdValue[0:maxIndex]
  357. // 找出所有的N值,并进行正序排列
  358. nList := make([]int, 0)
  359. for _, n := range nMap {
  360. nList = append(nList, n)
  361. }
  362. sort.Slice(nList, func(i, j int) bool {
  363. return nList[i] < nList[j]
  364. })
  365. for k, v := range yDataList {
  366. if len(v.XEdbInfoIdList) >= maxIndex+1 {
  367. yDataList[k].XEdbInfoIdList = v.XEdbInfoIdList[0 : maxIndex+1]
  368. }
  369. if len(v.Value) >= maxIndex+1 {
  370. yDataList[k].Value = v.Value[0 : maxIndex+1]
  371. }
  372. }
  373. baseEdbLen := len(edbInfoMappingList)
  374. tmpXDataList, newYDataList, err := handleResultData(regionType, futureGoodEdbType, baseEdbLen, yDataList, futureGoodEdbInfoList, maxIndex)
  375. if err != nil {
  376. return
  377. }
  378. if len(barChartInfoConf.XDataList) == 0 {
  379. xDataList = append(xDataList, tmpXDataList...)
  380. }
  381. yDataList = newYDataList
  382. return
  383. }
  384. // BarChartData 获取数据
  385. func BarChartData(baseEdbInfoMapping *models.ChartEdbInfoMapping, edbInfoMappingList []*models.ChartEdbInfoMapping, futureGoodMappingList []*future_good2.FutureGoodEdbInfo, edbDataListMap map[int][]*models.EdbDataList, barChartInfoDateList []data_manage.BarChartInfoDateReq, regionType, latestDate string) (edbIdList []int, yDataList []models.YData, err error) {
  386. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  387. // 现货指标数据map
  388. baseEdbDataMap := make(map[int]map[string]float64)
  389. edbInfoMappingMap := make(map[int]struct{})
  390. for _, v := range edbInfoMappingList {
  391. edbInfoMappingMap[v.EdbInfoId] = struct{}{}
  392. }
  393. for edbInfoId, edbDataList := range edbDataListMap {
  394. if _, ok1 := edbInfoMappingMap[edbInfoId]; ok1 {
  395. edbDateData := make(map[string]float64)
  396. for _, edbData := range edbDataList {
  397. edbDateData[edbData.DataTime] = edbData.Value
  398. }
  399. baseEdbDataMap[edbInfoId] = edbDateData
  400. }
  401. }
  402. // 期货指标数据map
  403. futureGoodEdbDataMap := make(map[int]map[string]float64)
  404. for edbInfoId, edbDataList := range edbDataListMap {
  405. if _, ok := edbInfoMappingMap[edbInfoId]; ok {
  406. continue
  407. }
  408. edbDateData := make(map[string]float64)
  409. for _, edbData := range edbDataList {
  410. edbDateData[edbData.DataTime] = edbData.Value
  411. }
  412. futureGoodEdbDataMap[edbInfoId] = edbDateData
  413. }
  414. // edbIdList 指标展示顺序;x轴的指标顺序
  415. edbIdList = make([]int, 0)
  416. for _, v := range edbInfoMappingList {
  417. edbIdList = append(edbIdList, v.EdbInfoId)
  418. }
  419. for _, v := range futureGoodMappingList {
  420. edbIdList = append(edbIdList, v.FutureGoodEdbInfoId)
  421. }
  422. latestDateTime, _ := time.ParseInLocation(utils.FormatDate, latestDate, time.Local)
  423. yDataList = make([]models.YData, 0) //y轴的数据列表
  424. for _, barChartInfoDate := range barChartInfoDateList {
  425. yDataMap := make(map[int]float64)
  426. var maxDate time.Time
  427. var findDateTime time.Time
  428. switch barChartInfoDate.Type {
  429. case 1: //最新值
  430. findDateTime = latestDateTime
  431. case 2: //近期几天
  432. findDateTime = latestDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  433. case 3: // 固定日期
  434. //寻找固定日期的数据
  435. tmpFindDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  436. if tmpErr != nil {
  437. err = tmpErr
  438. return
  439. }
  440. findDateTime = tmpFindDateTime
  441. default:
  442. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  443. return
  444. }
  445. if findDateTime.IsZero() {
  446. err = errors.New("错误的日期")
  447. return
  448. }
  449. findDataList := make([]float64, 0) // 当前日期的数据值
  450. noDataIdList := make([]int, 0) // 没有数据的指标id
  451. noDataIdMap := make(map[int]int, 0) // 没有数据的指标map
  452. xEdbInfoIdList := make([]int, 0) // 当前数据的指标id列表
  453. // 先找到基准日期
  454. var realDateTime time.Time
  455. realDateTime, findDataValue, isFind, tmpErr := GetNeedDateData(findDateTime, edbDataListMap[baseEdbInfoMapping.EdbInfoId], baseEdbDataMap[baseEdbInfoMapping.EdbInfoId], futureGoodEdbDataMap)
  456. if tmpErr != nil {
  457. err = tmpErr
  458. return
  459. }
  460. // 处理其余现货指标
  461. for _, v := range edbInfoMappingList {
  462. if v.EdbInfoId == baseEdbInfoMapping.EdbInfoId {
  463. findDataList = append(findDataList, findDataValue)
  464. yDataMap[baseEdbInfoMapping.EdbInfoId] = findDataValue
  465. if isFind {
  466. maxDate = realDateTime
  467. } else {
  468. noDataIdList = append(noDataIdList, baseEdbInfoMapping.EdbInfoId)
  469. noDataIdMap[baseEdbInfoMapping.EdbInfoId] = baseEdbInfoMapping.EdbInfoId
  470. }
  471. xEdbInfoIdList = append(xEdbInfoIdList, v.EdbInfoId)
  472. continue
  473. }
  474. tmpRealDateTime, findDataValueTmp, isFindTmp, e := GetNeedDateData(realDateTime, edbDataListMap[v.EdbInfoId], baseEdbDataMap[v.EdbInfoId], futureGoodEdbDataMap)
  475. if e != nil {
  476. err = e
  477. return
  478. }
  479. findDataList = append(findDataList, findDataValueTmp)
  480. yDataMap[v.EdbInfoId] = findDataValue
  481. if isFindTmp {
  482. // todo 是否需要变更 maxDate
  483. if maxDate.IsZero() || maxDate.Before(tmpRealDateTime) {
  484. maxDate = tmpRealDateTime
  485. }
  486. } else {
  487. noDataIdList = append(noDataIdList, v.EdbInfoId)
  488. noDataIdMap[v.EdbInfoId] = v.EdbInfoId
  489. }
  490. xEdbInfoIdList = append(xEdbInfoIdList, v.EdbInfoId)
  491. }
  492. //currMonth := findDateTime.Month() // 当前月份
  493. //currYear := findDateTime.Year() // 当前年份
  494. // 用实际日期的月份作为基准,往前推12个月(2024-5-13 16:26:43修改)
  495. currMonth := realDateTime.Month() // 当前月份
  496. currYear := realDateTime.Year() // 当前年份
  497. mList := make([]int, 0) // 间隔月份
  498. indexList := make([]int, 0)
  499. if regionType == `国内` {
  500. for i := currMonth + 1; i <= 12; i++ {
  501. indexList = append(indexList, int(i))
  502. mList = append(mList, int(i-currMonth))
  503. }
  504. for i := 1; i < int(currMonth); i++ {
  505. indexList = append(indexList, i)
  506. mList = append(mList, 12+i-int(currMonth))
  507. }
  508. } else {
  509. for i, v := range futureGoodMappingList {
  510. if v.FutureGoodEdbType == 2 {
  511. if i == 0 {
  512. continue
  513. }
  514. indexList = append(indexList, i)
  515. mList = append(mList, v.Month)
  516. } else {
  517. if v.Year > currYear || (v.Year == currYear && v.Month > int(currMonth)) {
  518. indexList = append(indexList, i)
  519. mList = append(mList, (v.Year-currYear)*12+v.Month-int(currMonth))
  520. }
  521. }
  522. }
  523. }
  524. for _, i := range indexList {
  525. futureGoodMapping := futureGoodMappingList[i] // 当前的期货指标
  526. //tmpRealDateTime, tmpFindDataValue, tmpIsFind, tmpErr := GetNeedDateData(findDateTime, edbDataListMap[futureGoodMapping.FutureGoodEdbInfoId], edbDataMap[futureGoodMapping.FutureGoodEdbInfoId])
  527. //if tmpErr != nil {
  528. // err = tmpErr
  529. // return
  530. //}
  531. //tmpRealDateTime := findDateTime // 按照配置找到的日期
  532. tmpRealDateTime := realDateTime // 实际现货的日期
  533. tmpFindDataValue, tmpIsFind := futureGoodEdbDataMap[futureGoodMapping.FutureGoodEdbInfoId][tmpRealDateTime.Format(utils.FormatDate)]
  534. yDataMap[futureGoodMapping.FutureGoodEdbInfoId] = tmpFindDataValue
  535. findDataList = append(findDataList, tmpFindDataValue)
  536. if tmpIsFind {
  537. if maxDate.IsZero() || maxDate.Before(tmpRealDateTime) {
  538. maxDate = tmpRealDateTime
  539. }
  540. } else {
  541. noDataIdList = append(noDataIdList, futureGoodMapping.FutureGoodEdbInfoId)
  542. noDataIdMap[futureGoodMapping.FutureGoodEdbInfoId] = futureGoodMapping.FutureGoodEdbInfoId
  543. }
  544. // 当前期货合约的指标
  545. xEdbInfoIdList = append(xEdbInfoIdList, futureGoodMapping.FutureGoodEdbInfoId)
  546. }
  547. yName := barChartInfoDate.Name
  548. yNameEn := barChartInfoDate.Name
  549. if yName == `` {
  550. if barChartInfoDate.Type == 2 {
  551. yName = strconv.Itoa(barChartInfoDate.Value) + "天前"
  552. if barChartInfoDate.Value == 1 {
  553. yNameEn = strconv.Itoa(barChartInfoDate.Value) + "day ago"
  554. } else {
  555. yNameEn = strconv.Itoa(barChartInfoDate.Value) + " days ago"
  556. }
  557. } else {
  558. yName = maxDate.Format(utils.FormatDate)
  559. yNameEn = maxDate.Format(utils.FormatDate)
  560. }
  561. }
  562. yDate := "0000-00-00"
  563. if !maxDate.IsZero() {
  564. yDate = maxDate.Format(utils.FormatDate)
  565. }
  566. // 数据处理,将没有数据的下标,赋值平均值
  567. {
  568. hasDataIndexList := make([]int, 0)
  569. for dataK, edbInfoId := range xEdbInfoIdList {
  570. if _, ok := noDataIdMap[edbInfoId]; !ok { // 如果是没有数据的指标id
  571. hasDataIndexList = append(hasDataIndexList, dataK)
  572. }
  573. }
  574. lenHasDataIndex := len(hasDataIndexList)
  575. if lenHasDataIndex > 0 {
  576. for lenHasDataI := 1; lenHasDataI < lenHasDataIndex; lenHasDataI++ {
  577. perK := hasDataIndexList[lenHasDataI-1] //上一个有数据的指标下标
  578. currK := hasDataIndexList[lenHasDataI] //当前有数据的指标下标
  579. preVal := findDataList[perK] //上一个有数据的坐标的值
  580. currVal := findDataList[currK] //当前有数据的指标的值
  581. // 环差值
  582. hcValDeci := decimal.NewFromFloat(currVal).Sub(decimal.NewFromFloat(preVal)).Div(decimal.NewFromInt(int64(currK - perK)))
  583. var tmpI int64
  584. // 将两个中间的数据做平均值补全
  585. for hcI := perK + 1; hcI < currK; hcI++ {
  586. tmpI++
  587. findDataList[hcI], _ = decimal.NewFromFloat(preVal).Add(hcValDeci.Mul(decimal.NewFromInt(tmpI))).RoundCeil(4).Float64()
  588. }
  589. }
  590. }
  591. }
  592. yDataList = append(yDataList, models.YData{
  593. Date: yDate,
  594. ConfigDate: realDateTime,
  595. Value: findDataList,
  596. NoDataEdbList: noDataIdList,
  597. XEdbInfoIdList: xEdbInfoIdList,
  598. Color: barChartInfoDate.Color,
  599. Name: yName,
  600. NameEn: yNameEn,
  601. EdbValMap: yDataMap,
  602. M: mList,
  603. })
  604. }
  605. return
  606. }
  607. // handleResultData 处理成最终的结果数据
  608. func handleResultData(regionType string, futureGoodEdbType, baseEdbLen int, yDataList []models.YData, futureGoodEdbInfoList []*future_good2.FutureGoodEdbInfo, maxIndex int) (xDataList []data_manage.XData, newYDataList []models.YData, err error) {
  609. xDataList = make([]data_manage.XData, 0)
  610. newYDataList = yDataList
  611. if regionType == `国内` {
  612. for i := 1; i < 12; i++ {
  613. if i > maxIndex {
  614. break
  615. }
  616. // todo x轴 内容调整
  617. xDataList = append(xDataList, data_manage.XData{
  618. Name: fmt.Sprint("M+", i),
  619. NameEn: fmt.Sprint("M+", i),
  620. })
  621. }
  622. return
  623. }
  624. futureGoodEdbInfoMap := make(map[int]*future_good2.FutureGoodEdbInfo)
  625. for _, v := range futureGoodEdbInfoList {
  626. futureGoodEdbInfoMap[v.FutureGoodEdbInfoId] = v
  627. }
  628. if futureGoodEdbType == 2 { // FutureGoodEdbType int `description:"指标类型,1:年月是固定的合约;2:只有M+N期的合约,未固定年月"`
  629. nList := []int{3, 15, 27}
  630. for _, i := range nList {
  631. xDataList = append(xDataList, data_manage.XData{
  632. Name: fmt.Sprint("M+", i),
  633. NameEn: fmt.Sprint("M+", i),
  634. })
  635. }
  636. for yIndex, yData := range yDataList {
  637. newYDataList[yIndex].XEdbInfoIdList = []int{}
  638. newYDataList[yIndex].Value = []float64{}
  639. tmpNList := nList
  640. //当前的主力合约
  641. newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, yData.XEdbInfoIdList[0:baseEdbLen]...)
  642. newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, yData.Value[0:baseEdbLen]...)
  643. xEdbInfoIdList := yData.XEdbInfoIdList[baseEdbLen:]
  644. valIndex := baseEdbLen
  645. needNum := 0
  646. for _, n := range tmpNList {
  647. if len(xEdbInfoIdList) > 0 {
  648. edbInfoId := xEdbInfoIdList[0]
  649. futureGoodEdbInfo, ok := futureGoodEdbInfoMap[edbInfoId]
  650. if !ok {
  651. err = errors.New("找不到指标")
  652. return
  653. }
  654. // 当前距离最早的日期相差的N数
  655. if futureGoodEdbInfo.Month == n {
  656. if needNum > 0 {
  657. currVal := yData.Value[valIndex]
  658. preVal := yData.Value[valIndex-1]
  659. hcValDeci := decimal.NewFromFloat(currVal).Sub(decimal.NewFromFloat(preVal)).Div(decimal.NewFromInt(int64(needNum + 1)))
  660. for tmpNum := 0; tmpNum < needNum; tmpNum++ {
  661. newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, 0)
  662. // 赋值平均值
  663. tmpVal, _ := decimal.NewFromFloat(preVal).Add(hcValDeci.Mul(decimal.NewFromInt(int64(tmpNum + 1)))).RoundCeil(4).Float64()
  664. newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, tmpVal)
  665. }
  666. }
  667. newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, edbInfoId)
  668. newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, yData.Value[valIndex])
  669. valIndex++
  670. needNum = 0
  671. if len(xEdbInfoIdList) > 0 {
  672. xEdbInfoIdList = xEdbInfoIdList[1:]
  673. }
  674. } else {
  675. needNum++
  676. }
  677. }
  678. }
  679. }
  680. maxI := 0
  681. for _, yData := range newYDataList {
  682. lenEdb := len(yData.XEdbInfoIdList)
  683. for i := 0; i < lenEdb; i++ {
  684. if yData.XEdbInfoIdList[i] != 0 && !utils.InArrayByInt(yData.NoDataEdbList, yData.XEdbInfoIdList[i]) {
  685. if maxI < i {
  686. maxI = i
  687. }
  688. }
  689. }
  690. }
  691. xDataList = xDataList[0:maxI]
  692. for yIndex, yData := range newYDataList {
  693. if len(yData.XEdbInfoIdList) > maxI+1 {
  694. newYDataList[yIndex].XEdbInfoIdList = yData.XEdbInfoIdList[0 : maxI+1]
  695. }
  696. if len(yData.Value) > maxI+1 {
  697. newYDataList[yIndex].Value = yData.Value[0 : maxI+1]
  698. }
  699. }
  700. return
  701. }
  702. nMap := make(map[int]int)
  703. for _, v := range yDataList {
  704. findDateTime := v.ConfigDate
  705. currMonth := findDateTime.Month() // 当前月份
  706. currYear := findDateTime.Year() // 当前年份
  707. //v.XEdbInfoIdList
  708. for edbInfoIndex, edbInfoId := range v.XEdbInfoIdList {
  709. // 第一个不处理
  710. if edbInfoIndex <= baseEdbLen-1 {
  711. continue
  712. }
  713. futureGoodEdbInfo, ok := futureGoodEdbInfoMap[edbInfoId]
  714. if !ok {
  715. err = errors.New("找不到指标")
  716. return
  717. }
  718. n := (futureGoodEdbInfo.Year-currYear)*12 + futureGoodEdbInfo.Month - int(currMonth)
  719. nMap[n] = n
  720. }
  721. }
  722. // 找出所有的N值,并进行正序排列
  723. nList := make([]int, 0)
  724. for _, n := range nMap {
  725. nList = append(nList, n)
  726. }
  727. sort.Slice(nList, func(i, j int) bool {
  728. return nList[i] < nList[j]
  729. })
  730. //prevMonth := 1
  731. for _, n := range nList {
  732. xDataList = append(xDataList, data_manage.XData{
  733. Name: fmt.Sprint("M+", n),
  734. NameEn: fmt.Sprint("M+", n),
  735. })
  736. }
  737. for yIndex, yData := range yDataList {
  738. newYDataList[yIndex].XEdbInfoIdList = []int{}
  739. newYDataList[yIndex].Value = []float64{}
  740. tmpNList := nList
  741. //当前的主力合约
  742. newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, yData.XEdbInfoIdList[0:baseEdbLen]...)
  743. newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, yData.Value[0:baseEdbLen]...)
  744. xEdbInfoIdList := yData.XEdbInfoIdList[baseEdbLen:]
  745. currDataTime := yData.ConfigDate
  746. valIndex := baseEdbLen
  747. needNum := 0
  748. for _, n := range tmpNList {
  749. if len(xEdbInfoIdList) > 0 {
  750. edbInfoId := xEdbInfoIdList[0]
  751. futureGoodEdbInfo, ok := futureGoodEdbInfoMap[edbInfoId]
  752. if !ok {
  753. err = errors.New("找不到指标")
  754. return
  755. }
  756. // 当前距离最早的日期相差的N数
  757. divMonth := (futureGoodEdbInfo.Year-currDataTime.Year())*12 + (futureGoodEdbInfo.Month - int(currDataTime.Month()))
  758. if divMonth == n {
  759. if needNum > 0 {
  760. currVal := yData.Value[valIndex]
  761. preVal := yData.Value[valIndex-1]
  762. hcValDeci := decimal.NewFromFloat(currVal).Sub(decimal.NewFromFloat(preVal)).Div(decimal.NewFromInt(int64(needNum + 1)))
  763. for tmpNum := 0; tmpNum < needNum; tmpNum++ {
  764. newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, 0)
  765. // 赋值平均值
  766. tmpVal, _ := decimal.NewFromFloat(preVal).Add(hcValDeci.Mul(decimal.NewFromInt(int64(tmpNum + 1)))).RoundCeil(4).Float64()
  767. newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, tmpVal)
  768. }
  769. }
  770. newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, edbInfoId)
  771. newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, yData.Value[valIndex])
  772. valIndex++
  773. needNum = 0
  774. if len(xEdbInfoIdList) > 0 {
  775. xEdbInfoIdList = xEdbInfoIdList[1:]
  776. }
  777. } else {
  778. needNum++
  779. }
  780. }
  781. }
  782. }
  783. maxI := 0
  784. for _, yData := range newYDataList {
  785. lenEdb := len(yData.XEdbInfoIdList)
  786. for i := 0; i < lenEdb; i++ {
  787. if yData.XEdbInfoIdList[i] != 0 && !utils.InArrayByInt(yData.NoDataEdbList, yData.XEdbInfoIdList[i]) {
  788. if maxI < i {
  789. maxI = i
  790. }
  791. }
  792. }
  793. }
  794. xDataList = xDataList[0:maxI]
  795. for yIndex, yData := range newYDataList {
  796. if len(yData.XEdbInfoIdList) > maxI+1 {
  797. newYDataList[yIndex].XEdbInfoIdList = yData.XEdbInfoIdList[0 : maxI+1]
  798. }
  799. if len(yData.Value) > maxI+1 {
  800. newYDataList[yIndex].Value = yData.Value[0 : maxI+1]
  801. }
  802. }
  803. return
  804. }
  805. // getFutureGoodEdbInfoList 获取适用的指标列表
  806. func getFutureGoodEdbInfoList(latestDateTime time.Time, tmpFutureGoodEdbInfoList []*future_good2.FutureGoodEdbInfo, barChartInfoDateList []data_manage.BarChartInfoDateReq) (earliestDateTime time.Time, futureGoodEdbInfoList []*future_good2.FutureGoodEdbInfo, err error) {
  807. maxM := 36 //最大36期合约
  808. futureGoodEdbInfoList = make([]*future_good2.FutureGoodEdbInfo, 0)
  809. earliestDateTime = latestDateTime // 数据的最早日期,目的是为了找出最早的合约
  810. for _, barChartInfoDate := range barChartInfoDateList {
  811. var findDateTime time.Time
  812. switch barChartInfoDate.Type {
  813. case 1: //最新值
  814. findDateTime = latestDateTime
  815. case 2: //近期几天
  816. findDateTime = latestDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  817. case 3: // 固定日期
  818. //寻找固定日期的数据
  819. tmpFindDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  820. if tmpErr != nil {
  821. err = tmpErr
  822. return
  823. }
  824. findDateTime = tmpFindDateTime
  825. default:
  826. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  827. return
  828. }
  829. if findDateTime.IsZero() {
  830. err = errors.New("错误的日期")
  831. return
  832. }
  833. if findDateTime.Before(earliestDateTime) {
  834. earliestDateTime = findDateTime
  835. }
  836. }
  837. for _, v := range tmpFutureGoodEdbInfoList {
  838. if v.RegionType == `国内` {
  839. futureGoodEdbInfoList = append(futureGoodEdbInfoList, v)
  840. continue
  841. }
  842. //海外的连续日期,目前
  843. if v.FutureGoodEdbType == 2 {
  844. if v.Month <= maxM {
  845. //addMonth := int(earliestDateTime.Month()) + v.Month
  846. //v.Year = earliestDateTime.Year() + addMonth/12
  847. //realMonth := addMonth % 12
  848. //if realMonth == 0 {
  849. // realMonth = 12
  850. //}
  851. //v.Month = realMonth
  852. futureGoodEdbInfoList = append(futureGoodEdbInfoList, v)
  853. }
  854. continue
  855. }
  856. if v.Year < earliestDateTime.Year() {
  857. continue
  858. }
  859. // 小于等于当前年,那么就肯定是ok的
  860. if v.Year <= earliestDateTime.Year() {
  861. futureGoodEdbInfoList = append(futureGoodEdbInfoList, v)
  862. continue
  863. }
  864. // 如果(当前年-最新日期的年份) * 12个月 + (当前月-最新日期的月份) 小于总月份
  865. if (v.Year-earliestDateTime.Year())*12+(v.Month-int(earliestDateTime.Month())) <= maxM {
  866. futureGoodEdbInfoList = append(futureGoodEdbInfoList, v)
  867. continue
  868. }
  869. }
  870. return
  871. }
  872. // GetNeedDateData 获取合约内需要的日期数据
  873. func GetNeedDateData(needDateTime time.Time, dataList []*models.EdbDataList, edbDataMap map[string]float64, allEdbDataMap map[int]map[string]float64) (findDateTime time.Time, findDataValue float64, isFind bool, err error) {
  874. //dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  875. if len(dataList) <= 0 {
  876. // 没有数据的指标id
  877. return
  878. }
  879. //最早的日期
  880. minDateTime, err := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  881. if err != nil {
  882. return
  883. }
  884. // 该日期存在数据的期货指标的最小数量,目前是现货和期货各1个,总共2个
  885. maxCount := 1
  886. for tmpDateTime := needDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  887. tmpDate := tmpDateTime.Format(utils.FormatDate)
  888. tmpValue, ok := edbDataMap[tmpDate]
  889. if !ok {
  890. continue
  891. }
  892. // 该日期存在数据的指标数量
  893. count := 0
  894. for _, currEdbDataMap := range allEdbDataMap {
  895. _, tmpIsFind := currEdbDataMap[tmpDate]
  896. if tmpIsFind {
  897. count++
  898. if count >= maxCount {
  899. continue
  900. }
  901. }
  902. }
  903. // 该日期存在数据的期货指标数量小于2个,那么要继续往前找
  904. if count < maxCount {
  905. continue
  906. }
  907. //如果能找到数据,那么就返回
  908. // 数据为0,也直接返回,做无值处理
  909. if tmpValue == 0 {
  910. return
  911. }
  912. findDateTime, _ = time.ParseInLocation(utils.FormatDate, tmpDate, time.Local)
  913. findDataValue = tmpValue
  914. isFind = true
  915. return
  916. }
  917. return
  918. }
  919. // FutureGoodChartInfoRefresh
  920. // @author Roc
  921. // @datetime 2023-2-2 18:44:46
  922. // @description 商品价格曲线图表刷新
  923. func FutureGoodChartInfoRefresh(chartInfoId int) (err error) {
  924. var errMsg string
  925. defer func() {
  926. if err != nil {
  927. go alarm_msg.SendAlarmMsg("ChartInfoRefresh:"+errMsg, 3)
  928. fmt.Println("ChartInfoRefresh Err:" + errMsg)
  929. }
  930. }()
  931. edbInfoMappingList, err := models.GetEtaEdbChartEdbMappingList(chartInfoId)
  932. if err != nil {
  933. errMsg = "获取需要刷新的ETA指标失败:Err:" + err.Error()
  934. return
  935. }
  936. edbInfoIds := make([]int, 0)
  937. for _, edbInfoMapping := range edbInfoMappingList {
  938. edbInfoIds = append(edbInfoIds, edbInfoMapping.EdbInfoId)
  939. }
  940. // 获取期货指标
  941. futureGoodEdbInfoMapping, err := models.GetFutureGoodEdbChartEdbMapping(chartInfoId)
  942. if err != nil {
  943. errMsg = "获取需要刷新的商品期货指标失败:Err:" + err.Error()
  944. return
  945. }
  946. // 获取期货指标以及期货数据
  947. futureGoodEdbInfoList, err := future_good2.GetFutureGoodEdbInfoListByParentId(futureGoodEdbInfoMapping.EdbInfoId)
  948. if err != nil {
  949. return
  950. }
  951. // 批量刷新ETA指标
  952. err, _ = data.EdbInfoRefreshAllFromBase(edbInfoIds, false)
  953. if err != nil {
  954. return
  955. }
  956. // 批量刷新期货指标
  957. err = FutureGoodEdbInfoRefreshAllFromBase(futureGoodEdbInfoList, false)
  958. if err != nil {
  959. return
  960. }
  961. return
  962. }