package models import ( "errors" "fmt" "github.com/beego/beego/v2/client/orm" "github.com/shopspring/decimal" "hongze/hongze_edb_lib/utils" "strconv" "strings" "time" ) // AddCalculateBp 变频 func AddCalculateBp(req *EdbInfoCalculateBatchSaveReq, fromEdbInfo *EdbInfo, edbCode, uniqueCode string, sysUserId int, sysUserRealName string) (edbInfoId int, err error) { o := orm.NewOrm() to, err := o.Begin() if err != nil { return } defer func() { if err != nil { fmt.Println("AddCalculateBp,Err:" + err.Error()) _ = to.Rollback() } else { _ = to.Commit() } }() if req.EdbInfoId <= 0 { edbInfo := new(EdbInfo) edbInfo.Source = utils.DATA_SOURCE_CALCULATE_BP edbInfo.SourceName = "变频" edbInfo.EdbCode = edbCode edbInfo.EdbName = req.EdbName edbInfo.EdbNameSource = req.EdbName edbInfo.Frequency = req.Frequency edbInfo.Unit = req.Unit edbInfo.ClassifyId = req.ClassifyId edbInfo.SysUserId = sysUserId edbInfo.SysUserRealName = sysUserRealName edbInfo.CreateTime = time.Now() edbInfo.ModifyTime = time.Now() edbInfo.UniqueCode = uniqueCode edbInfo.CalculateFormula = req.Formula edbInfo.EdbType = 2 newEdbInfoId, tmpErr := to.Insert(edbInfo) if tmpErr != nil { return edbInfoId, tmpErr } edbInfoId = int(newEdbInfoId) //关联关系 { calculateMappingItem := new(EdbInfoCalculateMapping) calculateMappingItem.CreateTime = time.Now() calculateMappingItem.ModifyTime = time.Now() calculateMappingItem.Sort = 1 calculateMappingItem.EdbCode = edbCode calculateMappingItem.EdbInfoId = 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 _, err = to.Insert(calculateMappingItem) if err != nil { return } } } else { edbInfoId = req.EdbInfoId dataTableName := GetEdbDataTableName(utils.DATA_SOURCE_CALCULATE_BP) fmt.Println("dataTableName:" + dataTableName) deleteSql := ` DELETE FROM %s WHERE edb_info_id=? ` deleteSql = fmt.Sprintf(deleteSql, dataTableName) _, err = to.Raw(deleteSql, req.EdbInfoId).Exec() if err != nil { return } } edbInfoIdStr := strconv.Itoa(edbInfoId) fmt.Println("edbInfoIdStr:" + edbInfoIdStr) //计算数据 var condition string var pars []interface{} condition += " AND edb_info_id=? " if req.EdbInfoId <= 0 { pars = append(pars, req.FromEdbInfoId) } else { pars = append(pars, fromEdbInfo.EdbInfoId) } dataList, err := GetEdbDataListAll(condition, pars, fromEdbInfo.Source, 0) if err != nil { return edbInfoId, err } addSql := ` INSERT INTO edb_data_calculate_bp(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values ` var isAdd bool existMap := make(map[string]string) dataLen := len(dataList) fmt.Println("dataLen:", dataLen) for i := 0; i < dataLen; i++ { //当期 currentItem := dataList[i] currentDate, _ := time.Parse(utils.FormatDate, currentItem.DataTime) 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.Parse(utils.FormatDate, preItem.DataTime) 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) 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) isAdd = true } existMap[existKey] = needDayStr } existKey := edbCode + currentItem.DataTime if _, ok := existMap[existKey]; !ok { currentDate, _ := time.Parse(utils.FormatDate, currentItem.DataTime) timestamp := currentDate.UnixNano() / 1e6 timestampStr := fmt.Sprintf("%d", timestamp) valStr := decimal.NewFromFloat(currentItem.Value).String() addSql += GetAddSql(edbInfoIdStr, edbCode, currentItem.DataTime, timestampStr, valStr) isAdd = true } existMap[existKey] = currentItem.DataTime } if isAdd { addSql = strings.TrimRight(addSql, ",") _, err = to.Raw(addSql).Exec() if err != nil { return edbInfoId, err } } return } // EditCalculateBp 修改变频数据 func EditCalculateBp(req *EdbInfoCalculateBatchEditReq, fromEdbInfo *EdbInfo, edbCode string) (edbInfoId int, err error) { edbInfoId = req.EdbInfoId o := orm.NewOrm() to, err := o.Begin() if err != nil { return } defer func() { if err != nil { fmt.Println("EditCalculateBp,Err:" + err.Error()) _ = to.Rollback() } else { _ = to.Commit() } }() //修改指标信息 sql := ` UPDATE edb_info SET edb_name =?, edb_name_source=?, frequency = ?, unit = ?, classify_id = ?, modify_time = NOW() WHERE edb_info_id = ? ` _, err = o.Raw(sql, req.EdbName, req.EdbName, req.Frequency, req.Unit, req.ClassifyId, edbInfoId).Exec() if err != nil { return } var existCondition string var existPars []interface{} existCondition += " AND edb_info_id=? " existPars = append(existPars, edbInfoId) existCondition += " AND from_edb_info_id=? " existPars = append(existPars, req.FromEdbInfoId) //判断计算指标是否被更换 count, err := GetEdbInfoCalculateCountByCondition(existCondition, existPars) if err != nil { err = errors.New("判断指标是否改变失败,Err:" + err.Error()) return } if count <= 0 { //删除指标关联计算指标 //sql := ` DELETE FROM edb_info_calculate_bp WHERE edb_info_id = ? ` //_, err = o.Raw(sql, edbInfoId).Exec() //if err != nil { // return //} //calculateItem := new(EdbInfoCalculateHbz) //calculateItem.CreateTime = time.Now() //calculateItem.ModifyTime = time.Now() //calculateItem.Sort = 1 //calculateItem.EdbCode = edbCode //calculateItem.EdbInfoId = edbInfoId //calculateItem.FromEdbInfoId = fromEdbInfo.EdbInfoId //calculateItem.FromEdbCode = fromEdbInfo.EdbCode //calculateItem.FromEdbName = fromEdbInfo.EdbName //calculateItem.FromSource = fromEdbInfo.Source //calculateItem.FromSourceName = fromEdbInfo.SourceName // //_, err = o.Insert(calculateItem) //if err != nil { // return //} //删除,计算指标关联的,基础指标的关联关系 sql = ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? ` _, err = o.Raw(sql, edbInfoId).Exec() if err != nil { err = errors.New("删除计算指标关联关系失败,Err:" + err.Error()) return } //清空原有数据 sql = ` DELETE FROM edb_data_calculate_bp WHERE edb_info_id = ? ` _, err = o.Raw(sql, edbInfoId).Exec() if err != nil { return edbInfoId, err } //关联关系 { calculateMappingItem := new(EdbInfoCalculateMapping) calculateMappingItem.CreateTime = time.Now() calculateMappingItem.ModifyTime = time.Now() calculateMappingItem.Sort = 1 calculateMappingItem.EdbCode = edbCode calculateMappingItem.EdbInfoId = 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 = utils.DATA_SOURCE_CALCULATE_BP calculateMappingItem.SourceName = "变频" _, err = o.Insert(calculateMappingItem) if err != nil { return } } edbInfoIdStr := strconv.Itoa(edbInfoId) fmt.Println("edbInfoIdStr:" + edbInfoIdStr) //计算数据 var condition string var pars []interface{} condition += " AND edb_info_id=? " pars = append(pars, req.FromEdbInfoId) fmt.Println("EdbInfoId:", req.FromEdbInfoId) dataList, err := GetEdbDataListAll(condition, pars, fromEdbInfo.Source, 0) if err != nil { return edbInfoId, err } addSql := ` INSERT INTO edb_data_calculate_bp(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values ` var isAdd bool existMap := make(map[string]string) dataLen := len(dataList) for i := 0; i < dataLen; i++ { //当期 currentItem := dataList[i] currentDate, _ := time.Parse(utils.FormatDate, currentItem.DataTime) 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.Parse(utils.FormatDate, preItem.DataTime) day = int(currentDate.Sub(preDate).Hours() / float64(24)) } } 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) } existMap[existKey] = needDayStr } existKey := edbCode + currentItem.DataTime if _, ok := existMap[existKey]; !ok { currentDate, _ := time.Parse(utils.FormatDate, currentItem.DataTime) timestamp := currentDate.UnixNano() / 1e6 timestampStr := fmt.Sprintf("%d", timestamp) valStr := decimal.NewFromFloat(currentItem.Value).String() addSql += GetAddSql(edbInfoIdStr, edbCode, currentItem.DataTime, timestampStr, valStr) } existMap[existKey] = currentItem.DataTime } if isAdd { addSql = strings.TrimRight(addSql, ",") _, err = o.Raw(addSql).Exec() if err != nil { return edbInfoId, err } } } return } // RefreshAllCalculateBpBak 刷新所有变频数据 func RefreshAllCalculateBpBak(edbInfoId, source int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string) (err error) { o := orm.NewOrm() to, err := o.Begin() if err != nil { return } defer func() { if err != nil { fmt.Println("RefreshAllCalculateBp,Err:" + err.Error()) _ = to.Rollback() } else { _ = to.Commit() } }() if err != nil { return } edbInfoIdStr := strconv.Itoa(edbInfoId) //计算数据 //计算数据 var condition string var pars []interface{} condition += " AND edb_info_id=? " pars = append(pars, fromEdbInfo.EdbInfoId) if startDate != "" { condition += " AND data_time>=? " pars = append(pars, startDate) } if endDate != "" { condition += " AND data_time<=? " pars = append(pars, endDate) } dataList, err := GetEdbDataListAll(condition, pars, fromEdbInfo.Source, 0) if err != nil { return err } var dateArr []string dataMap := make(map[string]*EdbInfoSearchData) for _, v := range dataList { dateArr = append(dateArr, v.DataTime) dataMap[v.DataTime] = v } fmt.Println("source:", source) //获取指标所有数据 existDataList := make([]*EdbData, 0) dataTableName := GetEdbDataTableName(source) fmt.Println("dataTableName:", dataTableName) sql := `SELECT * FROM %s WHERE edb_info_id=? ` sql = fmt.Sprintf(sql, dataTableName) _, err = o.Raw(sql, edbInfoId).QueryRows(&existDataList) if err != nil { return err } existDataMap := make(map[string]string) for _, v := range existDataList { existDataMap[v.DataTime] = v.Value } fmt.Println("existDataMap:", existDataMap) addSql := ` INSERT INTO edb_data_calculate_bp(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values ` var isAdd bool existMap := make(map[string]string) dataLen := len(dataList) for i := 0; i < dataLen; i++ { //当期 currentItem := dataList[i] currentDate, _ := time.Parse(utils.FormatDate, currentItem.DataTime) 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.Parse(utils.FormatDate, preItem.DataTime) day = int(currentDate.Sub(preDate).Hours() / float64(24)) } } 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() if existVal, ok := existDataMap[needDayStr]; !ok { addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, valStr) } else { if existVal != valStr { sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? ` sql = fmt.Sprintf(sql, dataTableName) _, err = o.Raw(sql, valStr, edbInfoId, needDay).Exec() if err != nil { return err } } } } existMap[existKey] = needDayStr } existKey := edbCode + currentItem.DataTime if _, ok := existMap[existKey]; !ok { currentDate, _ := time.Parse(utils.FormatDate, currentItem.DataTime) timestamp := currentDate.UnixNano() / 1e6 timestampStr := fmt.Sprintf("%d", timestamp) valStr := decimal.NewFromFloat(currentItem.Value).String() if existVal, ok := existDataMap[currentItem.DataTime]; !ok { addSql += GetAddSql(edbInfoIdStr, edbCode, currentItem.DataTime, timestampStr, valStr) } else { if existVal != valStr { sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? ` sql = fmt.Sprintf(sql, dataTableName) _, err = o.Raw(sql, valStr, edbInfoId, currentItem.DataTime).Exec() if err != nil { return err } } } } existMap[existKey] = currentItem.DataTime } if isAdd { addSql = strings.TrimRight(addSql, ",") _, err = o.Raw(addSql).Exec() if err != nil { return err } } return } func RefreshAllCalculateBp(edbInfoId, source int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string) (err error) { o := orm.NewOrm() to, err := o.Begin() if err != nil { return } defer func() { if err != nil { fmt.Println("RefreshAllCalculateBp,Err:" + err.Error()) _ = to.Rollback() } else { _ = to.Commit() } }() if err != nil { return } edbInfoIdStr := strconv.Itoa(edbInfoId) //计算数据 //计算数据 var condition string var pars []interface{} condition += " AND edb_info_id=? " pars = append(pars, fromEdbInfo.EdbInfoId) //if startDate != "" { // condition += " AND data_time>=? " // pars = append(pars, startDate) //} //if endDate != "" { // condition += " AND data_time<=? " // pars = append(pars, endDate) //} //获取来源指标的数据 dataList, err := GetEdbDataListAll(condition, pars, fromEdbInfo.Source, 1) if err != nil { return err } var dateArr []string dataMap := make(map[string]*EdbInfoSearchData) fromDataMap := make(map[string]float64) //来源指指标数据 for _, v := range dataList { dateArr = append(dateArr, v.DataTime) dataMap[v.DataTime] = v fromDataMap[v.DataTime] = v.Value } fmt.Println("source:", source) //获取变频指标所有数据 existDataList, err := GetAllEdbDataList(edbInfoId, source) if err != nil { return } //计算指标的map existDataMap := make(map[string]*EdbData, 0) addSql := ` INSERT INTO edb_data_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) //待删除的日期 removeDateList := make([]string, 0) if len(existDataList) > 0 { //第一个已经入库的日期 firstExistDataTimeStr := existDataList[0].DataTime //计算指标数据第一条的日期字符串 if len(dateArr) > 0 { firstFromDataTimeStr := dateArr[0] //来源数据第一条的日期字符串 firstExistDataTime, _ := time.Parse(utils.FormatDate, firstExistDataTimeStr) //计算指标数据第一条的日期(time类型) firstFromDataTime, _ := time.Parse(utils.FormatDate, firstFromDataTimeStr) //来源数据第一条的日期(time类型) nowDateStr := time.Now().Format(utils.FormatDate) //当天日期字符串 nowDate, _ := time.ParseInLocation(utils.FormatDate, nowDateStr, firstFromDataTime.Location()) //当天日期(time类型) 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 { //如果计算指标第一条数据的开始时间 晚于 来源指标的第一条开始时间,那么需要对两个时间之间的计算指标数据做 新增处理 if v == firstExistDataTimeStr { if tmpLastValue, ok := fromDataMap[v]; ok { //来源指标当天的数据 lastValue = tmpLastValue lastValueStr = decimal.NewFromFloat(lastValue).String() } break } currentDate, _ := time.Parse(utils.FormatDate, v) timestamp := currentDate.UnixNano() / 1e6 timestampStr := fmt.Sprintf("%d", timestamp) addSql += GetAddSql(edbInfoIdStr, edbCode, v, timestampStr, lastValueStr) isAdd = true } } } //第二步 剩余数据每天修改 day := int(nowDate.Sub(firstExistDataTime).Hours() / float64(24)) //第二步: 已经入库的数据处理 for _, v := range existDataList { existDataMap[v.DataTime] = v } 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() } //如果对应的值不匹配 if tmpExistData.Value != lastValueStr { err = ModifyEdbDataById(source, tmpExistData.EdbDataId, lastValueStr) if err != nil { return err } } } else { timestamp := needDay.UnixNano() / 1e6 timestampStr := fmt.Sprintf("%d", timestamp) addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, lastValueStr) isAdd = true } } } else { //如果没有来源指标数据,那么已经入库的计算指标数据需要全部删除 tableName := GetEdbDataTableName(source) sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ?`, tableName) _, err = o.Raw(sql, edbInfoId).Exec() if err != nil { err = fmt.Errorf("删除所有的变频指标数据失败,Err:" + err.Error()) return } //for _, v := range existDataList { // removeDateList = append(removeDateList, v.DataTime) //} } } // 删除不需要的指标数据 if len(removeDateList) > 0 { removeDateStr := strings.Join(removeDateList, `","`) removeDateStr = `"` + removeDateStr + `"` //如果拼接指标变更了,那么需要删除所有的指标数据 tableName := GetEdbDataTableName(source) sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (%s) `, tableName, removeDateStr) _, err = o.Raw(sql, edbInfoId).Exec() if err != nil { err = fmt.Errorf("删除不存在的变频指标数据失败,Err:" + err.Error()) return } } if isAdd { addSql = strings.TrimRight(addSql, ",") _, err = o.Raw(addSql).Exec() } return }