residual_analysis_service.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644
  1. package residual_analysis_service
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/models/data_manage"
  5. "eta/eta_api/models/residual_analysis_model"
  6. "eta/eta_api/models/system"
  7. "eta/eta_api/utils"
  8. "fmt"
  9. "math"
  10. "strconv"
  11. "time"
  12. )
  13. func ResidualAnalysisPreview(req residual_analysis_model.ResidualAnalysisReq) (residual_analysis_model.ResidualAnalysisResp, error) {
  14. if req.DateType < 0 {
  15. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("时间类型错误", nil)
  16. }
  17. mappingList, err := data_manage.GetChartEdbMappingListByEdbInfoIdList([]int{req.EdbInfoIdA, req.EdbInfoIdB})
  18. if err != nil {
  19. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("获取图表,指标信息失败,Err:%s", err.Error())
  20. }
  21. var edbInfoMappingA, edbInfoMappingB *data_manage.ChartEdbInfoMapping
  22. for _, v := range mappingList {
  23. if v.EdbInfoId == req.EdbInfoIdA {
  24. edbInfoMappingA = v
  25. }
  26. if v.EdbInfoId == req.EdbInfoIdB {
  27. edbInfoMappingB = v
  28. }
  29. }
  30. if edbInfoMappingA == nil {
  31. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("指标A不存在", nil)
  32. }
  33. if edbInfoMappingB == nil {
  34. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("指标B不存在", nil)
  35. }
  36. // 时间处理
  37. var startDate, endDate string
  38. switch req.DateType {
  39. case 0:
  40. startDate = req.StartDate
  41. endDate = req.EndDate
  42. case 1:
  43. startDate = req.StartDate
  44. endDate = ""
  45. default:
  46. startDate = utils.GetPreYearTime(req.DateType)
  47. endDate = ""
  48. }
  49. resp := residual_analysis_model.ResidualAnalysisResp{}
  50. // 原始图表信息
  51. originalEdbList := make([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, 0)
  52. originalEdbList, err = fillOriginalChart(req, mappingList, startDate, endDate, edbInfoMappingA, edbInfoMappingB, originalEdbList)
  53. if err != nil {
  54. return residual_analysis_model.ResidualAnalysisResp{}, err
  55. }
  56. originalChartInfo := createChartInfoResp(req, startDate, endDate, edbInfoMappingA.EdbName+"与"+edbInfoMappingB.EdbName)
  57. resp.OriginalChartData = residual_analysis_model.ChartResp{
  58. ChartInfo: originalChartInfo,
  59. EdbInfoList: originalEdbList,
  60. }
  61. // 如果只需要第一张图表的数据 直接返回,避免继续处理
  62. if req.QueryType == 1 {
  63. return resp, nil
  64. }
  65. dataAList, ok := edbInfoMappingA.DataList.([]*data_manage.EdbDataList)
  66. if !ok {
  67. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("数据类型转换失败", nil)
  68. }
  69. indexADataMap := map[string]*data_manage.EdbDataList{}
  70. for _, indexData := range dataAList {
  71. indexADataMap[indexData.DataTime] = indexData
  72. }
  73. // 映射图表信息
  74. mappingEdbList, a, b, r, err := fillMappingChartInfo(req, edbInfoMappingA, edbInfoMappingB, originalEdbList, indexADataMap)
  75. if err != nil {
  76. return residual_analysis_model.ResidualAnalysisResp{}, err
  77. }
  78. mappingChartInfo := createChartInfoResp(req, startDate, endDate, edbInfoMappingA.EdbName+"与"+edbInfoMappingB.EdbName+"映射"+edbInfoMappingA.EdbName)
  79. resp.MappingChartData = residual_analysis_model.ChartResp{
  80. ChartInfo: mappingChartInfo,
  81. EdbInfoList: mappingEdbList,
  82. }
  83. // 残差图表信息
  84. residualEdbList, R2, err := fillResidualChartInfo(edbInfoMappingA, edbInfoMappingB, mappingEdbList)
  85. if err != nil {
  86. return residual_analysis_model.ResidualAnalysisResp{}, err
  87. }
  88. residualChartInfo := createChartInfoResp(req, startDate, endDate, edbInfoMappingA.EdbName+"与"+edbInfoMappingA.EdbName+"映射残差/"+edbInfoMappingB.EdbName)
  89. resp.ResidualChartData = residual_analysis_model.ChartResp{
  90. ChartInfo: residualChartInfo,
  91. EdbInfoList: residualEdbList,
  92. }
  93. if req.ResidualType == 2 {
  94. resp.A = a
  95. resp.B = b
  96. resp.R = r
  97. resp.R2 = R2
  98. }
  99. return resp, nil
  100. }
  101. func createChartInfoResp(req residual_analysis_model.ResidualAnalysisReq, startDate, endDate, chartName string) residual_analysis_model.ResidualAnalysisChartInfo {
  102. return residual_analysis_model.ResidualAnalysisChartInfo{
  103. Calendar: `公历`,
  104. Source: utils.CHART_SOURCE_DEFAULT,
  105. DateType: req.DateType,
  106. StartDate: startDate,
  107. EndDate: endDate,
  108. ChartType: utils.CHART_TYPE_CURVE,
  109. ChartName: chartName,
  110. }
  111. }
  112. func fillResidualChartInfo(edbInfoMappingA *data_manage.ChartEdbInfoMapping, edbInfoMappingB *data_manage.ChartEdbInfoMapping, mappingEdbList []residual_analysis_model.ResidualAnalysisChartEdbInfoMapping) ([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, float64, error) {
  113. // 计算公式 映射残差 = 因变量指标 - 映射指标
  114. var edbInfoA, edbInfoB residual_analysis_model.ResidualAnalysisChartEdbInfoMapping
  115. if mappingEdbList[0].EdbInfoId == edbInfoMappingA.EdbInfoId {
  116. edbInfoA = mappingEdbList[0]
  117. edbInfoB = mappingEdbList[1]
  118. } else {
  119. edbInfoA = mappingEdbList[1]
  120. edbInfoB = mappingEdbList[0]
  121. }
  122. dataAList, ok := edbInfoA.DataList.([]*data_manage.EdbDataList)
  123. if !ok {
  124. return nil, 0, fmt.Errorf("数据类型转换失败", nil)
  125. }
  126. edbData := dataAList
  127. dataBList, ok := edbInfoB.DataList.([]*data_manage.EdbDataList)
  128. if !ok {
  129. return nil, 0, fmt.Errorf("数据类型转换失败", nil)
  130. }
  131. var indexDataBMap = make(map[string]*data_manage.EdbDataList)
  132. for _, data := range dataBList {
  133. indexDataBMap[data.DataTime] = data
  134. }
  135. // 求R2
  136. var valueB, sumValueA, averageValueA, residualQuadraticSum, totalQuadraticSum, R2 float64
  137. for _, indexData := range edbData {
  138. // 因变量的值总和
  139. sumValueA += indexData.Value
  140. }
  141. // 因变量平均值
  142. averageValueA = sumValueA / float64(len(edbData))
  143. for _, indexData := range edbData {
  144. if dataB, ok := indexDataBMap[indexData.DataTime]; ok {
  145. valueB = dataB.Value
  146. } else {
  147. valueB = 0
  148. }
  149. // 总因变量平方和
  150. totalQuadraticSum += math.Pow(indexData.Value-averageValueA, 2)
  151. // 补全残差值
  152. indexData.Value = indexData.Value - valueB
  153. // 残差平方和
  154. residualQuadraticSum += math.Pow(indexData.Value, 2)
  155. }
  156. // 计算R2 公式:R2=1-SSE/SST R2越大,越符合线性 R2 = 1 - 残差平方和/总平方和
  157. R2 = 1 - residualQuadraticSum/totalQuadraticSum
  158. mappingEdb := make([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, len(mappingEdbList))
  159. copy(mappingEdb, mappingEdbList)
  160. for i, mapping := range mappingEdb {
  161. if mapping.EdbInfoId != edbInfoMappingA.EdbInfoId {
  162. mappingEdb[i].DataList = edbData
  163. mappingEdb[i].EdbName = edbInfoMappingA.EdbName + "映射残差/" + edbInfoMappingB.EdbName
  164. mappingEdb[i].IsAxis = 1
  165. } else {
  166. mappingEdb[i].IsAxis = 0
  167. }
  168. }
  169. return mappingEdb, R2, nil
  170. }
  171. func fillMappingChartInfo(req residual_analysis_model.ResidualAnalysisReq, edbInfoMappingA *data_manage.ChartEdbInfoMapping, edbInfoMappingB *data_manage.ChartEdbInfoMapping, originalEdbList []residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, indexADataMap map[string]*data_manage.EdbDataList) ([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, float64, float64, float64, error) {
  172. // 计算公式:Y=aX+b,Y为映射后的指标,X为自变量指标
  173. // 正序:a=(L2-L1)/(R2-R1) b=L2-R2*a
  174. // 逆序:a=(L2-L1)/(R1-R2) b=L2-R1*a
  175. // L2:左轴下限 R2:右轴上限 L1:左轴上限 R1:右轴下限
  176. var a, b, r float64
  177. // 映射残差 计算a,b
  178. if req.ResidualType == 1 {
  179. if req.IsOrder {
  180. a = (req.LeftIndexMax - req.LeftIndexMin) / (req.RightIndexMax - req.RightIndexMin)
  181. b = req.LeftIndexMax - req.RightIndexMax*a
  182. } else {
  183. a = (req.LeftIndexMax - req.LeftIndexMin) / (req.RightIndexMin - req.RightIndexMax)
  184. b = req.LeftIndexMax - req.RightIndexMin*a
  185. }
  186. }
  187. dataList, ok := edbInfoMappingB.DataList.([]*data_manage.EdbDataList)
  188. if !ok {
  189. return nil, a, b, r, fmt.Errorf("数据类型转换失败", nil)
  190. }
  191. // 领先指标 dataList进行数据处理
  192. if req.IndexType == 2 {
  193. if req.LeadValue < 0 {
  194. return nil, a, b, r, fmt.Errorf("领先值不能小于0", nil)
  195. } else if req.LeadValue > 0 {
  196. for _, indexData := range dataList {
  197. switch req.LeadFrequency {
  198. case "天":
  199. indexData.DataTime = utils.GetNextDayN(indexData.DataTime, req.LeadValue)
  200. case "周":
  201. indexData.DataTime = utils.GetNextDayN(indexData.DataTime, req.LeadValue*7)
  202. case "月":
  203. indexData.DataTime = utils.TimeToString(utils.AddDate(utils.StringToTime(indexData.DataTime), 0, req.LeadValue), utils.YearMonthDay)
  204. case "季":
  205. indexData.DataTime = utils.TimeToString(utils.AddDate(utils.StringToTime(indexData.DataTime), 0, req.LeadValue*3), utils.YearMonthDay)
  206. case "年":
  207. indexData.DataTime = utils.TimeToString(utils.AddDate(utils.StringToTime(indexData.DataTime), req.LeadValue, 0), utils.YearMonthDay)
  208. }
  209. }
  210. }
  211. }
  212. // 指标B数据补充
  213. for index := 0; index < len(dataList)-1; index++ {
  214. // 获取当前数据和下一个数据
  215. beforeIndexData := dataList[index]
  216. afterIndexData := dataList[index+1]
  217. // 从最早时间开始,补充时间为自然日
  218. if utils.IsMoreThanOneDay(beforeIndexData.DataTime, afterIndexData.DataTime) {
  219. // 创建补充数据
  220. replenishIndexData := data_manage.EdbDataList{
  221. DataTime: utils.GetNextDay(beforeIndexData.DataTime),
  222. DataTimestamp: time.Now().UnixMilli(),
  223. Value: beforeIndexData.Value,
  224. }
  225. // 将补充数据插入到数据列表
  226. dataList = append(dataList, &replenishIndexData)
  227. }
  228. }
  229. // 拟合残差 计算a,b
  230. var coordinateList []utils.Coordinate
  231. if req.ResidualType == 2 {
  232. for _, indexData := range dataList {
  233. if _, ok := indexADataMap[indexData.DataTime]; ok {
  234. coordinate := utils.Coordinate{
  235. X: indexData.Value,
  236. Y: indexADataMap[indexData.DataTime].Value,
  237. }
  238. coordinateList = append(coordinateList, coordinate)
  239. }
  240. }
  241. a, b = utils.GetLinearResult(coordinateList)
  242. r = utils.CalculationDecisive(coordinateList)
  243. }
  244. // 根据指标A的时间key,在B的映射指标中筛选出对应的值
  245. var dataBList []*data_manage.EdbDataList
  246. for _, indexData := range dataList {
  247. if _, ok := indexADataMap[indexData.DataTime]; ok {
  248. // 计算指标B映射值
  249. indexData.Value = math.Round(a*indexData.Value+b) / 10000
  250. dataBList = append(dataBList, indexData)
  251. }
  252. }
  253. mappingEdbList := make([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, len(originalEdbList))
  254. copy(mappingEdbList, originalEdbList)
  255. for i, mapping := range mappingEdbList {
  256. if mapping.EdbInfoId != req.EdbInfoIdA {
  257. mappingEdbList[i].EdbInfoId = 0
  258. mappingEdbList[i].EdbCode = ""
  259. mappingEdbList[i].IsAxis = 1
  260. mappingEdbList[i].EdbName = edbInfoMappingB.EdbName + "映射" + edbInfoMappingA.EdbName
  261. mappingEdbList[i].DataList = dataBList
  262. }
  263. }
  264. return mappingEdbList, a, b, r, nil
  265. }
  266. func fillOriginalChart(req residual_analysis_model.ResidualAnalysisReq, mappingList []*data_manage.ChartEdbInfoMapping, startDate string, endDate string, edbInfoMappingA *data_manage.ChartEdbInfoMapping, edbInfoMappingB *data_manage.ChartEdbInfoMapping, originalEdbList []residual_analysis_model.ResidualAnalysisChartEdbInfoMapping) ([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, error) {
  267. for _, v := range mappingList {
  268. var edbInfoMapping residual_analysis_model.ResidualAnalysisChartEdbInfoMapping
  269. edbInfoMapping.EdbInfoType = 1
  270. edbInfoMapping.IsOrder = false
  271. edbInfoMapping.IsAxis = 1
  272. edbInfoMapping.ChartColor = `#00F`
  273. edbInfoMapping.ChartWidth = 3
  274. // 获取图表中的指标数据
  275. dataList, err := data_manage.GetEdbDataList(v.Source, v.SubSource, v.EdbInfoId, startDate, endDate)
  276. if err != nil {
  277. return nil, fmt.Errorf("获取指标数据失败,Err:%s", err.Error())
  278. }
  279. if v.EdbInfoId == req.EdbInfoIdB {
  280. edbInfoMapping.LeadValue = req.LeadValue
  281. edbInfoMapping.LeadUnit = req.LeadFrequency
  282. edbInfoMapping.EdbInfoType = req.IndexType
  283. edbInfoMapping.IsOrder = req.IsOrder
  284. edbInfoMapping.IsAxis = 0
  285. edbInfoMapping.ChartColor = `#F00`
  286. edbInfoMapping.ChartWidth = 1
  287. edbInfoMappingB.DataList = dataList
  288. } else {
  289. edbInfoMappingA.DataList = dataList
  290. }
  291. edbInfoMapping.EdbInfoId = v.EdbInfoId
  292. edbInfoMapping.EdbName = v.EdbName
  293. edbInfoMapping.EdbCode = v.EdbCode
  294. edbInfoMapping.Unit = v.Unit
  295. edbInfoMapping.Frequency = v.Frequency
  296. edbInfoMapping.Source = v.Source
  297. edbInfoMapping.SourceName = v.SourceName
  298. edbInfoMapping.MinValue = v.MinValue
  299. edbInfoMapping.MaxValue = v.MaxValue
  300. edbInfoMapping.LatestDate = v.LatestDate
  301. edbInfoMapping.LatestValue = v.LatestValue
  302. edbInfoMapping.DataList = dataList
  303. originalEdbList = append(originalEdbList, edbInfoMapping)
  304. }
  305. return originalEdbList, nil
  306. }
  307. func ContrastPreview(indexCode string) (residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, error) {
  308. var condition string
  309. var pars []interface{}
  310. if indexCode != "" {
  311. condition += " and edb_code=?"
  312. pars = append(pars, indexCode)
  313. }
  314. edbInfo, err := data_manage.GetEdbInfoByCondition(condition, pars)
  315. if err != nil {
  316. return residual_analysis_model.ResidualAnalysisChartEdbInfoMapping{}, err
  317. }
  318. if edbInfo == nil {
  319. return residual_analysis_model.ResidualAnalysisChartEdbInfoMapping{}, fmt.Errorf("指标不存在")
  320. }
  321. dataList, err := data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, "", "")
  322. var resp residual_analysis_model.ResidualAnalysisChartEdbInfoMapping
  323. resp.EdbInfoId = edbInfo.EdbInfoId
  324. resp.EdbCode = edbInfo.EdbCode
  325. resp.SourceName = edbInfo.SourceName
  326. resp.EdbName = edbInfo.EdbName
  327. resp.Unit = edbInfo.Unit
  328. resp.Frequency = edbInfo.Frequency
  329. resp.DataList = dataList
  330. return resp, nil
  331. }
  332. func SaveResidualAnalysis(req residual_analysis_model.ResidualAnalysisIndexSaveReq, sysUser *system.Admin) error {
  333. // 验证分类是否存在
  334. classifyCount, err := data_manage.GetEdbClassifyCountById(req.ClassifyId)
  335. if err != nil {
  336. return err
  337. }
  338. if classifyCount <= 0 {
  339. return fmt.Errorf("分类不存在", nil)
  340. }
  341. // 获取指标数据最大值 最小值 最后更新时间 最后更新时间对应的值
  342. var indexMax, indexMin, indexLatestValue float64
  343. var indexLatestDate string
  344. latestTime, _ := time.Parse(utils.YearMonthDay, req.DataList[0].DataTime)
  345. for _, data := range req.DataList {
  346. // 比较最大值
  347. if data.Value > indexMax {
  348. indexMax = data.Value
  349. }
  350. // 比较最小值
  351. if data.Value < indexMin {
  352. indexMin = data.Value
  353. }
  354. // 比较最新时间和对应值
  355. currentTime, err := time.Parse(utils.YearMonthDay, data.DataTime)
  356. if err != nil {
  357. // 时间解析失败,跳过此项
  358. continue
  359. }
  360. // 如果当前时间更晚
  361. if currentTime.After(latestTime) {
  362. latestTime = currentTime
  363. indexLatestDate = data.DataTime
  364. indexLatestValue = data.Value
  365. }
  366. }
  367. var edbInfoId int64
  368. var edbCode string
  369. // 更新or新增
  370. if req.EdbCode != "" {
  371. // 查询指标库指标
  372. edbInfo, err := data_manage.GetEdbInfoByEdbCode(req.Source, req.EdbCode)
  373. if err != nil {
  374. return err
  375. }
  376. if edbInfo == nil {
  377. return fmt.Errorf("指标不存在", nil)
  378. }
  379. edbInfoId = int64(edbInfo.EdbInfoId)
  380. edbCode = edbInfo.EdbCode
  381. // 须补充更新指标最大值,最小值,数据最新时间,数据最新值
  382. edbInfo.MaxValue = indexMax
  383. edbInfo.MinValue = indexMin
  384. edbInfo.LatestDate = indexLatestDate
  385. edbInfo.LatestValue = indexLatestValue
  386. err = edbInfo.Update([]string{"min_value", "max_value", "latest_date", "latest_value"})
  387. if err != nil {
  388. return err
  389. }
  390. // 删除对应得指标数据
  391. err = residual_analysis_model.DeleteResidualAnalysisDataByEdbCode(req.EdbCode)
  392. if err != nil {
  393. return fmt.Errorf("删除指标数据失败", nil)
  394. }
  395. } else {
  396. // 新增指标
  397. edbCode, err = utils.GenerateEdbCode(1, "")
  398. if err != nil {
  399. return err
  400. }
  401. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  402. edbInfoId, err = data_manage.AddEdbInfo(&data_manage.EdbInfo{
  403. EdbCode: edbCode,
  404. UniqueCode: utils.MD5(utils.CHART_PREFIX + "_" + timestamp),
  405. EdbName: req.EdbName,
  406. EdbNameEn: req.EdbNameEn,
  407. ClassifyId: req.ClassifyId,
  408. EdbType: req.EdbType,
  409. Unit: req.Unit,
  410. UnitEn: req.UnitEn,
  411. Frequency: req.Frequency,
  412. Source: req.Source,
  413. SourceName: "残差分析",
  414. })
  415. if err != nil {
  416. return err
  417. }
  418. }
  419. // 新增数据
  420. for i := range req.DataList {
  421. req.DataList[i].EdbDataId = 0
  422. req.DataList[i].EdbInfoId = int(edbInfoId)
  423. req.DataList[i].EdbCode = edbCode
  424. }
  425. _, err = residual_analysis_model.AddResidualAnalysisData(req.DataList)
  426. if err != nil {
  427. return err
  428. }
  429. // 更新保存指标和配置的映射关系
  430. mappingList, err := residual_analysis_model.GetConfigMappingListByConfigId(req.ConfigId)
  431. if err != nil {
  432. return err
  433. }
  434. flag := true
  435. for _, mapping := range mappingList {
  436. if mapping.EdbInfoId == edbInfoId {
  437. flag = false
  438. break
  439. }
  440. }
  441. if flag {
  442. _, err = residual_analysis_model.SaveConfigMapping(residual_analysis_model.CalculateResidualAnalysisConfigMapping{
  443. CalculateResidualAnalysisConfigId: req.ConfigId,
  444. EdbInfoId: edbInfoId,
  445. ResidualType: req.ResidualType,
  446. CreateTime: time.Now(),
  447. ModifyTime: time.Now(),
  448. })
  449. if err != nil {
  450. return err
  451. }
  452. }
  453. return nil
  454. }
  455. func ResidualAnalysisDetail(edbInfoId int) (residual_analysis_model.ResidualAnalysisDetailResp, error) {
  456. // 通过指标配置映射表 拿到配置id,再获取关联的所有指标信息
  457. var condition string
  458. var pars []interface{}
  459. condition += " and edb_info_id=?"
  460. pars = append(pars, edbInfoId)
  461. mappingList, err := residual_analysis_model.GetConfigMappingListByCondition(condition, pars)
  462. if err != nil {
  463. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  464. }
  465. if len(mappingList) <= 0 {
  466. return residual_analysis_model.ResidualAnalysisDetailResp{}, fmt.Errorf("指标不存在", nil)
  467. }
  468. mapping := mappingList[0]
  469. configMappingList, err := residual_analysis_model.GetConfigMappingListByConfigId(mapping.CalculateResidualAnalysisConfigId)
  470. if err != nil {
  471. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  472. }
  473. var edbInfoIdList []int64
  474. for _, v := range configMappingList {
  475. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  476. }
  477. condition = ""
  478. pars = []interface{}{}
  479. condition += ` and edb_info_id in(` + utils.GetOrmInReplace(len(edbInfoIdList)) + `)`
  480. for _, id := range edbInfoIdList {
  481. pars = append(pars, id)
  482. }
  483. edbInfoList, err := data_manage.GetEdbInfoListByCond(condition, pars)
  484. if err != nil {
  485. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  486. }
  487. // 获取配置
  488. configInfo, err := residual_analysis_model.GetResidualAnalysisConfigById(mapping.CalculateResidualAnalysisConfigId)
  489. if err != nil {
  490. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  491. }
  492. resp := residual_analysis_model.ResidualAnalysisDetailResp{
  493. ConfigInfo: &configInfo,
  494. EdbInfoList: edbInfoList,
  495. }
  496. return resp, nil
  497. }
  498. func SaveResidualAnalysisConfig(req residual_analysis_model.ResidualAnalysisReq, sysUser *system.Admin) error {
  499. config := residual_analysis_model.ResidualAnalysisConfigVo{
  500. DateType: req.DateType,
  501. StartDate: req.StartDate,
  502. EndDate: req.EndDate,
  503. IsOrder: req.IsOrder,
  504. IndexType: req.IndexType,
  505. LeadValue: req.LeadValue,
  506. LeadFrequency: req.LeadFrequency,
  507. LeftIndexMin: req.LeftIndexMin,
  508. LeftIndexMax: req.LeftIndexMax,
  509. RightIndexMin: req.RightIndexMin,
  510. RightIndexMax: req.RightIndexMax,
  511. ResidualIndexMin: req.ResidualIndexMin,
  512. ResidualIndexMax: req.ResidualIndexMax,
  513. ContrastIndexMin: req.ContrastIndexMin,
  514. ContrastIndexMax: req.ContrastIndexMax,
  515. }
  516. // 转换为json格式
  517. configJson, err := json.Marshal(config)
  518. if err != nil {
  519. return err
  520. }
  521. var condition string
  522. var pars []interface{}
  523. // 新增or更新
  524. if req.EdbInfoId > 0 {
  525. condition += " and edb_info_id=?"
  526. pars = append(pars, req.EdbInfoId)
  527. configMappings, err := residual_analysis_model.GetConfigMappingListByCondition(condition, pars)
  528. if err != nil {
  529. return err
  530. }
  531. if len(configMappings) > 0 {
  532. mapping := configMappings[0]
  533. configInfo, err := residual_analysis_model.GetResidualAnalysisConfigById(mapping.CalculateResidualAnalysisConfigId)
  534. if err != nil {
  535. return err
  536. }
  537. configInfo.Config = string(configJson)
  538. err = residual_analysis_model.UpdateResidualAnalysisConfig(configInfo)
  539. if err != nil {
  540. return err
  541. }
  542. }
  543. }
  544. analysisConfig := residual_analysis_model.CalculateResidualAnalysisConfig{
  545. Config: string(configJson),
  546. SysUserId: sysUser.AdminId,
  547. CreateTime: time.Now(),
  548. ModifyTime: time.Now(),
  549. }
  550. _, err = residual_analysis_model.SaveResidualAnalysisConfig(analysisConfig)
  551. if err != nil {
  552. return err
  553. }
  554. return nil
  555. }