chart_info.go 35 KB

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