chart_info.go 32 KB

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