edb_data_residual_analysis.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662
  1. package models
  2. import (
  3. "encoding/json"
  4. "eta/eta_index_lib/global"
  5. "eta/eta_index_lib/utils"
  6. "fmt"
  7. "gorm.io/gorm"
  8. "math"
  9. "sort"
  10. "strconv"
  11. "time"
  12. )
  13. type edbDataResidualAnalysis struct {
  14. EdbDataId int `gorm:"column:edb_data_id;type:int(11);primaryKey;not null;"`
  15. EdbInfoId int `gorm:"column:edb_info_id;type:int(11);comment:指标id;default:NULL;"` // 指标id
  16. EdbCode string `gorm:"column:edb_code;type:varchar(190);comment:指标编码;default:NULL;"` // 指标编码
  17. DataTime string `gorm:"column:data_time;type:date;comment:数据日期;default:NULL;"` // 数据日期
  18. Value float64 `gorm:"column:value;type:double;comment:数据值;default:NULL;"` // 数据值
  19. CreateTime time.Time `gorm:"column:create_time;type:datetime;comment:创建时间;default:NULL;" ` // 创建时间
  20. ModifyTime time.Time `gorm:"column:modify_time;type:datetime;comment:修改时间;default:NULL;"` // 修改时间
  21. DataTimestamp int64 `gorm:"column:data_timestamp;type:bigint(20);comment:数据日期时间戳;default:0;"` // 数据日期时间戳
  22. }
  23. // AfterFind 在该模型上设置钩子函数,把日期转成正确的string,所以查询函数只能用Find函数,First或者Scan是不会触发该函数的来获取数据
  24. func (m *edbDataResidualAnalysis) AfterFind(db *gorm.DB) (err error) {
  25. m.DataTime = utils.GormDateStrToDateStr(m.DataTime)
  26. return
  27. }
  28. // ConvertTimeStr
  29. // @Description: 转成需要输出的格式
  30. // @receiver m
  31. func (m *edbDataResidualAnalysis) ConvertTimeStr() {
  32. m.DataTime = utils.GormDateStrToDateStr(m.DataTime)
  33. return
  34. }
  35. // AddResidualAnalysisData 新增指标数据
  36. func AddResidualAnalysisData(dataList []edbDataResidualAnalysis) (err error) {
  37. err = global.DEFAULT_DB.CreateInBatches(dataList, utils.MultiAddNum).Error
  38. if err != nil {
  39. return err
  40. }
  41. return nil
  42. }
  43. // RefreshAllCalculateResidualAnalysis 刷新残差分析
  44. func RefreshAllCalculateResidualAnalysis(edbInfoId, source, subSource, formulaInt, moveType int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate, moveFrequency string) (err error) {
  45. to := global.DEFAULT_DB
  46. /*to, err := o.Begin()
  47. if err != nil {
  48. return
  49. }
  50. defer func() {
  51. if err != nil {
  52. fmt.Println("RefreshAllCalculateResidualAnalysis,Err:" + err.Error())
  53. _ = to.Rollback()
  54. } else {
  55. _ = to.Commit()
  56. }
  57. }()*/
  58. configMapping, err := GetConfigMappingListByConditionNotBase(edbInfoId)
  59. if err != nil {
  60. return err
  61. }
  62. var edbInfoIdList []int64
  63. for _, v := range configMapping {
  64. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  65. }
  66. //清空原有数据
  67. sql := ` DELETE FROM edb_data_residual_analysis WHERE edb_info_id in (` + utils.GetOrmInReplace(len(edbInfoIdList)) + `) `
  68. err = to.Exec(sql, edbInfoIdList).Error
  69. if err != nil {
  70. return
  71. }
  72. //计算数据
  73. err = refreshAllCalculateResidualAnalysis(edbInfoId, source, subSource, formulaInt, moveType, fromEdbInfo, edbCode, startDate, endDate, moveFrequency, configMapping)
  74. return
  75. }
  76. // refreshAllCalculateResidualAnalysis 刷新所有残差分析
  77. func refreshAllCalculateResidualAnalysis(edbInfoId, source, subSource, formulaInt, moveType int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate, moveFrequency string, configMapping []CalculateResidualAnalysisConfigMapping) (err error) {
  78. fmt.Println("refreshAllCalculateResidualAnalysis startDate:", startDate)
  79. calculateMappingList, err := GetCalculateMappingListByEdbInfoId(edbInfoId)
  80. if err != nil {
  81. return err
  82. }
  83. edbInfoIdA := calculateMappingList[0].FromEdbInfoId
  84. edbInfoIdB := calculateMappingList[1].FromEdbInfoId
  85. mappingList, err := GetEdbInfoListByIds([]int{edbInfoIdA, edbInfoIdB})
  86. if err != nil {
  87. return err
  88. }
  89. var edbInfoMappingA, edbInfoMappingB *EdbInfoList
  90. for _, v := range mappingList {
  91. if v.Unit == "无" {
  92. v.Unit = ""
  93. }
  94. if v.EdbInfoId == edbInfoIdA {
  95. edbInfoMappingA = v
  96. }
  97. if v.EdbInfoId == edbInfoIdB {
  98. edbInfoMappingB = v
  99. }
  100. }
  101. if edbInfoMappingA == nil {
  102. return fmt.Errorf("指标A不存在")
  103. }
  104. if edbInfoMappingB == nil {
  105. return fmt.Errorf("指标B不存在")
  106. }
  107. // 从配置中取出时间范围
  108. analysisConfig, err := GetResidualAnalysisConfigById(edbInfoId)
  109. if err != nil {
  110. return err
  111. }
  112. configString := analysisConfig.Config
  113. var config ResidualAnalysisConfig
  114. err = json.Unmarshal([]byte(configString), &config)
  115. // 时间处理
  116. switch config.DateType {
  117. case 0:
  118. startDate = config.StartDate
  119. endDate = config.EndDate
  120. case 1:
  121. startDate = config.StartDate
  122. endDate = ""
  123. default:
  124. startDate = utils.GetPreYearTime(config.DateType)
  125. endDate = ""
  126. }
  127. // 原始图表信息
  128. originalEdbList := make([]EdbInfoList, 0)
  129. originalEdbList, fullADataList, fullBDataList, err := fillOriginalChart(config, fromEdbInfo, mappingList, startDate, endDate, edbInfoMappingA, edbInfoMappingB, originalEdbList)
  130. if err != nil {
  131. return err
  132. }
  133. indexADataMap := map[string]*EdbData{}
  134. for _, indexData := range edbInfoMappingA.DataList {
  135. indexADataMap[indexData.DataTime] = indexData
  136. }
  137. // 映射图表信息
  138. mappingEdbList, _, _, _, err := fillMappingChartInfo(config, fromEdbInfo, edbInfoMappingA, edbInfoMappingB, originalEdbList, indexADataMap, startDate, endDate, fullADataList, fullBDataList)
  139. if err != nil {
  140. return err
  141. }
  142. // 残差图表信息
  143. residualEdbList, _, err := fillResidualChartInfo(config, fromEdbInfo, edbInfoMappingA, edbInfoMappingB, mappingEdbList)
  144. // 映射指标 与 残差指标 同步刷新
  145. for _, mapping := range configMapping {
  146. var edbDataResidualAnalysisList []edbDataResidualAnalysis
  147. if mapping.IndexType == 1 {
  148. edbInfo, err := GetEdbInfoById(int(mapping.EdbInfoId))
  149. if err != nil {
  150. return err
  151. }
  152. for _, edbData := range mappingEdbList[1].DataList {
  153. value, _ := strconv.ParseFloat(edbData.Value, 64)
  154. edbDataResidualAnalysisList = append(edbDataResidualAnalysisList, edbDataResidualAnalysis{
  155. EdbInfoId: int(mapping.EdbInfoId),
  156. EdbCode: edbInfo.EdbCode,
  157. DataTime: edbData.DataTime,
  158. Value: value,
  159. CreateTime: time.Now(),
  160. ModifyTime: time.Now(),
  161. DataTimestamp: edbData.DataTimestamp,
  162. })
  163. }
  164. err = AddResidualAnalysisData(edbDataResidualAnalysisList)
  165. if err != nil {
  166. return err
  167. }
  168. } else if mapping.IndexType == 2 {
  169. edbInfo, err := GetEdbInfoById(int(mapping.EdbInfoId))
  170. if err != nil {
  171. return err
  172. }
  173. for _, edbData := range residualEdbList[1].DataList {
  174. value, _ := strconv.ParseFloat(edbData.Value, 64)
  175. edbDataResidualAnalysisList = append(edbDataResidualAnalysisList, edbDataResidualAnalysis{
  176. EdbInfoId: int(mapping.EdbInfoId),
  177. EdbCode: edbInfo.EdbCode,
  178. DataTime: edbData.DataTime,
  179. Value: value,
  180. CreateTime: time.Now(),
  181. ModifyTime: time.Now(),
  182. DataTimestamp: edbData.DataTimestamp,
  183. })
  184. }
  185. err = AddResidualAnalysisData(edbDataResidualAnalysisList)
  186. if err != nil {
  187. return err
  188. }
  189. }
  190. }
  191. return
  192. }
  193. func fillResidualChartInfo(config ResidualAnalysisConfig, req *EdbInfo, edbInfoMappingA *EdbInfoList, edbInfoMappingB *EdbInfoList, mappingEdbList []EdbInfoList) ([]EdbInfoList, float64, error) {
  194. // 计算公式 映射残差 = 因变量指标 - 映射指标
  195. var edbInfoA, edbInfoB EdbInfoList
  196. if mappingEdbList[0].EdbInfoId == edbInfoMappingA.EdbInfoId {
  197. edbInfoA = mappingEdbList[0]
  198. edbInfoB = mappingEdbList[1]
  199. } else {
  200. edbInfoA = mappingEdbList[1]
  201. edbInfoB = mappingEdbList[0]
  202. }
  203. dataAList := edbInfoA.DataList
  204. edbData := make([]*EdbDataList, len(dataAList))
  205. for i, data := range dataAList {
  206. f, _ := strconv.ParseFloat(data.Value, 64)
  207. edbData[i] = &EdbDataList{
  208. Value: f,
  209. DataTimestamp: data.DataTimestamp,
  210. DataTime: data.DataTime,
  211. EdbInfoId: data.EdbInfoId,
  212. EdbDataId: data.EdbDataId,
  213. }
  214. }
  215. dataBList := edbInfoB.DataList
  216. // 映射指标开始时间
  217. var startTime string
  218. if len(dataBList) > 0 {
  219. startTime = dataBList[0].DataTime
  220. }
  221. var indexDataBMap = make(map[string]*EdbData)
  222. for _, data := range dataBList {
  223. indexDataBMap[data.DataTime] = data
  224. }
  225. // 求R2
  226. var valueB, sumValueA, averageValueA, residualQuadraticSum, totalQuadraticSum, R2 float64
  227. for _, indexData := range edbData {
  228. // 因变量的值总和
  229. sumValueA += indexData.Value
  230. }
  231. // 因变量平均值
  232. averageValueA = sumValueA / float64(len(edbData))
  233. var indexMax, indexMin float64
  234. var edbDataResp []*EdbDataList
  235. if len(edbData) > 0 {
  236. indexMax = edbData[0].Value
  237. indexMin = edbData[0].Value
  238. for _, indexData := range edbData {
  239. if dataB, ok := indexDataBMap[indexData.DataTime]; ok {
  240. f, _ := strconv.ParseFloat(dataB.Value, 64)
  241. valueB = f
  242. } else {
  243. continue
  244. }
  245. // 总因变量平方和
  246. totalQuadraticSum += math.Pow(indexData.Value-averageValueA, 2)
  247. // 补全残差值
  248. indexData.Value = math.Round((indexData.Value-valueB)*10000) / 10000
  249. // 残差平方和
  250. residualQuadraticSum += math.Pow(indexData.Value, 2)
  251. if indexData.Value > indexMax {
  252. indexMax = indexData.Value
  253. }
  254. if indexData.Value < indexMin {
  255. indexMin = indexData.Value
  256. }
  257. // 获取映射指标之后的数据
  258. if startTime != "" && utils.CompareDate(startTime, indexData.DataTime) {
  259. edbDataResp = append(edbDataResp, indexData)
  260. }
  261. }
  262. }
  263. // 计算R2 公式:R2=1-SSE/SST R2越大,越符合线性 R2 = 1 - 残差平方和/总平方和
  264. R2 = 1 - residualQuadraticSum/totalQuadraticSum
  265. mappingEdb := make([]EdbInfoList, len(mappingEdbList))
  266. copy(mappingEdb, mappingEdbList)
  267. for i, mapping := range mappingEdb {
  268. if mapping.EdbInfoId != edbInfoMappingA.EdbInfoId {
  269. toEdbData := convertEdbDataListToEdbData(edbDataResp)
  270. mappingEdb[i].DataList = toEdbData
  271. mappingEdb[i].EdbName = edbInfoMappingA.EdbName + "映射残差/" + edbInfoMappingB.EdbName
  272. if config.IndexType == 2 {
  273. if config.LeadValue > 0 {
  274. mappingEdb[i].EdbName = edbInfoMappingA.EdbName + "映射残差/" + edbInfoMappingB.EdbName + "(领先" + strconv.Itoa(config.LeadValue) + config.LeadFrequency + ")"
  275. }
  276. }
  277. }
  278. }
  279. return mappingEdb, R2, nil
  280. }
  281. func fillMappingChartInfo(config ResidualAnalysisConfig, req *EdbInfo, edbInfoMappingA *EdbInfoList, edbInfoMappingB *EdbInfoList, originalEdbList []EdbInfoList, indexADataMap map[string]*EdbData, startDate string, endDate string, fullADataList []*EdbDataList, fullBDataList []*EdbDataList) ([]EdbInfoList, float64, float64, float64, error) {
  282. // 计算公式:Y=aX+b,Y为映射后的指标,X为自变量指标
  283. // 正序:a=(L2-L1)/(R2-R1) b=L2-R2*a
  284. // 逆序:a=(L2-L1)/(R1-R2) b=L2-R1*a
  285. // L2:左轴下限 R2:右轴上限 L1:左轴上限 R1:右轴下限
  286. var a, b, r float64
  287. // 映射残差 计算a,b
  288. if config.ResidualType == 1 {
  289. if config.IsOrder {
  290. a = (config.LeftIndexMax - config.LeftIndexMin) / (config.RightIndexMin - config.RightIndexMax)
  291. b = config.LeftIndexMax - config.RightIndexMin*a
  292. } else {
  293. a = (config.LeftIndexMax - config.LeftIndexMin) / (config.RightIndexMax - config.RightIndexMin)
  294. b = config.LeftIndexMax - config.RightIndexMax*a
  295. }
  296. }
  297. dataList := edbInfoMappingB.DataList
  298. // 指标B数据补充
  299. // 新建一个切片来保存补充的数据
  300. var replenishDataList []*EdbData
  301. for index := 0; index < len(dataList)-1; index++ {
  302. // 获取当前数据和下一个数据
  303. beforeIndexData := dataList[index]
  304. afterIndexData := dataList[index+1]
  305. // 从最早时间开始,补充时间为自然日
  306. for utils.IsMoreThanOneDay(beforeIndexData.DataTime, afterIndexData.DataTime) {
  307. // 创建补充数据
  308. nextDay := utils.GetNextDay(beforeIndexData.DataTime)
  309. toTime := utils.StringToTime(nextDay)
  310. replenishIndexData := EdbData{
  311. DataTime: nextDay, // 计算下一个自然日
  312. DataTimestamp: toTime.UnixMilli(),
  313. Value: beforeIndexData.Value, // 可以选择使用前一天的值,或者其他逻辑来计算值
  314. }
  315. // 将补充数据加入补充数据列表
  316. replenishDataList = append(replenishDataList, &replenishIndexData)
  317. // 更新 beforeIndexData 为新创建的补充数据
  318. beforeIndexData = &replenishIndexData
  319. }
  320. }
  321. // 将补充数据插入原始数据列表
  322. dataList = append(dataList, replenishDataList...)
  323. // 排序
  324. sort.Sort(ByDataTime(dataList))
  325. // 拟合残差 计算a,b
  326. var coordinateList []utils.Coordinate
  327. var replenishADataList []*EdbDataList
  328. var replenishBDataList []*EdbDataList
  329. if config.ResidualType == 2 {
  330. //
  331. // 因变量指标也转换为日度
  332. for index := 0; index < len(fullADataList)-1; index++ {
  333. // 获取当前数据和下一个数据
  334. beforeIndexData := fullADataList[index]
  335. afterIndexData := fullADataList[index+1]
  336. replenishADataList = append(replenishADataList, beforeIndexData)
  337. // 从最早时间开始,补充时间为自然日
  338. if utils.IsMoreThanOneDay(beforeIndexData.DataTime, afterIndexData.DataTime) {
  339. for {
  340. // 创建补充数据
  341. nextDay := utils.GetNextDay(beforeIndexData.DataTime)
  342. toTime := utils.StringToTime(nextDay)
  343. replenishIndexData := EdbDataList{
  344. DataTime: nextDay, // 计算下一个自然日
  345. DataTimestamp: toTime.UnixMilli(),
  346. Value: beforeIndexData.Value, // 可以选择使用前一天的值,或者其他逻辑来计算值
  347. }
  348. // 将补充数据加入补充数据列表
  349. replenishADataList = append(replenishADataList, &replenishIndexData)
  350. // 更新 beforeIndexData 为新创建的补充数据
  351. beforeIndexData = &replenishIndexData
  352. // 检查是否还需要继续补充数据
  353. if !utils.IsMoreThanOneDay(beforeIndexData.DataTime, afterIndexData.DataTime) {
  354. break
  355. }
  356. }
  357. }
  358. }
  359. replenishADataList = append(replenishADataList, fullADataList[len(fullADataList)-1])
  360. // 自变量指标也转换为日度
  361. for index := 0; index < len(fullBDataList)-1; index++ {
  362. // 获取当前数据和下一个数据
  363. beforeIndexData := fullBDataList[index]
  364. afterIndexData := fullBDataList[index+1]
  365. replenishBDataList = append(replenishBDataList, beforeIndexData)
  366. // 从最早时间开始,补充时间为自然日
  367. if utils.IsMoreThanOneDay(beforeIndexData.DataTime, afterIndexData.DataTime) {
  368. for {
  369. // 创建补充数据
  370. nextDay := utils.GetNextDay(beforeIndexData.DataTime)
  371. toTime := utils.StringToTime(nextDay)
  372. replenishIndexData := EdbDataList{
  373. DataTime: nextDay, // 计算下一个自然日
  374. DataTimestamp: toTime.UnixMilli(),
  375. Value: beforeIndexData.Value, // 可以选择使用前一天的值,或者其他逻辑来计算值
  376. }
  377. // 将补充数据加入补充数据列表
  378. replenishBDataList = append(replenishBDataList, &replenishIndexData)
  379. // 更新 beforeIndexData 为新创建的补充数据
  380. beforeIndexData = &replenishIndexData
  381. // 检查是否还需要继续补充数据
  382. if !utils.IsMoreThanOneDay(beforeIndexData.DataTime, afterIndexData.DataTime) {
  383. break
  384. }
  385. }
  386. }
  387. }
  388. replenishBDataList = append(replenishBDataList, fullBDataList[len(fullBDataList)-1])
  389. // replenishADataList --> map
  390. replenishADataMap := make(map[string]*EdbDataList)
  391. for _, indexData := range replenishADataList {
  392. if (utils.StringToTime(indexData.DataTime).After(utils.StringToTime(startDate)) || utils.StringToTime(indexData.DataTime).Equal(utils.StringToTime(startDate))) && (endDate == "" || utils.StringToTime(indexData.DataTime).Before(utils.StringToTime(endDate))) {
  393. replenishADataMap[indexData.DataTime] = indexData
  394. }
  395. }
  396. for _, indexData := range replenishBDataList {
  397. if _, ok := replenishADataMap[indexData.DataTime]; ok {
  398. coordinate := utils.Coordinate{
  399. X: indexData.Value,
  400. Y: replenishADataMap[indexData.DataTime].Value,
  401. }
  402. coordinateList = append(coordinateList, coordinate)
  403. }
  404. }
  405. a, b = utils.GetLinearResult(coordinateList)
  406. r = utils.ComputeCorrelation(coordinateList)
  407. }
  408. // 填充映射指标值 使得时间长度一致
  409. dataList = FillDataBList(dataList, edbInfoMappingA)
  410. // 根据指标A的时间key,在B的映射指标中筛选出对应的值
  411. var dataBList []*EdbDataList
  412. var indexMax, indexMin string
  413. if len(dataList) > 0 {
  414. indexMax = dataList[0].Value
  415. indexMin = dataList[0].Value
  416. for _, indexData := range dataList {
  417. if _, ok := indexADataMap[indexData.DataTime]; ok {
  418. indexDataCopy := *indexData
  419. // 计算指标B映射值
  420. f, _ := strconv.ParseFloat(indexData.Value, 64)
  421. indexDataCopy.Value = fmt.Sprintf("%f", math.Round((a*f+b)*10000)/10000)
  422. // 比较最大值
  423. if indexData.Value > indexMax {
  424. indexMax = indexData.Value
  425. }
  426. // 比较最小值
  427. if indexData.Value < indexMin {
  428. indexMin = indexData.Value
  429. }
  430. // 将副本添加到 dataBList
  431. copyValue, _ := strconv.ParseFloat(indexDataCopy.Value, 64)
  432. dataBList = append(dataBList, &EdbDataList{
  433. DataTime: indexDataCopy.DataTime,
  434. DataTimestamp: indexDataCopy.DataTimestamp,
  435. EdbDataId: indexDataCopy.EdbDataId,
  436. EdbInfoId: indexDataCopy.EdbInfoId,
  437. Value: copyValue,
  438. })
  439. }
  440. }
  441. }
  442. mappingEdbList := make([]EdbInfoList, len(originalEdbList))
  443. copy(mappingEdbList, originalEdbList)
  444. for i, mapping := range mappingEdbList {
  445. if mapping.EdbInfoId != edbInfoMappingA.EdbInfoId {
  446. mappingEdbList[i].EdbInfoId = 0
  447. mappingEdbList[i].EdbCode = ""
  448. mappingEdbList[i].EdbName = edbInfoMappingB.EdbName + "映射" + edbInfoMappingA.EdbName
  449. if config.IndexType == 2 {
  450. if config.LeadValue > 0 {
  451. mappingEdbList[i].EdbName = edbInfoMappingB.EdbName + "映射" + edbInfoMappingA.EdbName + "(领先" + strconv.Itoa(config.LeadValue) + config.LeadFrequency + ")"
  452. }
  453. }
  454. edbData := convertEdbDataListToEdbData(dataBList)
  455. mappingEdbList[i].DataList = edbData
  456. }
  457. }
  458. return mappingEdbList, a, b, r, nil
  459. }
  460. // FillDataBList 填充B的数据 使得与A的时间保持一致
  461. func FillDataBList(dataList []*EdbData, edbInfoMappingA *EdbInfoList) []*EdbData {
  462. dataAList := edbInfoMappingA.DataList
  463. for utils.StringToTime(dataList[len(dataList)-1].DataTime).Before(utils.StringToTime(dataAList[len(dataAList)-1].DataTime)) {
  464. // 使用A的时间填充时间差
  465. timeDiff := utils.GetNextDayN(dataList[len(dataList)-1].DataTime, 1)
  466. // 创建新的数据点并填充 前值填充
  467. newDataPoint := &EdbData{
  468. DataTime: timeDiff,
  469. Value: dataList[len(dataList)-1].Value,
  470. DataTimestamp: utils.StringToTime(timeDiff).UnixMilli(),
  471. }
  472. // 将新数据点添加到dataList末尾
  473. dataList = append(dataList, newDataPoint)
  474. }
  475. return dataList
  476. }
  477. func fillOriginalChart(config ResidualAnalysisConfig, req *EdbInfo, mappingList []*EdbInfoList, startDate string, endDate string, edbInfoMappingA *EdbInfoList, edbInfoMappingB *EdbInfoList, originalEdbList []EdbInfoList) ([]EdbInfoList, []*EdbDataList, []*EdbDataList, error) {
  478. var fullADataList, fullBDataList []*EdbDataList
  479. for _, v := range mappingList {
  480. var edbInfoMapping EdbInfoList
  481. edbInfoMapping.EdbName = v.EdbName
  482. // 获取图表中的指标数据
  483. dataList, err := GetEdbDataList(v.Source, v.SubSource, v.EdbInfoId, startDate, endDate)
  484. if err != nil {
  485. return nil, nil, nil, fmt.Errorf("获取指标数据失败,Err:%s", err.Error())
  486. }
  487. data := convertEdbDataListToEdbData(dataList)
  488. // 重新获取指标数据 产品要求需要和计算指标-拟合残差逻辑保持一致
  489. fullDataList, err := GetEdbDataList(v.Source, v.SubSource, v.EdbInfoId, "", "")
  490. if err != nil {
  491. return nil, nil, nil, fmt.Errorf("获取指标数据失败,Err:%s", err.Error())
  492. }
  493. if v.EdbInfoId == edbInfoMappingB.EdbInfoId {
  494. // 领先指标 dataList进行数据处理
  495. if config.IndexType == 1 {
  496. if config.LeadValue < 0 {
  497. return nil, nil, nil, fmt.Errorf("领先值不能小于0")
  498. } else if config.LeadValue > 0 {
  499. edbInfoMapping.EdbName = v.EdbName + "(领先" + strconv.Itoa(config.LeadValue) + config.LeadFrequency + ")"
  500. for _, indexData := range dataList {
  501. switch config.LeadFrequency {
  502. case "天":
  503. indexData.DataTime = utils.GetNextDayN(indexData.DataTime, config.LeadValue)
  504. case "周":
  505. indexData.DataTime = utils.GetNextDayN(indexData.DataTime, config.LeadValue*7)
  506. case "月":
  507. indexData.DataTime = utils.TimeToString(utils.AddDate(utils.StringToTime(indexData.DataTime), 0, config.LeadValue), utils.YearMonthDay)
  508. case "季":
  509. indexData.DataTime = utils.TimeToString(utils.AddDate(utils.StringToTime(indexData.DataTime), 0, config.LeadValue*3), utils.YearMonthDay)
  510. case "年":
  511. indexData.DataTime = utils.TimeToString(utils.AddDate(utils.StringToTime(indexData.DataTime), config.LeadValue, 0), utils.YearMonthDay)
  512. }
  513. indexData.DataTimestamp = utils.StringToTime(indexData.DataTime).UnixMilli()
  514. }
  515. }
  516. }
  517. edbInfoMappingB.DataList = data
  518. fullBDataList = fullDataList
  519. } else {
  520. edbInfoMappingA.DataList = data
  521. fullADataList = fullDataList
  522. }
  523. edbInfoMapping.EdbInfoId = v.EdbInfoId
  524. edbInfoMapping.EdbCode = v.EdbCode
  525. edbInfoMapping.Unit = v.Unit
  526. edbInfoMapping.Frequency = v.Frequency
  527. edbInfoMapping.Source = v.Source
  528. edbInfoMapping.SourceName = v.SourceName
  529. edbInfoMapping.LatestDate = v.LatestDate
  530. edbInfoMapping.LatestValue = v.LatestValue
  531. edbInfoMapping.DataList = data
  532. originalEdbList = append(originalEdbList, edbInfoMapping)
  533. }
  534. return originalEdbList, fullADataList, fullBDataList, nil
  535. }
  536. func convertEdbDataListToEdbData(edbDataLists []*EdbDataList) []*EdbData {
  537. var edbDataList []*EdbData
  538. for _, edbData := range edbDataLists {
  539. data := &EdbData{
  540. DataTime: edbData.DataTime,
  541. DataTimestamp: edbData.DataTimestamp,
  542. EdbDataId: edbData.EdbDataId,
  543. EdbInfoId: edbData.EdbInfoId,
  544. Value: fmt.Sprintf("%f", edbData.Value),
  545. }
  546. edbDataList = append(edbDataList, data)
  547. }
  548. return edbDataList
  549. }
  550. type ByDataTime []*EdbData
  551. func (a ByDataTime) Len() int {
  552. return len(a)
  553. }
  554. func (a ByDataTime) Swap(i, j int) {
  555. a[i], a[j] = a[j], a[i]
  556. }
  557. func (a ByDataTime) Less(i, j int) bool {
  558. t1 := utils.StringToTime(a[i].DataTime)
  559. t2 := utils.StringToTime(a[j].DataTime)
  560. return t1.Before(t2)
  561. }