123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740 |
- package models
- import (
- "encoding/json"
- "errors"
- "eta/eta_index_lib/utils"
- "fmt"
- "github.com/beego/beego/v2/client/orm"
- "github.com/shopspring/decimal"
- "reflect"
- "strconv"
- "strings"
- "time"
- )
- // Ljz 累计值
- type Ljz struct {
- }
- // Add 添加
- func (obj Ljz) Add(params AddCalculateBatchParams) (edbInfo *EdbInfo, err error, errMsg string) {
- req := params.Req
- fromEdbInfo := params.FromEdbInfo
- edbCode := params.EdbCode
- frequencyInt := utils.CheckFrequency(fromEdbInfo.Frequency, req.Frequency)
- if frequencyInt < 0 {
- errMsg = "频度异常,不允许低频转为高频"
- err = errors.New(errMsg)
- return
- }
- if frequencyInt == 0 {
- errMsg = "频度异常,不允许同频计算累计值"
- err = errors.New(errMsg)
- return
- }
- o := orm.NewOrm()
- to, err := o.Begin()
- if err != nil {
- return
- }
- defer func() {
- if err != nil {
- _ = to.Rollback()
- fmt.Println(reflect.TypeOf(obj).Name(), ";Add,Err:"+err.Error())
- } else {
- _ = to.Commit()
- }
- }()
- edbInfo = &EdbInfo{
- //EdbInfoId: 0,
- SourceName: obj.GetSourceName(),
- Source: obj.GetSource(),
- EdbCode: edbCode,
- EdbName: req.EdbName,
- EdbNameSource: req.EdbName,
- Frequency: req.Frequency,
- Unit: req.Unit,
- StartDate: "",
- EndDate: "",
- ClassifyId: req.ClassifyId,
- SysUserId: params.SysUserId,
- SysUserRealName: params.SysUserRealName,
- UniqueCode: params.UniqueCode,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- MinValue: 0,
- MaxValue: 0,
- CalculateFormula: req.Formula,
- EdbType: obj.GetEdbType(),
- Sort: GetAddEdbMaxSortByClassifyId(req.ClassifyId, utils.EDB_INFO_TYPE),
- MoveType: 0,
- MoveFrequency: "",
- NoUpdate: 0,
- ServerUrl: "",
- EdbInfoType: 0,
- EdbNameEn: req.EdbName,
- UnitEn: req.Unit,
- LatestDate: "",
- LatestValue: 0,
- ChartImage: "",
- Calendar: "",
- Extra: req.Extra,
- }
- newEdbInfoId, tmpErr := to.Insert(edbInfo)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- edbInfo.EdbInfoId = int(newEdbInfoId)
- //关联关系
- {
- calculateMappingItem := new(EdbInfoCalculateMapping)
- calculateMappingItem.CreateTime = time.Now()
- calculateMappingItem.ModifyTime = time.Now()
- calculateMappingItem.Sort = 1
- calculateMappingItem.EdbCode = edbCode
- calculateMappingItem.EdbInfoId = edbInfo.EdbInfoId
- calculateMappingItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
- calculateMappingItem.FromEdbCode = fromEdbInfo.EdbCode
- calculateMappingItem.FromEdbName = fromEdbInfo.EdbName
- calculateMappingItem.FromSource = fromEdbInfo.Source
- calculateMappingItem.FromSourceName = fromEdbInfo.SourceName
- calculateMappingItem.FromTag = ""
- calculateMappingItem.Source = edbInfo.Source
- calculateMappingItem.SourceName = edbInfo.SourceName
- calculateMappingItem.FromSubSource = fromEdbInfo.SubSource
- _, err = to.Insert(calculateMappingItem)
- if err != nil {
- return
- }
- }
- //计算数据
- err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", edbInfo.Extra)
- return
- }
- // Edit 编辑
- func (obj Ljz) Edit(params EditCalculateBatchParams) (err error, errMsg string) {
- req := params.Req
- edbInfo := params.EdbInfo
- fromEdbInfo := params.FromEdbInfo
- frequencyInt := utils.CheckFrequency(fromEdbInfo.Frequency, req.Frequency)
- if frequencyInt < 0 {
- errMsg = "频度异常,不允许低频转为高频"
- err = errors.New(errMsg)
- return
- }
- if frequencyInt == 0 {
- errMsg = "频度异常,不允许同频计算累计值"
- err = errors.New(errMsg)
- return
- }
- o := orm.NewOrm()
- to, err := o.Begin()
- if err != nil {
- return
- }
- defer func() {
- if err != nil {
- _ = to.Rollback()
- fmt.Println(reflect.TypeOf(obj).Name(), ";Edit,Err:"+err.Error())
- } else {
- _ = to.Commit()
- }
- }()
- tableName := GetEdbDataTableName(edbInfo.Source, edbInfo.SubSource)
- var isRecalculate bool
- if edbInfo.Frequency != req.Frequency || edbInfo.Extra != req.Extra {
- isRecalculate = true
- }
- //修改指标信息
- edbInfo.EdbName = req.EdbName
- edbInfo.EdbNameSource = req.EdbName
- edbInfo.Frequency = req.Frequency
- edbInfo.Unit = req.Unit
- edbInfo.ClassifyId = req.ClassifyId
- edbInfo.Extra = req.Extra
- edbInfo.EdbNameEn = req.EdbNameEn
- edbInfo.UnitEn = req.UnitEn
- edbInfo.ModifyTime = time.Now()
- _, err = to.Update(edbInfo, "EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime", "Extra", "EdbNameEn", "UnitEn")
- if err != nil {
- return
- }
- var existCondition string
- var existPars []interface{}
- existCondition += " AND edb_info_id=? AND from_edb_info_id=? "
- existPars = append(existPars, edbInfo.EdbInfoId, req.FromEdbInfoId)
- //判断计算指标是否被更换
- count, err := GetEdbInfoCalculateCountByCondition(existCondition, existPars)
- if err != nil {
- err = errors.New("判断指标是否改变失败,Err:" + err.Error())
- return
- }
- if count > 0 { // 指标未被替换,无需删除关联数据
- // todo 频度被换了,需要重新计算
- if isRecalculate {
- err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", req.Extra)
- }
- return
- }
- //删除,计算指标关联的,基础指标的关联关系
- sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? `
- _, err = to.Raw(sql, edbInfo.EdbInfoId).Exec()
- if err != nil {
- return
- }
- //清空原有数据
- sql = ` DELETE FROM ` + tableName + ` WHERE edb_info_id = ? `
- _, err = to.Raw(sql, edbInfo.EdbInfoId).Exec()
- if err != nil {
- return
- }
- //关联关系
- {
- calculateMappingItem := &EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: edbInfo.EdbInfoId,
- Source: obj.GetSource(),
- SourceName: obj.GetSourceName(),
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- FromTag: "",
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- FromSubSource: fromEdbInfo.SubSource,
- }
- _, err = to.Insert(calculateMappingItem)
- if err != nil {
- return
- }
- }
- //计算数据
- err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", edbInfo.Extra)
- return
- }
- // Refresh 刷新
- func (obj Ljz) Refresh(params RefreshParams) (err error, errMsg string) {
- calculateMapping, err := GetEdbInfoCalculateMappingDetail(params.EdbInfo.EdbInfoId)
- if err != nil {
- errMsg = "GetEdbInfoCalculateLjzDetail Err:" + err.Error()
- return
- }
- fromEdbInfo, err := GetEdbInfoById(calculateMapping.FromEdbInfoId)
- if err != nil {
- errMsg = "GetEdbInfoById Err:" + err.Error()
- return
- }
- o := orm.NewOrm()
- to, err := o.Begin()
- if err != nil {
- return
- }
- defer func() {
- if err != nil {
- _ = to.Rollback()
- fmt.Println(reflect.TypeOf(obj).Name(), ";Refresh,Err:"+err.Error())
- } else {
- _ = to.Commit()
- }
- }()
- // 计算数据
- 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
- }
- // GetSource 获取来源编码id
- func (obj Ljz) GetSource() int {
- return utils.DATA_SOURCE_CALCULATE_LJZ
- }
- // GetSourceName 获取来源名称
- func (obj Ljz) GetSourceName() string {
- return utils.DATA_SOURCE_NAME_CALCULATE_LJZ
- }
- // GetEdbType 获取指标类型
- 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, extra string) (err error) {
- dataTableName := GetEdbDataTableName(source, subSource)
- edbInfoIdStr := strconv.Itoa(edbInfoId)
- //计算数据
- var isWeekData bool // 是否周度数据,如果是周度数据的话,是需要变频的,最后结果还需要除以7
- // 周度数据需要先变成日度的
- if fromEdbInfo.Frequency == `周度` {
- isWeekData = true
- }
- dataList, err := GetEdbDataListAllByTo(to, fromEdbInfo.Source, fromEdbInfo.SubSource, FindEdbDataListAllCond{
- EdbInfoId: fromEdbInfo.EdbInfoId,
- }, 1)
- if err != nil {
- return err
- }
- fromEdbDataMap := make(map[string]float64)
- if isWeekData {
- dataList, err = HandleDataByLinearRegression(dataList, fromEdbDataMap)
- if err != nil {
- 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
- }
- //日度转周度:日期选周五,计算上周六到本周五的日度值的加总,最新日期为最新值对应的周五。
- //日度转旬度:日期选每个旬的最后一天,计算当旬所有日度值的加总,最新日期为最新值对应当旬的最后一天。
- //日度转月度:日期选每个月最后一天,计算当月所有日度值的加总,最新日期为最新值对应当月最后一天。
- //日度转季度、年度:方法类似转月度。
- //周度转月度/季度/年度:将周度值转成日度,空值用插值法插值,计算当月/当季/当年所有值的加总,然后除以7。
- //月度转季度/年度: 当季/当年月度值相加。
- dateList := make([]time.Time, 0)
- valueMap := make(map[time.Time]float64)
- switch edbInfo.Frequency {
- 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 {
- err = tmpErr
- return
- }
- year := itemDate.Year()
- yearVal, ok := yearMap[year]
- if ok {
- yearMap[year] = item.Value + yearVal
- } else {
- yearList = append(yearList, year)
- yearMap[year] = item.Value
- }
- }
- for _, v := range yearList {
- currTime := time.Date(v, 12, 31, 0, 0, 0, 0, time.Local)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMap[v]
- lastNewDate = currTime
- }
- // 根据配置处理最新值, 1 表示均值填充
- 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, _ := initValAndMonthT.Div(daysT).Round(4).Float64()
- fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
- 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 {
- err = tmpErr
- return
- }
- year := itemDate.Year()
- var tmpK string
- if itemDate.Month() <= 6 {
- tmpK = fmt.Sprint(year, "06")
- } else {
- tmpK = fmt.Sprint(year, "12")
- }
- yearVal, ok := yearMonthMap[tmpK]
- if ok {
- yearMonthMap[tmpK] = item.Value + yearVal
- } else {
- yearMonthList = append(yearMonthList, tmpK)
- yearMonthMap[tmpK] = item.Value
- }
- }
- for _, v := range yearMonthList {
- currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- currTime = currTime.AddDate(0, 1, -1)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMonthMap[v]
- lastNewDate = currTime
- }
- // 根据配置处理最新值, 1 表示均值填充
- 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, _ := initValAndMonthT.Div(daysT).Round(4).Float64()
- fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
- 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 {
- err = tmpErr
- return
- }
- year := itemDate.Year()
- var tmpK string
- if itemDate.Month() <= 3 {
- tmpK = fmt.Sprint(year, "03")
- } else if itemDate.Month() <= 6 {
- tmpK = fmt.Sprint(year, "06")
- } else if itemDate.Month() <= 9 {
- tmpK = fmt.Sprint(year, "09")
- } else {
- tmpK = fmt.Sprint(year, "12")
- }
- yearVal, ok := yearMonthMap[tmpK]
- if ok {
- yearMonthMap[tmpK] = item.Value + yearVal
- } else {
- yearMonthList = append(yearMonthList, tmpK)
- yearMonthMap[tmpK] = item.Value
- }
- }
- for _, v := range yearMonthList {
- currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- currTime = currTime.AddDate(0, 1, -1)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMonthMap[v]
- lastNewDate = currTime
- }
- // 根据配置处理最新值, 1 表示均值填充
- 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, _ := initValAndMonthT.Div(daysT).Round(4).Float64()
- fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
- 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 {
- err = tmpErr
- return
- }
- year := itemDate.Year()
- var tmpK string
- tmpK = fmt.Sprint(year*100 + int(itemDate.Month()))
- yearVal, ok := yearMonthMap[tmpK]
- if ok {
- yearMonthMap[tmpK] = item.Value + yearVal
- fmt.Printf("%s 累加值:%f = %.6f + %.6f\n", tmpK, yearMonthMap[tmpK], item.Value, yearVal)
- } else {
- yearMonthList = append(yearMonthList, tmpK)
- yearMonthMap[tmpK] = item.Value
- }
- }
- for _, v := range yearMonthList {
- currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- currTime = currTime.AddDate(0, 1, -1)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMonthMap[v]
- lastNewDate = currTime
- }
- // 根据配置处理最新值, 1 表示均值填充
- 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, _ := initValAndMonthT.Div(daysT).Round(4).Float64()
- fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", monthDays, initVal, days, val)
- valueMap[lastNewDate] = val
- }
- case "旬度":
- tmpDateDataMap := make(map[time.Time]float64)
- tmpDateList := make([]time.Time, 0)
- for _, item := range dataList {
- itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- dayInt := itemDate.Year()*100 + int(itemDate.Month())
- var currTime time.Time
- 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.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
- }
- currTime = currTime.AddDate(0, 1, -1)
- }
- yearVal, ok := tmpDateDataMap[currTime]
- if ok {
- tmpDateDataMap[currTime] = item.Value + yearVal
- } else {
- tmpDateList = append(tmpDateList, currTime)
- tmpDateDataMap[currTime] = item.Value
- }
- }
- for _, currTime := range tmpDateList {
- dateList = append(dateList, currTime)
- valueMap[currTime] = tmpDateDataMap[currTime]
- }
- case "周度":
- tmpDateDataMap := make(map[time.Time]float64)
- tmpDateList := make([]time.Time, 0)
- for _, item := range dataList {
- itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- var currTime time.Time
- // 周六周日,这是下一个周五的数据
- if itemDate.Weekday() == 0 {
- currTime = itemDate.AddDate(0, 0, 5)
- } else if itemDate.Weekday() == 6 {
- currTime = itemDate.AddDate(0, 0, 6)
- } else {
- currTime = itemDate.AddDate(0, 0, 5-int(itemDate.Weekday()))
- }
- yearVal, ok := tmpDateDataMap[currTime]
- if ok {
- tmpDateDataMap[currTime] = item.Value + yearVal
- } else {
- tmpDateList = append(tmpDateList, currTime)
- tmpDateDataMap[currTime] = item.Value
- }
- }
- for _, currTime := range tmpDateList {
- dateList = append(dateList, currTime)
- valueMap[currTime] = tmpDateDataMap[currTime]
- }
- }
- //获取指标所有数据
- existDataList := make([]*EdbData, 0)
- sql := `SELECT * FROM %s WHERE edb_info_id=? `
- sql = fmt.Sprintf(sql, dataTableName)
- _, err = to.Raw(sql, edbInfoId).QueryRows(&existDataList)
- if err != nil {
- return err
- }
- existDataMap := make(map[string]string)
- removeDataTimeMap := make(map[string]int) //需要移除的日期数据
- for _, v := range existDataList {
- existDataMap[v.DataTime] = v.Value
- removeDataTimeMap[v.DataTime] = 1
- }
- needAddDateMap := make(map[time.Time]int)
- addSql := ` INSERT INTO ` + dataTableName + `(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
- var isAdd bool
- for _, currTime := range dateList {
- currDateStr := currTime.Format(utils.FormatDate)
- existVal, ok := existDataMap[currDateStr]
- tmpVal, ok2 := valueMap[currTime]
- if !ok2 {
- err = errors.New("数据异常,date:" + currDateStr)
- return
- }
- var saveValue string
- if isWeekData { //周度指标转的话,最后结果要除以7
- saveValue = decimal.NewFromFloat(tmpVal).Div(decimal.NewFromInt(7)).Round(4).String()
- } else {
- saveValue = decimal.NewFromFloat(tmpVal).Round(4).String()
- }
- // 如果库中已经存在该数据的话,那么就进行值的变更操作
- if ok {
- //校验待删除日期数据里面是否存在该元素,如果存在的话,那么移除该日期
- delete(removeDataTimeMap, currDateStr)
- if existVal != saveValue {
- sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
- sql = fmt.Sprintf(sql, dataTableName)
- _, err = to.Raw(sql, saveValue, edbInfoId, currDateStr).Exec()
- if err != nil {
- return
- }
- }
- continue
- }
- // 库中不存在该日期的数据
- timestamp := currTime.UnixNano() / 1e6
- timeStr := fmt.Sprintf("%d", timestamp)
- if _, existOk := needAddDateMap[currTime]; !existOk {
- addSql += GetAddSql(edbInfoIdStr, edbCode, currDateStr, timeStr, saveValue)
- isAdd = true
- }
- needAddDateMap[currTime] = 1
- }
- //删除已经不存在的指标数据(由于该指标当日的数据删除了)
- {
- removeDateList := make([]string, 0)
- for dateTime := range removeDataTimeMap {
- removeDateList = append(removeDateList, dateTime)
- }
- removeNum := len(removeDateList)
- if removeNum > 0 {
- sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (`+utils.GetOrmInReplace(removeNum)+`) `, dataTableName)
- _, err = to.Raw(sql, edbInfo.EdbInfoId, removeDateList).Exec()
- if err != nil {
- fmt.Println(reflect.TypeOf(obj).Name(), " add data ;delete Err", err.Error())
- err = fmt.Errorf("删除不存在的指标数据失败,Err:" + err.Error())
- return
- }
- }
- }
- if isAdd {
- addSql = strings.TrimRight(addSql, ",")
- _, err = to.Raw(addSql).Exec()
- if err != nil {
- fmt.Println(reflect.TypeOf(obj).Name(), " add data Err", err.Error())
- return
- }
- }
- return
- }
|