123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821 |
- package models
- import (
- "errors"
- "eta_gn/eta_index_lib/global"
- "eta_gn/eta_index_lib/utils"
- "fmt"
- "github.com/shopspring/decimal"
- "gorm.io/gorm"
- "math"
- "strings"
- "time"
- )
- // EdbDataCalculateNhcc 拟合残差数据结构体
- type EdbDataCalculateNhcc struct {
- EdbDataId int `gorm:"primaryKey;autoIncrement;column:edb_data_id"`
- EdbInfoId int `gorm:"column:edb_info_id" description:"指标ID"`
- EdbCode string `gorm:"column:edb_code" description:"指标编码"`
- DataTime string `gorm:"column:data_time" description:"数据日期"`
- Value float64 `gorm:"column:value" description:"数据值"`
- Status int `gorm:"column:status" description:"状态"`
- CreateTime time.Time `gorm:"column:create_time" description:"创建时间"`
- ModifyTime time.Time `gorm:"column:modify_time" description:"修改时间"`
- DataTimestamp int64 `gorm:"column:data_timestamp" description:"数据日期时间戳"`
- }
- // NhccDate 拟合残差的开始、结束日期
- type NhccDate struct {
- StartDate time.Time
- EndDate time.Time
- }
- // AddCalculateNhcc 新增拟合残差数据
- func AddCalculateNhcc(req *EdbInfoCalculateBatchSaveReq, firstEdbInfo, secondEdbInfo *EdbInfo, edbCode, uniqueCode string, nhccDate NhccDate, sysUserId int, sysUserRealName string) (edbInfo *EdbInfo, err error, errMsg string) {
- to := global.DEFAULT_DmSQL.Begin()
- defer func() {
- if err != nil {
- to.Rollback()
- } else {
- to.Commit()
- }
- }()
- if req.EdbInfoId <= 0 {
- edbInfo = &EdbInfo{
- SourceName: "拟合残差",
- Source: utils.DATA_SOURCE_CALCULATE_NHCC,
- EdbCode: edbCode,
- EdbName: req.EdbName,
- EdbNameSource: req.EdbName,
- Frequency: req.Frequency,
- Unit: req.Unit,
- StartDate: firstEdbInfo.StartDate,
- EndDate: firstEdbInfo.EndDate,
- ClassifyId: req.ClassifyId,
- SysUserId: sysUserId,
- SysUserRealName: sysUserRealName,
- UniqueCode: uniqueCode,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- CalculateFormula: req.Formula,
- EdbNameEn: req.EdbName,
- UnitEn: req.Unit,
- EdbType: 2,
- Sort: GetAddEdbMaxSortByClassifyId(req.ClassifyId, utils.EDB_INFO_TYPE),
- }
- tmpErr := to.Create(edbInfo).Error
- if tmpErr != nil {
- err = tmpErr
- return
- }
- } else {
- edbInfo, err = GetEdbInfoById(req.EdbInfoId)
- if err != nil {
- return
- }
- //查询
- tmpEdbInfo, tmpErr := GetEdbInfoById(edbInfo.EdbInfoId)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- tmpEdbInfo.EdbName = req.EdbName
- tmpEdbInfo.ClassifyId = req.ClassifyId
- tmpEdbInfo.Frequency = req.Frequency
- tmpEdbInfo.Unit = req.Unit
- tmpEdbInfo.CalculateFormula = req.Formula
- edbInfo = tmpEdbInfo
- //删除指标数据
- dataTableName := GetEdbDataTableName(utils.DATA_SOURCE_CALCULATE_NHCC, utils.DATA_SUB_SOURCE_EDB)
- fmt.Println("dataTableName:" + dataTableName)
- deleteSql := ` DELETE FROM %s WHERE edb_info_id=? `
- deleteSql = fmt.Sprintf(deleteSql, dataTableName)
- err = to.Exec(deleteSql, req.EdbInfoId).Error
- if err != nil {
- return
- }
- //删除指标关系
- sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id=? `
- err = to.Exec(sql, edbInfo.EdbInfoId).Error
- }
- //关联关系
- var existItemA, existItemB *EdbInfoCalculateMapping
- //第一个指标
- {
- existItemA = &EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: edbInfo.EdbInfoId,
- Source: edbInfo.Source,
- SourceName: edbInfo.SourceName,
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: firstEdbInfo.EdbInfoId,
- FromEdbCode: firstEdbInfo.EdbCode,
- FromEdbName: firstEdbInfo.EdbName,
- FromSource: firstEdbInfo.Source,
- FromSourceName: firstEdbInfo.SourceName,
- FromTag: "A",
- MoveValue: req.EdbInfoIdArr[0].MoveValue,
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- FromSubSource: firstEdbInfo.SubSource,
- }
- tmpErr := to.Create(existItemA).Error
- if tmpErr != nil {
- err = tmpErr
- return
- }
- }
- //第二个指标
- {
- existItemB = &EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: edbInfo.EdbInfoId,
- Source: edbInfo.Source,
- SourceName: edbInfo.SourceName,
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: secondEdbInfo.EdbInfoId,
- FromEdbCode: secondEdbInfo.EdbCode,
- FromEdbName: secondEdbInfo.EdbName,
- FromSource: secondEdbInfo.Source,
- FromSourceName: secondEdbInfo.SourceName,
- FromTag: "B",
- MoveValue: req.EdbInfoIdArr[1].MoveValue,
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- FromSubSource: secondEdbInfo.SubSource,
- }
- tmpErr := to.Create(existItemB).Error
- if tmpErr != nil {
- err = tmpErr
- return
- }
- }
- //拼接数据
- err, errMsg = refreshAllCalculateNhcc(to, edbInfo, existItemA, existItemB, nhccDate)
- return
- }
- // EditCalculateNhcc 编辑拟合残差数据
- func EditCalculateNhcc(req *EdbInfoCalculateBatchEditReq, edbInfo, firstEdbInfo, secondEdbInfo *EdbInfo, nhccDate NhccDate) (err error, errMsg string) {
- to := global.DEFAULT_DmSQL.Begin()
- defer func() {
- if err != nil {
- to.Rollback()
- } else {
- to.Commit()
- }
- }()
- nowEdbInfo := *edbInfo // 现在的指标信息
- //修改指标信息
- edbInfo.EdbName = req.EdbName
- edbInfo.EdbNameSource = req.EdbName
- edbInfo.Frequency = req.Frequency
- edbInfo.Unit = req.Unit
- edbInfo.ClassifyId = req.ClassifyId
- edbInfo.CalculateFormula = req.Formula
- edbInfo.EdbNameEn = req.EdbNameEn
- edbInfo.UnitEn = req.UnitEn
- edbInfo.ModifyTime = time.Now()
- err = to.Model(edbInfo).Select([]string{"EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "CalculateFormula", "ModifyTime", "EdbNameEn", "UnitEn"}).Updates(edbInfo).Error
- if err != nil {
- return
- }
- var existCondition string
- var existPars []interface{}
- existCondition += " AND edb_info_id=? "
- existPars = append(existPars, edbInfo.EdbInfoId)
- //查询出所有的关联指标
- existList, err := GetEdbInfoCalculateListByCondition(existCondition, existPars)
- if err != nil {
- err = fmt.Errorf("判断指标是否改变失败,Err:" + err.Error())
- return
- }
- var existItemA, existItemB *EdbInfoCalculateMapping
- for _, existItem := range existList {
- if existItem.FromTag == "A" {
- existItemA = existItem
- } else if existItem.FromTag == "B" {
- existItemB = existItem
- }
- }
- // 是否需要删除数据重新计算
- isNeedCalculateData := false
- // 如果截止日期变更,那么需要重新计算
- if req.Formula != nowEdbInfo.CalculateFormula {
- isNeedCalculateData = true
- }
- var isDeleteA, isDeleteB bool
- // 如果指标变了,那么需要删除关系
- {
- if existItemA.FromEdbInfoId != firstEdbInfo.EdbInfoId {
- //删除之前的A指标关联关系
- sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? and from_edb_info_id = ?`
- err = to.Exec(sql, edbInfo.EdbInfoId, existItemA.FromEdbInfoId).Error
- if err != nil {
- err = fmt.Errorf("删除指标A关联关系失败,Err:" + err.Error())
- return
- }
- isDeleteA = true
- }
- //并重新计算
- if existItemB.FromEdbInfoId != secondEdbInfo.EdbInfoId {
- //删除之前的B指标关联关系
- sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? and from_edb_info_id = ?`
- err = to.Exec(sql, edbInfo.EdbInfoId, existItemB.FromEdbInfoId).Error
- if err != nil {
- err = fmt.Errorf("删除指标B关联关系失败,Err:" + err.Error())
- return
- }
- isDeleteB = true
- }
- }
- //第一个指标数据
- {
- // 如果指标变了,那么需要删除关系,并重新计算
- if isDeleteA {
- //添加新的指标关系
- {
- existItemA = &EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: edbInfo.EdbInfoId,
- Source: edbInfo.Source,
- SourceName: edbInfo.SourceName,
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: firstEdbInfo.EdbInfoId,
- FromEdbCode: firstEdbInfo.EdbCode,
- FromEdbName: firstEdbInfo.EdbName,
- FromSource: firstEdbInfo.Source,
- FromSourceName: firstEdbInfo.SourceName,
- FromTag: "A",
- MoveValue: req.EdbInfoIdArr[0].MoveValue,
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- FromSubSource: firstEdbInfo.SubSource,
- }
- tmpErr := to.Create(existItemA).Error
- if tmpErr != nil {
- err = tmpErr
- return
- }
- isNeedCalculateData = true
- }
- } else if existItemA.MoveValue != req.EdbInfoIdArr[0].MoveValue {
- // 如果平移天数不一致的话,
- existItemA.ModifyTime = time.Now()
- existItemA.MoveValue = req.EdbInfoIdArr[0].MoveValue
- err = to.Model(existItemA).Select([]string{"ModifyTime", "MoveValue"}).Updates(existItemA).Error
- if err != nil {
- return
- }
- isNeedCalculateData = true
- }
- }
- //第二个指标数据
- {
- // 如果指标变了,那么需要删除关系,并重新计算
- if isDeleteB {
- // 添加新的指标关联关系
- existItemB = &EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: edbInfo.EdbInfoId,
- Source: edbInfo.Source,
- SourceName: edbInfo.SourceName,
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: secondEdbInfo.EdbInfoId,
- FromEdbCode: secondEdbInfo.EdbCode,
- FromEdbName: secondEdbInfo.EdbName,
- FromSource: secondEdbInfo.Source,
- FromSourceName: secondEdbInfo.SourceName,
- FromTag: "B",
- MoveValue: req.EdbInfoIdArr[1].MoveValue,
- Sort: 2,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- FromSubSource: secondEdbInfo.SubSource,
- }
- tmpErr := to.Create(existItemB).Error
- if tmpErr != nil {
- err = tmpErr
- return
- }
- isNeedCalculateData = true
- } else if existItemB.MoveValue != req.EdbInfoIdArr[1].MoveValue {
- // 如果平移天数不一致的话,
- existItemB.ModifyTime = time.Now()
- existItemB.MoveValue = req.EdbInfoIdArr[1].MoveValue
- err = to.Model(existItemB).Select([]string{"ModifyTime", "MoveValue"}).Updates(existItemB).Error
- if err != nil {
- return
- }
- isNeedCalculateData = true
- }
- }
- // 如果需要重新计算,那么先删除所有的指标数据,然后再重新计算
- if isNeedCalculateData {
- // 删除之前所有的指标数据
- tableName := GetEdbDataTableName(edbInfo.Source, edbInfo.SubSource)
- sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? `, tableName)
- err = to.Exec(sql, edbInfo.EdbInfoId).Error
- if err != nil {
- err = fmt.Errorf("删除历史数据失败,Err:" + err.Error())
- return
- }
- err, errMsg = refreshAllCalculateNhcc(to, edbInfo, existItemA, existItemB, nhccDate)
- }
- return
- }
- // RefreshAllCalculateNhcc 刷新所有 拟合残差 数据
- func RefreshAllCalculateNhcc(edbInfo *EdbInfo) (err error) {
- to := global.DEFAULT_DmSQL.Begin()
- defer func() {
- if err != nil {
- to.Rollback()
- } else {
- to.Commit()
- }
- }()
- //查询关联指标信息
- var existCondition string
- var existPars []interface{}
- existCondition += " AND edb_info_id=? "
- existPars = append(existPars, edbInfo.EdbInfoId)
- existList, err := GetEdbInfoCalculateListByCondition(existCondition, existPars)
- if err != nil {
- err = fmt.Errorf("判断指标是否改变失败,Err:" + err.Error())
- return
- }
- var existItemA, existItemB *EdbInfoCalculateMapping
- for _, existItem := range existList {
- if existItem.FromTag == "A" {
- existItemA = existItem
- } else if existItem.FromTag == "B" {
- existItemB = existItem
- }
- }
- timeList := strings.Split(edbInfo.CalculateFormula, ",")
- startDate, err := time.ParseInLocation(utils.FormatDate, timeList[0], time.Local)
- if err != nil {
- return
- }
- endDate, err := time.ParseInLocation(utils.FormatDate, timeList[1], time.Local)
- if err != nil {
- return
- }
- if endDate.Sub(startDate).Hours() < 48 {
- return
- }
- nhccDate := NhccDate{
- StartDate: startDate,
- EndDate: endDate,
- }
- // 刷新数据
- err, _ = refreshAllCalculateNhcc(to, edbInfo, existItemA, existItemB, nhccDate)
- return
- }
- // refreshAllCalculateNhcc 刷新所有 拟合残差 数据
- func refreshAllCalculateNhcc(to *gorm.DB, edbInfo *EdbInfo, existItemA, existItemB *EdbInfoCalculateMapping, nhccDate NhccDate) (err error, errMsg string) {
- //查询当前指标现有的数据
- var condition string
- var pars []interface{}
- condition += " AND edb_info_id=? "
- pars = append(pars, edbInfo.EdbInfoId)
- //获取年化指标所有数据
- existDataList, err := GetAllEdbDataListByTo(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource)
- //var dataList []*EdbDataCalculateNhcc
- //sql := ` SELECT * FROM edb_data_calculate_nhcc WHERE edb_info_id=? ORDER BY data_time DESC `
- //err = to.Raw(sql, edbInfo.EdbInfoId).Scan(&dataList).Error
- if err != nil {
- return
- }
- var dateArr []string
- dataMap := make(map[string]*EdbData)
- removeDataTimeMap := make(map[string]int) //需要移除的日期数据
- for _, v := range existDataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime] = v
- removeDataTimeMap[v.DataTime] = 1
- }
- addDataList := make([]*EdbDataCalculateNhcc, 0)
- aDataList, secondDataList, aDataMap, bDataMap, nhccDate, err, errMsg := getNhccData(to, existItemA, existItemB, nhccDate)
- if err != nil {
- return
- }
- // 计算线性方程公式
- var a, b float64
- {
- coordinateData := make([]utils.Coordinate, 0)
- for i := nhccDate.StartDate; i.Before(nhccDate.EndDate) || i.Equal(nhccDate.EndDate); i = i.AddDate(0, 0, 1) {
- dateStr := i.Format(utils.FormatDate)
- xValue, ok := aDataMap[dateStr]
- if !ok {
- errMsg = "指标A日期:" + dateStr + "数据异常,导致计算线性方程公式失败"
- err = errors.New(errMsg)
- return
- }
- yValue, ok := bDataMap[dateStr]
- if !ok {
- errMsg = "指标B日期:" + dateStr + "数据异常,导致计算线性方程公式失败"
- err = errors.New(errMsg)
- return
- }
- tmpCoordinate := utils.Coordinate{
- X: xValue,
- Y: yValue,
- }
- coordinateData = append(coordinateData, tmpCoordinate)
- }
- a, b = utils.GetLinearResult(coordinateData)
- }
- if math.IsNaN(a) || math.IsNaN(b) {
- errMsg = "线性方程公式生成失败"
- err = errors.New(errMsg)
- return
- }
- //fmt.Println("a:", a, ";======b:", b)
- //计算B’
- newBDataMap := make(map[string]float64)
- {
- //B’=aA+b
- aDecimal := decimal.NewFromFloat(a)
- bDecimal := decimal.NewFromFloat(b)
- for _, aData := range aDataList {
- xDecimal := decimal.NewFromFloat(aData.Value)
- val, _ := aDecimal.Mul(xDecimal).Add(bDecimal).Round(4).Float64()
- newBDataMap[aData.DataTime] = val
- }
- }
- // Delta=B-B‘
- for _, bData := range secondDataList {
- currDate := bData.DataTime
- //校验待删除日期数据里面是否存在该元素,如果存在的话,那么移除该元素
- if _, ok := removeDataTimeMap[currDate]; ok {
- delete(removeDataTimeMap, currDate)
- }
- b2Val, ok := newBDataMap[currDate]
- if !ok {
- // 如果B`不存在数据,那么就退出当前循环
- continue
- }
- bDecimal := decimal.NewFromFloat(bData.Value)
- b2Decimal := decimal.NewFromFloat(b2Val)
- valDeci := bDecimal.Sub(b2Decimal)
- val, _ := valDeci.Round(4).Float64()
- // 判断之前有没有该数据
- existData, ok := dataMap[currDate]
- if !ok { //不存在那么就添加吧
- currentDate, _ := time.ParseInLocation(utils.FormatDate, currDate, time.Local)
- timestamp := currentDate.UnixNano() / 1e6
- edbDataNhcc := &EdbDataCalculateNhcc{
- EdbInfoId: edbInfo.EdbInfoId,
- EdbCode: edbInfo.EdbCode,
- DataTime: currDate,
- Value: val,
- Status: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- DataTimestamp: timestamp,
- }
- addDataList = append(addDataList, edbDataNhcc)
- } else {
- // 如果有的话,还需要判断值是否一致,一致则不处理,不一致则修改
- tmpExistVal, tmpErr := decimal.NewFromString(existData.Value)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- if !tmpExistVal.Equal(valDeci) {
- err = ModifyEdbDataById(edbInfo.Source, edbInfo.SubSource, existData.EdbDataId, valDeci.String())
- if err != nil {
- return
- }
- }
- }
- }
- //删除已经不存在的拟合残差指标数据(由于该指标当日的数据删除了)
- {
- removeDateList := make([]string, 0)
- for dateTime := range removeDataTimeMap {
- removeDateList = append(removeDateList, dateTime)
- }
- removeNum := len(removeDateList)
- if removeNum > 0 {
- //如果拼接指标变更了,那么需要删除所有的指标数据
- tableName := GetEdbDataTableName(edbInfo.Source, edbInfo.SubSource)
- sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (`+utils.GetOrmInReplace(removeNum)+`) `, tableName)
- err = to.Exec(sql, edbInfo.EdbInfoId, removeDateList).Error
- if err != nil {
- err = fmt.Errorf("删除不存在的拟合残差指标数据失败,Err:" + err.Error())
- return
- }
- }
- }
- //数据入库
- if len(addDataList) > 0 {
- tmpAddDataList := make([]*EdbDataCalculateNhcc, 0)
- i := 0
- for _, v := range addDataList {
- tmpAddDataList = append(tmpAddDataList, v)
- i++
- if i >= 500 {
- err = to.CreateInBatches(tmpAddDataList, 500).Error
- if err != nil {
- return
- }
- i = 0
- tmpAddDataList = make([]*EdbDataCalculateNhcc, 0)
- }
- }
- if len(tmpAddDataList) > 0 {
- err = to.CreateInBatches(tmpAddDataList, 500).Error
- if err != nil {
- return
- }
- }
- }
- return
- }
- // CalculateComputeCorrelation 计算相关性结果
- func CalculateComputeCorrelation(req *EdbInfoCalculateBatchSaveReq, firstEdbInfo, secondEdbInfo *EdbInfo, nhccDate NhccDate) (a, b, r float64, err error, errMsg string) {
- var existItemA, existItemB *EdbInfoCalculateMapping
- //第一个指标
- {
- existItemA = &EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- FromEdbInfoId: firstEdbInfo.EdbInfoId,
- FromEdbCode: firstEdbInfo.EdbCode,
- FromEdbName: firstEdbInfo.EdbName,
- FromSource: firstEdbInfo.Source,
- FromSourceName: firstEdbInfo.SourceName,
- FromTag: "A",
- MoveValue: req.EdbInfoIdArr[0].MoveValue,
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- FromSubSource: firstEdbInfo.SubSource,
- }
- }
- //第二个指标
- {
- existItemB = &EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- FromEdbInfoId: secondEdbInfo.EdbInfoId,
- FromEdbCode: secondEdbInfo.EdbCode,
- FromEdbName: secondEdbInfo.EdbName,
- FromSource: secondEdbInfo.Source,
- FromSourceName: secondEdbInfo.SourceName,
- FromTag: "B",
- MoveValue: req.EdbInfoIdArr[1].MoveValue,
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- FromSubSource: secondEdbInfo.SubSource,
- }
- }
- to := global.DEFAULT_DmSQL.Begin()
- defer func() {
- if err != nil {
- to.Rollback()
- } else {
- to.Commit()
- }
- }()
- // 获取相关数据
- _, _, aDataMap, bDataMap, nhccDate, err, errMsg := getNhccData(to, existItemA, existItemB, nhccDate)
- if err != nil {
- return
- }
- // 计算相关性
- coordinateData := make([]utils.Coordinate, 0)
- for i := nhccDate.StartDate; i.Before(nhccDate.EndDate) || i.Equal(nhccDate.EndDate); i = i.AddDate(0, 0, 1) {
- dateStr := i.Format(utils.FormatDate)
- xValue, ok := aDataMap[dateStr]
- if !ok {
- errMsg = "指标A日期:" + dateStr + "数据异常,导致计算线性方程公式失败"
- err = errors.New(errMsg)
- return
- }
- yValue, ok := bDataMap[dateStr]
- if !ok {
- errMsg = "指标B日期:" + dateStr + "数据异常,导致计算线性方程公式失败"
- err = errors.New(errMsg)
- return
- }
- tmpCoordinate := utils.Coordinate{
- X: xValue,
- Y: yValue,
- }
- coordinateData = append(coordinateData, tmpCoordinate)
- }
- // 计算线性方程公式
- a, b = utils.GetLinearResult(coordinateData)
- // 计算相关性
- r = utils.ComputeCorrelation(coordinateData)
- return
- }
- // getNhccData 获取拟合残差需要的数据
- func getNhccData(to *gorm.DB, existItemA, existItemB *EdbInfoCalculateMapping, nhccDate NhccDate) (aDataList []EdbInfoSearchData, secondDataList []*EdbInfoSearchData, aDataMap, bDataMap map[string]float64, newNhccDate NhccDate, err error, errMsg string) {
- newNhccDate = nhccDate
- //第一个指标
- aDataList = make([]EdbInfoSearchData, 0)
- aDataMap = make(map[string]float64)
- {
- //第一个指标的数据列表
- firstDataList, tmpErr := GetEdbDataListAllByTo(to, existItemA.FromSource, existItemA.FromSubSource, FindEdbDataListAllCond{
- EdbInfoId: existItemA.FromEdbInfoId,
- }, 0)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- aDataList, aDataMap = handleNhccData(firstDataList, existItemA.MoveValue)
- }
- //第二个指标
- bDataList := make([]EdbInfoSearchData, 0)
- secondDataList = make([]*EdbInfoSearchData, 0)
- bDataMap = make(map[string]float64)
- {
- //第二个指标的数据列表
- secondDataList, err = GetEdbDataListAllByTo(to, existItemB.FromSource, existItemB.FromSubSource, FindEdbDataListAllCond{
- EdbInfoId: existItemB.FromEdbInfoId,
- }, 0)
- if err != nil {
- return
- }
- bDataList, bDataMap = handleNhccData(secondDataList, existItemB.MoveValue)
- }
- if len(aDataList) <= 0 {
- errMsg = "指标A没有数据"
- err = errors.New(errMsg)
- return
- }
- if len(bDataList) <= 0 {
- errMsg = "指标B没有数据"
- err = errors.New(errMsg)
- return
- }
- // 拟合残差计算的开始日期判断
- {
- startAData := aDataList[0]
- tmpStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, startAData.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- // 如果A指标的最新数据日期晚于拟合残差的结束日期,那么就用A指标的最新数据日期
- if tmpStartDate.After(newNhccDate.StartDate) {
- newNhccDate.StartDate = tmpStartDate
- }
- startBData := bDataList[0]
- tmpStartDate, tmpErr = time.ParseInLocation(utils.FormatDate, startBData.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- // 如果B指标的最新数据日期晚于拟合残差的结束日期,那么就用A指标的最新数据日期
- if tmpStartDate.After(newNhccDate.StartDate) {
- newNhccDate.StartDate = tmpStartDate
- }
- }
- // 拟合残差计算的结束日期判断
- {
- endAData := aDataList[len(aDataList)-1]
- tmpEndDate, tmpErr := time.ParseInLocation(utils.FormatDate, endAData.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- // 如果A指标的最新数据日期早于拟合残差的结束日期,那么就用A指标的最新数据日期
- if tmpEndDate.Before(newNhccDate.EndDate) {
- newNhccDate.EndDate = tmpEndDate
- }
- endBData := bDataList[len(bDataList)-1]
- tmpEndDate, tmpErr = time.ParseInLocation(utils.FormatDate, endBData.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- // 如果B指标的最新数据日期早于拟合残差的结束日期,那么就用A指标的最新数据日期
- if tmpEndDate.Before(newNhccDate.EndDate) {
- newNhccDate.EndDate = tmpEndDate
- }
- }
- return
- }
- // handleNhccData 处理拟合残差需要的数据
- func handleNhccData(dataList []*EdbInfoSearchData, moveDay int) (newDataList []EdbInfoSearchData, dateDataMap map[string]float64) {
- dateMap := make(map[time.Time]float64)
- var minDate, maxDate time.Time
- dateDataMap = make(map[string]float64)
- for _, v := range dataList {
- currDate, _ := time.ParseInLocation(utils.FormatDate, v.DataTime, time.Local)
- if minDate.IsZero() || currDate.Before(minDate) {
- minDate = currDate
- }
- if maxDate.IsZero() || currDate.After(maxDate) {
- maxDate = currDate
- }
- dateMap[currDate] = v.Value
- }
- // 处理领先、滞后数据
- newDateMap := make(map[time.Time]float64)
- for currDate, value := range dateMap {
- newDate := currDate.AddDate(0, 0, moveDay)
- newDateMap[newDate] = value
- }
- // 开始平移天数
- minDate = minDate.AddDate(0, 0, moveDay)
- maxDate = maxDate.AddDate(0, 0, moveDay)
- // 计算最早日期与最晚日期之间的相差天数
- dayNum := utils.GetTimeSubDay(minDate, maxDate)
- // 将最早日期与最晚日期之间不存在的日期进行数据补全(按前一期数据补充)
- for i := 0; i <= dayNum; i++ {
- currDate := minDate.AddDate(0, 0, i)
- tmpValue, ok := newDateMap[currDate]
- if !ok {
- // 万一没有数据,那么就过滤当次循环
- if len(newDataList) <= 0 {
- continue
- }
- //找不到数据,那么就用前面的数据吧
- tmpValue = newDataList[len(newDataList)-1].Value
- }
- tmpData := EdbInfoSearchData{
- //EdbDataId: 0,
- DataTime: currDate.Format(utils.FormatDate),
- Value: tmpValue,
- }
- dateDataMap[tmpData.DataTime] = tmpData.Value
- newDataList = append(newDataList, tmpData)
- }
- return
- }
|