|
@@ -46,6 +46,7 @@ func SavePredictCalculateBp(req *EdbInfoCalculateBatchSaveReq, fromEdbInfo *EdbI
|
|
|
edbInfo.UnitEn = req.Unit
|
|
|
edbInfo.CalculateFormula = req.Formula
|
|
|
edbInfo.EdbType = 2
|
|
|
+ edbInfo.EmptyType = req.EmptyType
|
|
|
newEdbInfoId, tmpErr := to.Insert(edbInfo)
|
|
|
if tmpErr != nil {
|
|
|
err = tmpErr
|
|
@@ -90,11 +91,12 @@ func SavePredictCalculateBp(req *EdbInfoCalculateBatchSaveReq, fromEdbInfo *EdbI
|
|
|
edbInfo.Unit = req.Unit
|
|
|
edbInfo.EdbNameSource = req.EdbName
|
|
|
}
|
|
|
-
|
|
|
+ oldEmptyType := edbInfo.EmptyType
|
|
|
edbInfo.Frequency = req.Frequency
|
|
|
edbInfo.ClassifyId = req.ClassifyId
|
|
|
+ edbInfo.EmptyType = req.EmptyType
|
|
|
edbInfo.ModifyTime = time.Now()
|
|
|
- _, err = to.Update(edbInfo, "EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime", "EdbNameEn", "UnitEn")
|
|
|
+ _, err = to.Update(edbInfo, "EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime", "EdbNameEn", "UnitEn", "EmptyType")
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -110,7 +112,7 @@ func SavePredictCalculateBp(req *EdbInfoCalculateBatchSaveReq, fromEdbInfo *EdbI
|
|
|
err = errors.New("判断指标是否改变失败,Err:" + e.Error())
|
|
|
return
|
|
|
}
|
|
|
- if count > 0 { // 指标未被替换,无需重新计算
|
|
|
+ if count > 0 && oldEmptyType == edbInfo.EmptyType { // 指标未被替换,无需重新计算
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -153,12 +155,12 @@ func SavePredictCalculateBp(req *EdbInfoCalculateBatchSaveReq, fromEdbInfo *EdbI
|
|
|
}
|
|
|
|
|
|
//计算数据
|
|
|
- latestDateStr, latestValue, err = refreshAllPredictCalculateBp(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, fromEdbInfo, edbInfo.EdbCode, "", "", 0)
|
|
|
+ latestDateStr, latestValue, err = refreshAllPredictCalculateBp(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, fromEdbInfo, edbInfo.EdbCode, "", "", 0, edbInfo.EmptyType)
|
|
|
|
|
|
return
|
|
|
}
|
|
|
|
|
|
-func RefreshAllPredictCalculateBp(edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string) (latestDateStr string, latestValue float64, err error) {
|
|
|
+func RefreshAllPredictCalculateBp(edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string, emptyType int) (latestDateStr string, latestValue float64, err error) {
|
|
|
o := orm.NewOrm()
|
|
|
to, err := o.Begin()
|
|
|
if err != nil {
|
|
@@ -174,14 +176,16 @@ func RefreshAllPredictCalculateBp(edbInfoId, source, subSource int, fromEdbInfo
|
|
|
}()
|
|
|
|
|
|
// 计算数据
|
|
|
- latestDateStr, latestValue, err = refreshAllPredictCalculateBp(to, edbInfoId, source, subSource, fromEdbInfo, edbCode, startDate, endDate, 1)
|
|
|
+ latestDateStr, latestValue, err = refreshAllPredictCalculateBp(to, edbInfoId, source, subSource, fromEdbInfo, edbCode, startDate, endDate, 0, emptyType)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// refreshAllPredictCalculateBp 刷新变频数据
|
|
|
-func refreshAllPredictCalculateBp(to orm.TxOrmer, edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string, order int) (latestDateStr string, latestValue float64, err error) {
|
|
|
+func refreshAllPredictCalculateBp(to orm.TxOrmer, edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string, order, emptyType int) (latestDateStr string, latestValue float64, err error) {
|
|
|
edbInfoIdStr := strconv.Itoa(edbInfoId)
|
|
|
- // 获取关联指标数据
|
|
|
+ //计算数据
|
|
|
+
|
|
|
+ //获取来源指标的数据
|
|
|
dataList, err := GetPredictEdbDataListAllByStartDate(fromEdbInfo, order, "")
|
|
|
if err != nil {
|
|
|
return
|
|
@@ -199,189 +203,144 @@ func refreshAllPredictCalculateBp(to orm.TxOrmer, edbInfoId, source, subSource i
|
|
|
}
|
|
|
fmt.Println("source:", source)
|
|
|
|
|
|
- //获取变频指标所有数据
|
|
|
+ //获取升频指标所有数据
|
|
|
existDataList, err := GetAllEdbDataListByTo(to, edbInfoId, source, subSource)
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
//计算指标的map
|
|
|
existDataMap := make(map[string]*EdbData, 0)
|
|
|
-
|
|
|
+ removeDateMap := make(map[string]struct{})
|
|
|
addSql := ` INSERT INTO edb_data_predict_calculate_bp(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
|
|
|
- var isAdd bool
|
|
|
|
|
|
- var lastValue float64 //最后数据的值(float64)
|
|
|
- var lastValueStr string //最后数据的值(string)
|
|
|
+ var isAdd bool
|
|
|
//待删除的日期
|
|
|
removeDateList := make([]string, 0)
|
|
|
- if len(existDataList) > 0 {
|
|
|
- //第一个已经入库的日期
|
|
|
- firstExistDataTimeStr := existDataList[0].DataTime //计算指标数据第一条的日期字符串
|
|
|
- if len(dateArr) > 0 {
|
|
|
- firstFromDataTimeStr := dateArr[0] //来源数据第一条的日期字符串
|
|
|
- firstExistDataTime, _ := time.ParseInLocation(utils.FormatDate, firstExistDataTimeStr, time.Local) //计算指标数据第一条的日期(time类型)
|
|
|
- firstFromDataTime, _ := time.ParseInLocation(utils.FormatDate, firstFromDataTimeStr, time.Local) //来源数据第一条的日期(time类型)
|
|
|
- nowDateStr := time.Now().Format(utils.FormatDate) //当天日期字符串
|
|
|
- nowDate, _ := time.ParseInLocation(utils.FormatDate, nowDateStr, firstFromDataTime.Location()) //当天日期(time类型)
|
|
|
+ if len(existDataList) > 0 && len(dateArr) == 0 {
|
|
|
+ //如果没有来源指标数据,那么已经入库的计算指标数据需要全部删除
|
|
|
+ tableName := GetEdbDataTableName(source, subSource)
|
|
|
+ sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ?`, tableName)
|
|
|
+ _, err = to.Raw(sql, edbInfoId).Exec()
|
|
|
+ if err != nil {
|
|
|
+ err = fmt.Errorf("删除所有的升频指标数据失败,Err:" + err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+ return
|
|
|
+ }
|
|
|
|
|
|
- lastValue = fromDataMap[firstFromDataTimeStr]
|
|
|
- lastValueStr = decimal.NewFromFloat(lastValue).String()
|
|
|
- //第一步: 判断来源指标的开始时间与计算指标的开始时间是否相等,相等的话,那么就不需要对两个时间之间的数据做处理
|
|
|
- if firstExistDataTimeStr != firstFromDataTimeStr {
|
|
|
- if firstExistDataTime.Before(firstFromDataTime) { //如果计算指标第一条数据的开始时间 早于 来源指标的第一条开始时间,那么需要对两个时间之间的计算指标数据做 删除处理
|
|
|
- for _, v := range existDataList {
|
|
|
- if v.DataTime == firstFromDataTimeStr {
|
|
|
- if tmpLastValue, ok := fromDataMap[firstFromDataTimeStr]; ok { //来源指标当天的数据
|
|
|
- lastValue = tmpLastValue
|
|
|
- lastValueStr = decimal.NewFromFloat(lastValue).String()
|
|
|
- }
|
|
|
- break
|
|
|
- }
|
|
|
- removeDateList = append(removeDateList, v.DataTime)
|
|
|
- }
|
|
|
- } else {
|
|
|
- for _, v := range dateArr { //如果计算指标第一条数据的开始时间 晚于 来源指标的第一条开始时间,那么需要对两个时间之间的计算指标数据做 新增处理
|
|
|
- vDataTime, _ := time.ParseInLocation(utils.FormatDate, v, time.Local) //当前日期(time类型)
|
|
|
- if firstExistDataTime.Equal(vDataTime) || firstExistDataTime.Before(vDataTime) {
|
|
|
- if tmpLastValue, ok := fromDataMap[v]; ok { //来源指标当天的数据
|
|
|
- lastValue = tmpLastValue
|
|
|
- lastValueStr = decimal.NewFromFloat(lastValue).String()
|
|
|
- }
|
|
|
- break
|
|
|
- }
|
|
|
+ existMap := make(map[string]string)
|
|
|
|
|
|
- currentDate, _ := time.ParseInLocation(utils.FormatDate, v, time.Local)
|
|
|
- timestamp := currentDate.UnixNano() / 1e6
|
|
|
- timestampStr := fmt.Sprintf("%d", timestamp)
|
|
|
- addSql += GetAddSql(edbInfoIdStr, edbCode, v, timestampStr, lastValueStr)
|
|
|
- // 实际数据的值
|
|
|
- if fromEdbInfo.LatestDate == v {
|
|
|
- latestValue = lastValue
|
|
|
- }
|
|
|
- isAdd = true
|
|
|
- }
|
|
|
- }
|
|
|
+ dataLen := len(dataList)
|
|
|
+ //第三步: 已经入库的数据处理
|
|
|
+ for _, v := range existDataList {
|
|
|
+ existDataMap[v.DataTime] = v
|
|
|
+ removeDateMap[v.DataTime] = struct{}{}
|
|
|
+ }
|
|
|
+ for i := 0; i < dataLen; i++ {
|
|
|
+ //当期
|
|
|
+ currentItem := dataList[i]
|
|
|
+ var prevItem *EdbInfoSearchData
|
|
|
+ if emptyType == 3 { //3后值填充,其余前值填充
|
|
|
+ if i >= 1 {
|
|
|
+ prevItem = dataList[i-1]
|
|
|
}
|
|
|
-
|
|
|
- //第二步 剩余数据每天修改
|
|
|
-
|
|
|
- day := int(nowDate.Sub(firstExistDataTime).Hours() / float64(24))
|
|
|
-
|
|
|
- //第三步: 已经入库的数据处理
|
|
|
- for _, v := range existDataList {
|
|
|
- existDataMap[v.DataTime] = v
|
|
|
+ }
|
|
|
+ currentDate, _ := time.ParseInLocation(utils.FormatDate, currentItem.DataTime, time.Local)
|
|
|
+ var day int
|
|
|
+ var preItem *EdbInfoSearchData
|
|
|
+ var preDate time.Time
|
|
|
+ if i == 0 {
|
|
|
+ if emptyType == 3 { //后值填充
|
|
|
+ day = 0 //最新的时间就是来源指标的最新日期
|
|
|
+ preDate = currentDate
|
|
|
+ } else {
|
|
|
+ day = int(time.Now().Sub(currentDate).Hours() / float64(24))
|
|
|
+ preDate = time.Date(time.Now().Year(), time.Now().Month(), time.Now().Day(), 0, 0, 0, 0, time.Local)
|
|
|
}
|
|
|
-
|
|
|
- for k := day; k >= 0; k-- {
|
|
|
- needDay := nowDate.AddDate(0, 0, -k)
|
|
|
- needDayStr := needDay.Format(utils.FormatDate)
|
|
|
- tmpExistData, ok := existDataMap[needDayStr]
|
|
|
- if ok {
|
|
|
- if tmpLastValue, ok := fromDataMap[tmpExistData.DataTime]; ok { //来源指标当天的数据
|
|
|
- lastValue = tmpLastValue
|
|
|
- //lastValueStr = decimal.NewFromFloat(lastValue).String()
|
|
|
- lastValueStr = fmt.Sprintf("%.4f", lastValue)
|
|
|
- }
|
|
|
- if fromEdbInfo.LatestDate == tmpExistData.DataTime {
|
|
|
- latestValue = lastValue
|
|
|
+ } else {
|
|
|
+ j := i - 1
|
|
|
+ if j < dataLen {
|
|
|
+ preItem = dataList[j]
|
|
|
+ preDate, _ = time.ParseInLocation(utils.FormatDate, preItem.DataTime, time.Local)
|
|
|
+ day = int(preDate.Sub(currentDate).Hours() / float64(24))
|
|
|
+ utils.FileLog.Info("preItem.DataTime:" + preItem.DataTime + ";currentItem.DataTime" + currentItem.DataTime)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for k := 0; k <= day; k++ {
|
|
|
+ needDay := preDate.AddDate(0, 0, -k)
|
|
|
+ needDayStr := needDay.Format(utils.FormatDate)
|
|
|
+ delete(removeDateMap, needDayStr)
|
|
|
+ existKey := edbCode + needDayStr
|
|
|
+ if _, ok := existMap[existKey]; !ok {
|
|
|
+ timestamp := needDay.UnixNano() / 1e6
|
|
|
+ timestampStr := fmt.Sprintf("%d", timestamp)
|
|
|
+ valStr := decimal.NewFromFloat(currentItem.Value).String()
|
|
|
+ if prevItem != nil && needDayStr != currentItem.DataTime {
|
|
|
+ valStr = decimal.NewFromFloat(prevItem.Value).String()
|
|
|
+ }
|
|
|
+ if fromEdbInfo.LatestDate == needDayStr {
|
|
|
+ if prevItem != nil && needDayStr != currentItem.DataTime {
|
|
|
+ latestValue = prevItem.Value
|
|
|
+ } else {
|
|
|
+ latestValue = currentItem.Value
|
|
|
}
|
|
|
+ }
|
|
|
+
|
|
|
+ tmpExistData, ok2 := existDataMap[needDayStr]
|
|
|
+ if !ok2 {
|
|
|
+ addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, valStr)
|
|
|
+ isAdd = true
|
|
|
+ } else {
|
|
|
//如果对应的值不匹配
|
|
|
- if tmpExistData.Value != lastValueStr {
|
|
|
- err = ModifyEdbDataById(source, subSource, tmpExistData.EdbDataId, lastValueStr)
|
|
|
+ if tmpExistData.Value != valStr {
|
|
|
+ err = ModifyEdbDataById(source, subSource, tmpExistData.EdbDataId, valStr)
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
- } else {
|
|
|
- timestamp := needDay.UnixNano() / 1e6
|
|
|
- timestampStr := fmt.Sprintf("%d", timestamp)
|
|
|
- addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, lastValueStr)
|
|
|
- if fromEdbInfo.LatestDate == needDayStr {
|
|
|
- latestValue = lastValue
|
|
|
- }
|
|
|
- isAdd = true
|
|
|
}
|
|
|
- }
|
|
|
- } else {
|
|
|
- //如果没有来源指标数据,那么已经入库的计算指标数据需要全部删除
|
|
|
- tableName := GetEdbDataTableName(source, subSource)
|
|
|
- sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ?`, tableName)
|
|
|
- _, err = to.Raw(sql, edbInfoId).Exec()
|
|
|
- if err != nil {
|
|
|
- err = fmt.Errorf("删除所有的变频指标数据失败,Err:" + err.Error())
|
|
|
- return
|
|
|
- }
|
|
|
|
|
|
- //for _, v := range existDataList {
|
|
|
- // removeDateList = append(removeDateList, v.DataTime)
|
|
|
- //}
|
|
|
+ }
|
|
|
+ existMap[existKey] = needDayStr
|
|
|
}
|
|
|
- } else {
|
|
|
- existMap := make(map[string]string)
|
|
|
- dataLen := len(dataList)
|
|
|
-
|
|
|
- for i := 0; i < dataLen; i++ {
|
|
|
- //当期
|
|
|
- currentItem := dataList[i]
|
|
|
+ existKey := edbCode + currentItem.DataTime
|
|
|
+ if _, ok := existMap[existKey]; !ok {
|
|
|
currentDate, _ := time.ParseInLocation(utils.FormatDate, currentItem.DataTime, time.Local)
|
|
|
- var day int
|
|
|
- var preItem *EdbInfoSearchData
|
|
|
- var preDate time.Time
|
|
|
- if i == 0 {
|
|
|
- day = int(time.Now().Sub(currentDate).Hours() / float64(24))
|
|
|
- preDate = time.Now()
|
|
|
- } else {
|
|
|
- j := i - 1
|
|
|
- if j < dataLen {
|
|
|
- preItem = dataList[j]
|
|
|
- preDate, _ = time.ParseInLocation(utils.FormatDate, preItem.DataTime, time.Local)
|
|
|
- day = int(preDate.Sub(currentDate).Hours() / float64(24))
|
|
|
- utils.FileLog.Info("preItem.DataTime:" + preItem.DataTime + ";currentItem.DataTime" + currentItem.DataTime)
|
|
|
- }
|
|
|
+ timestamp := currentDate.UnixNano() / 1e6
|
|
|
+ timestampStr := fmt.Sprintf("%d", timestamp)
|
|
|
+ valStr := decimal.NewFromFloat(currentItem.Value).String()
|
|
|
+ if fromEdbInfo.LatestDate == currentItem.DataTime {
|
|
|
+ latestValue = currentItem.Value
|
|
|
}
|
|
|
- for k := 0; k <= day; k++ {
|
|
|
- needDay := preDate.AddDate(0, 0, -k)
|
|
|
- needDayStr := needDay.Format(utils.FormatDate)
|
|
|
- existKey := edbCode + needDayStr
|
|
|
- if _, ok := existMap[existKey]; !ok {
|
|
|
- timestamp := needDay.UnixNano() / 1e6
|
|
|
- timestampStr := fmt.Sprintf("%d", timestamp)
|
|
|
- valStr := decimal.NewFromFloat(currentItem.Value).String()
|
|
|
- addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, valStr)
|
|
|
- if fromEdbInfo.LatestDate == needDayStr {
|
|
|
- latestValue = currentItem.Value
|
|
|
- }
|
|
|
- isAdd = true
|
|
|
- }
|
|
|
- existMap[existKey] = needDayStr
|
|
|
- }
|
|
|
- existKey := edbCode + currentItem.DataTime
|
|
|
- if _, ok := existMap[existKey]; !ok {
|
|
|
- currentDate, _ := time.ParseInLocation(utils.FormatDate, currentItem.DataTime, time.Local)
|
|
|
- timestamp := currentDate.UnixNano() / 1e6
|
|
|
- timestampStr := fmt.Sprintf("%d", timestamp)
|
|
|
- valStr := decimal.NewFromFloat(currentItem.Value).String()
|
|
|
+ tmpExistData, ok2 := existDataMap[currentItem.DataTime]
|
|
|
+ if !ok2 {
|
|
|
addSql += GetAddSql(edbInfoIdStr, edbCode, currentItem.DataTime, timestampStr, valStr)
|
|
|
- if fromEdbInfo.LatestDate == currentItem.DataTime {
|
|
|
- latestValue = currentItem.Value
|
|
|
- }
|
|
|
isAdd = true
|
|
|
+ } else {
|
|
|
+ //如果对应的值不匹配
|
|
|
+ if tmpExistData.Value != valStr {
|
|
|
+ err = ModifyEdbDataById(source, subSource, tmpExistData.EdbDataId, valStr)
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
- existMap[existKey] = currentItem.DataTime
|
|
|
+
|
|
|
}
|
|
|
+ existMap[existKey] = currentItem.DataTime
|
|
|
}
|
|
|
|
|
|
+ for k, _ := range removeDateMap {
|
|
|
+ removeDateList = append(removeDateList, k)
|
|
|
+ }
|
|
|
// 删除不需要的指标数据
|
|
|
if len(removeDateList) > 0 {
|
|
|
- removeDateStr := strings.Join(removeDateList, `","`)
|
|
|
- removeDateStr = `"` + removeDateStr + `"`
|
|
|
//如果拼接指标变更了,那么需要删除所有的指标数据
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
- sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (%s) `, tableName, removeDateStr)
|
|
|
-
|
|
|
- _, err = to.Raw(sql, edbInfoId).Exec()
|
|
|
+ sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (`+utils.GetOrmInReplace(len(removeDateList))+`) `, tableName)
|
|
|
+ _, err = to.Raw(sql, edbInfoId, removeDateList).Exec()
|
|
|
if err != nil {
|
|
|
- err = fmt.Errorf("删除不存在的变频指标数据失败,Err:" + err.Error())
|
|
|
+ err = fmt.Errorf("删除不存在的升频指标数据失败,Err:" + err.Error())
|
|
|
return
|
|
|
}
|
|
|
}
|