|
@@ -1,6 +1,7 @@
|
|
|
package models
|
|
|
|
|
|
import (
|
|
|
+ "encoding/json"
|
|
|
"errors"
|
|
|
"eta/eta_index_lib/utils"
|
|
|
"fmt"
|
|
@@ -80,6 +81,7 @@ func (obj Ljz) Add(params AddCalculateBatchParams) (edbInfo *EdbInfo, err error,
|
|
|
LatestValue: 0,
|
|
|
ChartImage: "",
|
|
|
Calendar: "",
|
|
|
+ Extra: req.Extra,
|
|
|
}
|
|
|
|
|
|
newEdbInfoId, tmpErr := to.Insert(edbInfo)
|
|
@@ -113,7 +115,7 @@ func (obj Ljz) Add(params AddCalculateBatchParams) (edbInfo *EdbInfo, err error,
|
|
|
}
|
|
|
|
|
|
|
|
|
- err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "")
|
|
|
+ err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", edbInfo.Extra)
|
|
|
|
|
|
return
|
|
|
}
|
|
@@ -152,7 +154,7 @@ func (obj Ljz) Edit(params EditCalculateBatchParams) (err error, errMsg string)
|
|
|
tableName := GetEdbDataTableName(edbInfo.Source, edbInfo.SubSource)
|
|
|
|
|
|
var isRecalculate bool
|
|
|
- if edbInfo.Frequency != req.Frequency {
|
|
|
+ if edbInfo.Frequency != req.Frequency || edbInfo.Extra != req.Extra {
|
|
|
isRecalculate = true
|
|
|
}
|
|
|
|
|
@@ -161,6 +163,7 @@ func (obj Ljz) Edit(params EditCalculateBatchParams) (err error, errMsg string)
|
|
|
edbInfo.Frequency = req.Frequency
|
|
|
edbInfo.Unit = req.Unit
|
|
|
edbInfo.ClassifyId = req.ClassifyId
|
|
|
+ edbInfo.Extra = req.Extra
|
|
|
edbInfo.ModifyTime = time.Now()
|
|
|
_, err = to.Update(edbInfo, "EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime")
|
|
|
if err != nil {
|
|
@@ -180,9 +183,9 @@ func (obj Ljz) Edit(params EditCalculateBatchParams) (err error, errMsg string)
|
|
|
}
|
|
|
if count > 0 {
|
|
|
|
|
|
-
|
|
|
+
|
|
|
if isRecalculate {
|
|
|
- err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "")
|
|
|
+ err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", req.Extra)
|
|
|
}
|
|
|
|
|
|
return
|
|
@@ -228,7 +231,7 @@ func (obj Ljz) Edit(params EditCalculateBatchParams) (err error, errMsg string)
|
|
|
}
|
|
|
|
|
|
|
|
|
- err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "")
|
|
|
+ err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", edbInfo.Extra)
|
|
|
|
|
|
return
|
|
|
}
|
|
@@ -261,7 +264,7 @@ func (obj Ljz) Refresh(params RefreshParams) (err error, errMsg string) {
|
|
|
}()
|
|
|
|
|
|
|
|
|
- err = obj.refresh(to, params.EdbInfo.EdbInfoId, params.EdbInfo.Source, params.EdbInfo.SubSource, params.EdbInfo, fromEdbInfo, params.EdbInfo.EdbCode, params.StartDate)
|
|
|
+ err = obj.refresh(to, params.EdbInfo.EdbInfoId, params.EdbInfo.Source, params.EdbInfo.SubSource, params.EdbInfo, fromEdbInfo, params.EdbInfo.EdbCode, params.StartDate, params.EdbInfo.Extra)
|
|
|
|
|
|
return
|
|
|
}
|
|
@@ -281,7 +284,7 @@ func (obj Ljz) GetEdbType() int {
|
|
|
return utils.CALCULATE_EDB_TYPE
|
|
|
}
|
|
|
|
|
|
-func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo, fromEdbInfo *EdbInfo, edbCode, startDate string) (err error) {
|
|
|
+func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo, fromEdbInfo *EdbInfo, edbCode, startDate string, extra string) (err error) {
|
|
|
dataTableName := GetEdbDataTableName(source, subSource)
|
|
|
edbInfoIdStr := strconv.Itoa(edbInfoId)
|
|
|
|
|
@@ -313,8 +316,19 @@ func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
+ var lastValType int
|
|
|
+ if extra != "" {
|
|
|
+ var lastValConfig CalculateLjzEdbExtra
|
|
|
+ err = json.Unmarshal([]byte(extra), &lastValConfig)
|
|
|
+ if err != nil {
|
|
|
+ err = fmt.Errorf("refreshAllCalculate,extra解析失败,Err:%s", err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+ lastValType = lastValConfig.LastValType
|
|
|
+ }
|
|
|
|
|
|
|
|
|
+
|
|
|
|
|
|
|
|
|
|
|
@@ -326,6 +340,7 @@ func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo
|
|
|
case "年度":
|
|
|
yearMap := make(map[int]float64)
|
|
|
yearList := make([]int, 0)
|
|
|
+ var lastNewDate time.Time
|
|
|
for _, item := range dataList {
|
|
|
itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
|
|
|
if tmpErr != nil {
|
|
@@ -345,10 +360,41 @@ func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo
|
|
|
currTime := time.Date(v, 12, 31, 0, 0, 0, 0, time.Local)
|
|
|
dateList = append(dateList, currTime)
|
|
|
valueMap[currTime] = yearMap[v]
|
|
|
+ lastNewDate = currTime
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ if lastValType == 1 {
|
|
|
+ lastItem := dataList[len(dataList)-1]
|
|
|
+
|
|
|
+ initVal := valueMap[lastNewDate]
|
|
|
+ itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
|
|
|
+ if tmpErr != nil {
|
|
|
+ err = tmpErr
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ startDateT := time.Date(itemDate.Year(), 1, 1, 0, 0, 0, 0, time.Local)
|
|
|
+ allDays := int(lastNewDate.Sub(startDateT).Hours() / 24)
|
|
|
+
|
|
|
+ days := int(itemDate.Sub(startDateT).Hours() / 24)
|
|
|
+
|
|
|
+ daysT := decimal.NewFromInt(int64(days))
|
|
|
+ initValAndMonthT := decimal.NewFromFloat(initVal * float64(allDays))
|
|
|
+ val, ok := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
|
|
|
+ fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
|
|
|
+ if !ok {
|
|
|
+ err = fmt.Errorf("最新值,均值计算失败")
|
|
|
+ return
|
|
|
+ }
|
|
|
+ valueMap[lastNewDate] = val
|
|
|
}
|
|
|
case "半年度":
|
|
|
yearMonthMap := make(map[string]float64)
|
|
|
yearMonthList := make([]string, 0)
|
|
|
+ var lastNewDate time.Time
|
|
|
for _, item := range dataList {
|
|
|
itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
|
|
|
if tmpErr != nil {
|
|
@@ -380,10 +426,41 @@ func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo
|
|
|
currTime = currTime.AddDate(0, 1, -1)
|
|
|
dateList = append(dateList, currTime)
|
|
|
valueMap[currTime] = yearMonthMap[v]
|
|
|
+ lastNewDate = currTime
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ if lastValType == 1 {
|
|
|
+ lastItem := dataList[len(dataList)-1]
|
|
|
+
|
|
|
+ initVal := valueMap[lastNewDate]
|
|
|
+ itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
|
|
|
+ if tmpErr != nil {
|
|
|
+ err = tmpErr
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ startDateT := lastNewDate.AddDate(0, -6, 0)
|
|
|
+ allDays := int(lastNewDate.Sub(startDateT).Hours() / 24)
|
|
|
+
|
|
|
+
|
|
|
+ days := int(itemDate.Sub(startDateT).Hours() / 24)
|
|
|
+
|
|
|
+ daysT := decimal.NewFromInt(int64(days))
|
|
|
+ initValAndMonthT := decimal.NewFromFloat(initVal * float64(allDays))
|
|
|
+ val, ok := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
|
|
|
+ fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
|
|
|
+ if !ok {
|
|
|
+ err = fmt.Errorf("最新值,均值计算失败")
|
|
|
+ return
|
|
|
+ }
|
|
|
+ valueMap[lastNewDate] = val
|
|
|
}
|
|
|
case "季度":
|
|
|
yearMonthMap := make(map[string]float64)
|
|
|
yearMonthList := make([]string, 0)
|
|
|
+ var lastNewDate time.Time
|
|
|
for _, item := range dataList {
|
|
|
itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
|
|
|
if tmpErr != nil {
|
|
@@ -419,10 +496,40 @@ func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo
|
|
|
currTime = currTime.AddDate(0, 1, -1)
|
|
|
dateList = append(dateList, currTime)
|
|
|
valueMap[currTime] = yearMonthMap[v]
|
|
|
+ lastNewDate = currTime
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ if lastValType == 1 {
|
|
|
+ lastItem := dataList[len(dataList)-1]
|
|
|
+
|
|
|
+ initVal := valueMap[lastNewDate]
|
|
|
+ itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
|
|
|
+ if tmpErr != nil {
|
|
|
+ err = tmpErr
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ startDateT := lastNewDate.AddDate(0, -3, 0)
|
|
|
+ allDays := int(lastNewDate.Sub(startDateT).Hours() / 24)
|
|
|
+
|
|
|
+ days := int(itemDate.Sub(startDateT).Hours() / 24)
|
|
|
+
|
|
|
+ daysT := decimal.NewFromInt(int64(days))
|
|
|
+ initValAndMonthT := decimal.NewFromFloat(initVal * float64(allDays))
|
|
|
+ val, ok := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
|
|
|
+ fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
|
|
|
+ if !ok {
|
|
|
+ err = fmt.Errorf("最新值,均值计算失败")
|
|
|
+ return
|
|
|
+ }
|
|
|
+ valueMap[lastNewDate] = val
|
|
|
}
|
|
|
case "月度":
|
|
|
yearMonthMap := make(map[string]float64)
|
|
|
yearMonthList := make([]string, 0)
|
|
|
+ var lastNewDate time.Time
|
|
|
for _, item := range dataList {
|
|
|
itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
|
|
|
if tmpErr != nil {
|
|
@@ -450,6 +557,34 @@ func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo
|
|
|
currTime = currTime.AddDate(0, 1, -1)
|
|
|
dateList = append(dateList, currTime)
|
|
|
valueMap[currTime] = yearMonthMap[v]
|
|
|
+ lastNewDate = currTime
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ if lastValType == 1 {
|
|
|
+ lastItem := dataList[len(dataList)-1]
|
|
|
+
|
|
|
+ initVal := valueMap[lastNewDate]
|
|
|
+ itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
|
|
|
+ if tmpErr != nil {
|
|
|
+ err = tmpErr
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ monthDays := lastNewDate.Day()
|
|
|
+
|
|
|
+ days := itemDate.Day()
|
|
|
+
|
|
|
+ daysT := decimal.NewFromInt(int64(days))
|
|
|
+ initValAndMonthT := decimal.NewFromFloat(initVal * float64(monthDays))
|
|
|
+ val, ok := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
|
|
|
+ fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", monthDays, initVal, days, val)
|
|
|
+ if !ok {
|
|
|
+ err = fmt.Errorf("最新值,均值计算失败")
|
|
|
+ return
|
|
|
+ }
|
|
|
+ valueMap[lastNewDate] = val
|
|
|
}
|
|
|
case "旬度":
|
|
|
tmpDateDataMap := make(map[time.Time]float64)
|
|
@@ -462,20 +597,23 @@ func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo
|
|
|
}
|
|
|
dayInt := itemDate.Year()*100 + int(itemDate.Month())
|
|
|
var currTime time.Time
|
|
|
- if itemDate.Month() <= 10 {
|
|
|
- tmpK := fmt.Sprint(dayInt*100, "10")
|
|
|
+ if itemDate.Day() <= 10 {
|
|
|
+
|
|
|
+ tmpK := fmt.Sprint(dayInt, "10")
|
|
|
currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
- } else if itemDate.Month() <= 20 {
|
|
|
- tmpK := fmt.Sprint(dayInt*100, "20")
|
|
|
+ } else if itemDate.Day() <= 20 {
|
|
|
+
|
|
|
+ tmpK := fmt.Sprint(dayInt, "20")
|
|
|
currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
} else {
|
|
|
+
|
|
|
currTime, err = time.ParseInLocation(utils.FormatYearMonthUnSpace, fmt.Sprint(dayInt), time.Local)
|
|
|
if err != nil {
|
|
|
return
|