chart_info.go 26 KB

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