predict_edb_info.go 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. package chart
  2. import (
  3. "errors"
  4. edbDataModel "hongze/hongze_yb/models/tables/edb_data"
  5. edbInfoModel "hongze/hongze_yb/models/tables/edb_info"
  6. predictEdbConfModel "hongze/hongze_yb/models/tables/predict_edb_conf"
  7. "hongze/hongze_yb/utils"
  8. "time"
  9. )
  10. // GetChartPredictEdbInfoDataList 获取图表的预测指标的未来数据
  11. func GetChartPredictEdbInfoDataList(predictEdbConf predictEdbConfModel.PredictEdbConf, filtrateStartDateStr, latestDateStr string, lastDataValue float64, endDateStr, frequency string) (predictEdbInfoData []*edbDataModel.EdbDataList, err error) {
  12. endDate, err := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
  13. if err != nil {
  14. return
  15. }
  16. latestDate, err := time.ParseInLocation(utils.FormatDate, latestDateStr, time.Local)
  17. if err != nil {
  18. return
  19. }
  20. // 开始预测数据的时间
  21. startDate := latestDate
  22. // 如果有筛选时间的话
  23. if filtrateStartDateStr != `` {
  24. filtrateStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, filtrateStartDateStr, time.Local)
  25. if tmpErr != nil {
  26. err = tmpErr
  27. return
  28. }
  29. //如果筛选时间晚于实际数据时间,那么就以筛选时间作为获取预测数据的时间
  30. if filtrateStartDate.After(latestDate) {
  31. startDate = filtrateStartDate.AddDate(0, 0, -1)
  32. }
  33. }
  34. dataValue := lastDataValue
  35. if predictEdbConf.RuleType == 2 {
  36. dataValue = predictEdbConf.FixedValue
  37. }
  38. //获取后面的预测数据
  39. dayList := getPredictEdbDayList(startDate, endDate, frequency)
  40. predictEdbInfoData = make([]*edbDataModel.EdbDataList, 0)
  41. for k, v := range dayList {
  42. predictEdbInfoData = append(predictEdbInfoData, &edbDataModel.EdbDataList{
  43. EdbDataId: int(predictEdbConf.PredictEdbInfoID) + 10000000000 + k,
  44. EdbInfoId: int(predictEdbConf.PredictEdbInfoID),
  45. DataTime: v.Format(utils.FormatDate),
  46. Value: dataValue,
  47. DataTimestamp: (v.UnixNano() / 1e6) + 1000, //前端需要让加1s,说是2022-09-01 00:00:00 这样的整点不合适
  48. })
  49. }
  50. return
  51. }
  52. // GetPredictEdbDayList 获取预测指标日期列表
  53. func getPredictEdbDayList(startDate, endDate time.Time, frequency string) (dayList []time.Time) {
  54. //if !utils.InArrayByStr([]string{"日度", "周度", "月度"}, frequency)
  55. switch frequency {
  56. case "日度":
  57. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 1) {
  58. //周六、日排除
  59. if currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday {
  60. continue
  61. }
  62. dayList = append(dayList, currDate)
  63. }
  64. case "周度":
  65. //nextDate := startDate.AddDate(0, 0, 7)
  66. for currDate := startDate.AddDate(0, 0, 7); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 7) {
  67. dayList = append(dayList, currDate)
  68. }
  69. case "月度":
  70. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  71. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 2, -1)
  72. if !currDate.After(endDate) {
  73. dayList = append(dayList, currDate)
  74. }
  75. }
  76. }
  77. return
  78. }
  79. // GetPredictDataListByPredictEdbInfoId 根据预测指标id获取预测指标的数据
  80. func GetPredictDataListByPredictEdbInfoId(edbInfoId int, startDate, endDate string, isTimeBetween bool) (dataList []*edbDataModel.EdbDataList, sourceEdbInfoItem *edbInfoModel.EdbInfo, predictEdbConf *predictEdbConfModel.PredictEdbConf, err error, errMsg string) {
  81. edbInfo, err := edbInfoModel.GetEdbInfoById(edbInfoId)
  82. if err != nil {
  83. errMsg = `获取预测指标信息失败`
  84. return
  85. }
  86. return GetPredictDataListByPredictEdbInfo(edbInfo, startDate, endDate, isTimeBetween)
  87. }
  88. // GetPredictDataListByPredictEdbInfo 根据预测指标信息获取预测指标的数据
  89. func GetPredictDataListByPredictEdbInfo(edbInfo *edbInfoModel.EdbInfo, startDate, endDate string, isTimeBetween bool) (dataList []*edbDataModel.EdbDataList, sourceEdbInfoItem *edbInfoModel.EdbInfo, predictEdbConf *predictEdbConfModel.PredictEdbConf, err error, errMsg string) {
  90. // 非计算指标,直接从表里获取数据
  91. if edbInfo.EdbType != 1 {
  92. if !isTimeBetween {
  93. endDate = ``
  94. }
  95. return GetPredictCalculateDataListByPredictEdbInfo(edbInfo, startDate, endDate)
  96. }
  97. // 查找该预测指标配置
  98. predictEdbConf, err = predictEdbConfModel.GetPredictEdbConfById(edbInfo.EdbInfoId)
  99. if err != nil && err != utils.ErrNoRow {
  100. errMsg = "获取预测指标配置信息失败"
  101. return
  102. }
  103. if predictEdbConf == nil {
  104. errMsg = "获取预测指标配置信息失败"
  105. err = errors.New(errMsg)
  106. return
  107. }
  108. // 来源指标
  109. sourceEdbInfoItem, err = edbInfoModel.GetEdbInfoById(int(predictEdbConf.SourceEdbInfoID))
  110. if err != nil {
  111. if err == utils.ErrNoRow {
  112. errMsg = "找不到来源指标信息"
  113. err = errors.New(errMsg)
  114. }
  115. return
  116. }
  117. //获取指标数据(实际已生成)
  118. dataList, err = edbDataModel.GetEdbDataList(sourceEdbInfoItem.Source, sourceEdbInfoItem.EdbInfoId, startDate, endDate)
  119. if err != nil {
  120. return
  121. }
  122. // 获取预测指标未来的数据
  123. predictDataList := make([]*edbDataModel.EdbDataList, 0)
  124. endDateStr := edbInfo.EndDate.Format(utils.FormatDate) //预测指标的结束日期
  125. if isTimeBetween { //如果是时间区间,那么
  126. reqEndDateTime, _ := time.ParseInLocation(utils.FormatDate, endDate, time.Local)
  127. // 如果选择的时间区间结束日期 晚于 当天,那么预测数据截止到当天
  128. if reqEndDateTime.Before(edbInfo.EndDate) {
  129. endDateStr = endDate
  130. }
  131. }
  132. predictDataList, err = GetChartPredictEdbInfoDataList(*predictEdbConf, startDate, sourceEdbInfoItem.LatestDate.Format(utils.FormatDate), sourceEdbInfoItem.LatestValue, endDateStr, edbInfo.Frequency)
  133. if err != nil {
  134. return
  135. }
  136. dataList = append(dataList, predictDataList...)
  137. if len(predictDataList) > 0 {
  138. tmpValue := predictDataList[0]
  139. // 如果最大值 小于 预测值,那么将预测值作为最大值数据返回
  140. if edbInfo.MaxValue < tmpValue.Value {
  141. edbInfo.MaxValue = tmpValue.Value
  142. }
  143. // 如果最小值 大于 预测值,那么将预测值作为最小值数据返回
  144. if edbInfo.MinValue > tmpValue.Value {
  145. edbInfo.MinValue = tmpValue.Value
  146. }
  147. }
  148. return
  149. }
  150. // GetPredictCalculateDataListByPredictEdbInfo 根据预测运算指标信息获取预测指标的数据
  151. func GetPredictCalculateDataListByPredictEdbInfo(edbInfo *edbInfoModel.EdbInfo, startDate, endDate string) (dataList []*edbDataModel.EdbDataList, sourceEdbInfoItem *edbInfoModel.EdbInfo, predictEdbConf *predictEdbConfModel.PredictEdbConf, err error, errMsg string) {
  152. dataList, err = edbDataModel.GetEdbDataList(edbInfo.Source, edbInfo.EdbInfoId, startDate, endDate)
  153. return
  154. }