residual_analysis_service.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822
  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. mappingList, err := data_manage.GetChartEdbMappingListByEdbInfoIdList([]int{req.EdbInfoIdA, req.EdbInfoIdB})
  15. if err != nil {
  16. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("获取图表,指标信息失败,Err:%s", err.Error())
  17. }
  18. var edbInfoMappingA, edbInfoMappingB *data_manage.ChartEdbInfoMapping
  19. for _, v := range mappingList {
  20. if v.EdbInfoId == req.EdbInfoIdA {
  21. edbInfoMappingA = v
  22. }
  23. if v.EdbInfoId == req.EdbInfoIdB {
  24. edbInfoMappingB = v
  25. }
  26. }
  27. if edbInfoMappingA == nil {
  28. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("指标A不存在")
  29. }
  30. if edbInfoMappingB == nil {
  31. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("指标B不存在")
  32. }
  33. // 时间处理
  34. var startDate, endDate string
  35. switch req.DateType {
  36. case 0:
  37. startDate = req.StartDate
  38. endDate = req.EndDate
  39. case 1:
  40. startDate = req.StartDate
  41. endDate = ""
  42. default:
  43. startDate = utils.GetPreYearTime(req.DateType)
  44. endDate = ""
  45. }
  46. resp := residual_analysis_model.ResidualAnalysisResp{}
  47. // 原始图表信息
  48. originalEdbList := make([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, 0)
  49. originalEdbList, err = fillOriginalChart(req, mappingList, startDate, endDate, edbInfoMappingA, edbInfoMappingB, originalEdbList)
  50. if err != nil {
  51. return residual_analysis_model.ResidualAnalysisResp{}, err
  52. }
  53. originalChartInfo := createChartInfoResp(req, startDate, endDate, edbInfoMappingA.EdbName+"与"+edbInfoMappingB.EdbName)
  54. resp.OriginalChartData = residual_analysis_model.ChartResp{
  55. ChartInfo: originalChartInfo,
  56. EdbInfoList: originalEdbList,
  57. }
  58. // 如果只需要第一张图表的数据 直接返回,避免继续处理
  59. if req.QueryType == 1 {
  60. return resp, nil
  61. }
  62. dataAList, ok := edbInfoMappingA.DataList.([]*data_manage.EdbDataList)
  63. if !ok {
  64. return residual_analysis_model.ResidualAnalysisResp{}, fmt.Errorf("数据类型转换失败")
  65. }
  66. indexADataMap := map[string]*data_manage.EdbDataList{}
  67. for _, indexData := range dataAList {
  68. indexADataMap[indexData.DataTime] = indexData
  69. }
  70. // 映射图表信息
  71. mappingEdbList, a, b, r, err := fillMappingChartInfo(req, edbInfoMappingA, edbInfoMappingB, originalEdbList, indexADataMap)
  72. if err != nil {
  73. return residual_analysis_model.ResidualAnalysisResp{}, err
  74. }
  75. mappingChartInfo := createChartInfoResp(req, startDate, endDate, edbInfoMappingA.EdbName+"与"+edbInfoMappingB.EdbName+"映射"+edbInfoMappingA.EdbName)
  76. resp.MappingChartData = residual_analysis_model.ChartResp{
  77. ChartInfo: mappingChartInfo,
  78. EdbInfoList: mappingEdbList,
  79. }
  80. // 残差图表信息
  81. residualEdbList, R2, err := fillResidualChartInfo(edbInfoMappingA, edbInfoMappingB, mappingEdbList)
  82. if err != nil {
  83. return residual_analysis_model.ResidualAnalysisResp{}, err
  84. }
  85. residualChartInfo := createChartInfoResp(req, startDate, endDate, edbInfoMappingA.EdbName+"与"+edbInfoMappingA.EdbName+"映射残差/"+edbInfoMappingB.EdbName)
  86. resp.ResidualChartData = residual_analysis_model.ChartResp{
  87. ChartInfo: residualChartInfo,
  88. EdbInfoList: residualEdbList,
  89. }
  90. if req.ResidualType == 2 {
  91. resp.A = a
  92. resp.B = b
  93. resp.R = r
  94. resp.R2 = R2
  95. }
  96. return resp, nil
  97. }
  98. func createChartInfoResp(req residual_analysis_model.ResidualAnalysisReq, startDate, endDate, chartName string) residual_analysis_model.ResidualAnalysisChartInfo {
  99. return residual_analysis_model.ResidualAnalysisChartInfo{
  100. Calendar: `公历`,
  101. Source: utils.CHART_SOURCE_DEFAULT,
  102. DateType: req.DateType,
  103. StartDate: startDate,
  104. EndDate: endDate,
  105. ChartType: utils.CHART_TYPE_CURVE,
  106. ChartName: chartName,
  107. }
  108. }
  109. func fillResidualChartInfo(edbInfoMappingA *data_manage.ChartEdbInfoMapping, edbInfoMappingB *data_manage.ChartEdbInfoMapping, mappingEdbList []residual_analysis_model.ResidualAnalysisChartEdbInfoMapping) ([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, float64, error) {
  110. // 计算公式 映射残差 = 因变量指标 - 映射指标
  111. var edbInfoA, edbInfoB residual_analysis_model.ResidualAnalysisChartEdbInfoMapping
  112. if mappingEdbList[0].EdbInfoId == edbInfoMappingA.EdbInfoId {
  113. edbInfoA = mappingEdbList[0]
  114. edbInfoB = mappingEdbList[1]
  115. } else {
  116. edbInfoA = mappingEdbList[1]
  117. edbInfoB = mappingEdbList[0]
  118. }
  119. dataAList, ok := edbInfoA.DataList.([]*data_manage.EdbDataList)
  120. if !ok {
  121. return nil, 0, fmt.Errorf("数据类型转换失败")
  122. }
  123. edbData := dataAList
  124. dataBList, ok := edbInfoB.DataList.([]*data_manage.EdbDataList)
  125. if !ok {
  126. return nil, 0, fmt.Errorf("数据类型转换失败")
  127. }
  128. var indexDataBMap = make(map[string]*data_manage.EdbDataList)
  129. for _, data := range dataBList {
  130. indexDataBMap[data.DataTime] = data
  131. }
  132. // 求R2
  133. var valueB, sumValueA, averageValueA, residualQuadraticSum, totalQuadraticSum, R2 float64
  134. for _, indexData := range edbData {
  135. // 因变量的值总和
  136. sumValueA += indexData.Value
  137. }
  138. // 因变量平均值
  139. averageValueA = sumValueA / float64(len(edbData))
  140. for _, indexData := range edbData {
  141. if dataB, ok := indexDataBMap[indexData.DataTime]; ok {
  142. valueB = dataB.Value
  143. } else {
  144. valueB = 0
  145. }
  146. // 总因变量平方和
  147. totalQuadraticSum += math.Pow(indexData.Value-averageValueA, 2)
  148. // 补全残差值
  149. indexData.Value = indexData.Value - valueB
  150. // 残差平方和
  151. residualQuadraticSum += math.Pow(indexData.Value, 2)
  152. }
  153. // 计算R2 公式:R2=1-SSE/SST R2越大,越符合线性 R2 = 1 - 残差平方和/总平方和
  154. R2 = 1 - residualQuadraticSum/totalQuadraticSum
  155. mappingEdb := make([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, len(mappingEdbList))
  156. copy(mappingEdb, mappingEdbList)
  157. for i, mapping := range mappingEdb {
  158. if mapping.EdbInfoId != edbInfoMappingA.EdbInfoId {
  159. mappingEdb[i].DataList = edbData
  160. mappingEdb[i].EdbName = edbInfoMappingA.EdbName + "映射残差/" + edbInfoMappingB.EdbName
  161. mappingEdb[i].IsAxis = 1
  162. mappingEdb[i].ChartColor = `#00F`
  163. mappingEdb[i].IsOrder = false
  164. } else {
  165. mappingEdb[i].IsAxis = 0
  166. mappingEdb[i].ChartColor = `#F00`
  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.RightIndexMin - req.RightIndexMax)
  181. b = req.LeftIndexMax - req.RightIndexMin*a
  182. } else {
  183. a = (req.LeftIndexMax - req.LeftIndexMin) / (req.RightIndexMax - req.RightIndexMin)
  184. b = req.LeftIndexMax - req.RightIndexMax*a
  185. }
  186. }
  187. dataList, ok := edbInfoMappingB.DataList.([]*data_manage.EdbDataList)
  188. if !ok {
  189. return nil, a, b, r, fmt.Errorf("数据类型转换失败")
  190. }
  191. // 领先指标 dataList进行数据处理
  192. if req.IndexType == 2 {
  193. if req.LeadValue < 0 {
  194. return nil, a, b, r, fmt.Errorf("领先值不能小于0")
  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. indexDataCopy := *indexData
  249. // 计算指标B映射值
  250. indexDataCopy.Value = math.Round((a*indexData.Value+b)*10000) / 10000
  251. // 将副本添加到 dataBList
  252. dataBList = append(dataBList, &indexDataCopy)
  253. }
  254. }
  255. mappingEdbList := make([]residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, len(originalEdbList))
  256. copy(mappingEdbList, originalEdbList)
  257. for i, mapping := range mappingEdbList {
  258. if mapping.EdbInfoId != req.EdbInfoIdA {
  259. mappingEdbList[i].EdbInfoId = 0
  260. mappingEdbList[i].EdbCode = ""
  261. mappingEdbList[i].IsAxis = 1
  262. mappingEdbList[i].EdbName = edbInfoMappingB.EdbName + "映射" + edbInfoMappingA.EdbName
  263. mappingEdbList[i].DataList = dataBList
  264. }
  265. }
  266. return mappingEdbList, a, b, r, nil
  267. }
  268. 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) {
  269. for _, v := range mappingList {
  270. var edbInfoMapping residual_analysis_model.ResidualAnalysisChartEdbInfoMapping
  271. edbInfoMapping.EdbInfoType = 1
  272. edbInfoMapping.IsOrder = false
  273. edbInfoMapping.IsAxis = 1
  274. edbInfoMapping.ChartColor = `#00F`
  275. edbInfoMapping.ChartWidth = 3
  276. // 获取图表中的指标数据
  277. dataList, err := data_manage.GetEdbDataList(v.Source, v.SubSource, v.EdbInfoId, startDate, endDate)
  278. if err != nil {
  279. return nil, fmt.Errorf("获取指标数据失败,Err:%s", err.Error())
  280. }
  281. if v.EdbInfoId == req.EdbInfoIdB {
  282. edbInfoMapping.LeadValue = req.LeadValue
  283. edbInfoMapping.LeadUnit = req.LeadFrequency
  284. edbInfoMapping.EdbInfoType = req.IndexType
  285. edbInfoMapping.IsOrder = req.IsOrder
  286. edbInfoMapping.IsAxis = 0
  287. edbInfoMapping.ChartColor = `#F00`
  288. edbInfoMapping.ChartWidth = 1
  289. edbInfoMappingB.DataList = dataList
  290. } else {
  291. edbInfoMappingA.DataList = dataList
  292. }
  293. edbInfoMapping.EdbInfoId = v.EdbInfoId
  294. edbInfoMapping.EdbName = v.EdbName
  295. edbInfoMapping.EdbCode = v.EdbCode
  296. edbInfoMapping.Unit = v.Unit
  297. edbInfoMapping.Frequency = v.Frequency
  298. edbInfoMapping.Source = v.Source
  299. edbInfoMapping.SourceName = v.SourceName
  300. edbInfoMapping.MinValue = v.MinValue
  301. edbInfoMapping.MaxValue = v.MaxValue
  302. edbInfoMapping.LatestDate = v.LatestDate
  303. edbInfoMapping.LatestValue = v.LatestValue
  304. edbInfoMapping.DataList = dataList
  305. originalEdbList = append(originalEdbList, edbInfoMapping)
  306. }
  307. return originalEdbList, nil
  308. }
  309. func ContrastPreview(indexCode string) (residual_analysis_model.ResidualAnalysisChartEdbInfoMapping, error) {
  310. var condition string
  311. var pars []interface{}
  312. if indexCode != "" {
  313. condition += " and edb_code=?"
  314. pars = append(pars, indexCode)
  315. }
  316. edbInfo, err := data_manage.GetEdbInfoByCondition(condition, pars)
  317. if err != nil {
  318. return residual_analysis_model.ResidualAnalysisChartEdbInfoMapping{}, err
  319. }
  320. if edbInfo == nil {
  321. return residual_analysis_model.ResidualAnalysisChartEdbInfoMapping{}, fmt.Errorf("指标不存在")
  322. }
  323. dataList, err := data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, "", "")
  324. var resp residual_analysis_model.ResidualAnalysisChartEdbInfoMapping
  325. resp.EdbInfoId = edbInfo.EdbInfoId
  326. resp.EdbCode = edbInfo.EdbCode
  327. resp.ChartColor = "#F00"
  328. resp.SourceName = edbInfo.SourceName
  329. resp.EdbName = edbInfo.EdbName
  330. resp.Unit = edbInfo.Unit
  331. resp.Frequency = edbInfo.Frequency
  332. resp.MinValue = edbInfo.MinValue
  333. resp.MaxValue = edbInfo.MaxValue
  334. resp.DataList = dataList
  335. return resp, nil
  336. }
  337. func SaveResidualAnalysis(req residual_analysis_model.ResidualAnalysisIndexSaveReq, sysUser *system.Admin) error {
  338. // 验证分类是否存在
  339. classifyCount, err := data_manage.GetEdbClassifyCountById(req.ClassifyId)
  340. if err != nil {
  341. return err
  342. }
  343. if classifyCount <= 0 {
  344. return fmt.Errorf("分类不存在")
  345. }
  346. // 校验名称是否重复
  347. var condition string
  348. var pars []interface{}
  349. condition += " and source = ? AND edb_name=?"
  350. pars = append(pars, req.Source, req.EdbName)
  351. edbInfoByCondition, err := data_manage.GetEdbInfoByCondition(condition, pars)
  352. if err != nil && err.Error() != utils.ErrNoRow() {
  353. return err
  354. }
  355. // 获取指标数据最大值 最小值 最后更新时间 最后更新时间对应的值
  356. var indexMax, indexMin, indexLatestValue float64
  357. var indexLatestDate string
  358. if len(req.DataList) > 0 {
  359. latestTime, _ := time.Parse(utils.YearMonthDay, req.DataList[0].DataTime)
  360. for _, data := range req.DataList {
  361. // 比较最大值
  362. if data.Value > indexMax {
  363. indexMax = data.Value
  364. }
  365. // 比较最小值
  366. if data.Value < indexMin {
  367. indexMin = data.Value
  368. }
  369. // 比较最新时间和对应值
  370. currentTime, err := time.Parse(utils.YearMonthDay, data.DataTime)
  371. if err != nil {
  372. // 时间解析失败,跳过此项
  373. continue
  374. }
  375. // 如果当前时间更晚
  376. if currentTime.After(latestTime) {
  377. latestTime = currentTime
  378. indexLatestDate = data.DataTime
  379. indexLatestValue = data.Value
  380. }
  381. }
  382. }
  383. // 更新保存指标和配置的映射关系
  384. mappingList, err := residual_analysis_model.GetConfigMappingListByConfigId(req.ConfigId)
  385. if err != nil {
  386. return err
  387. }
  388. // 判断是更新还是修改 看指标配置映射中,是否存在对应指标 存在 则更新 不存在 则新增
  389. var edbInfoMapping residual_analysis_model.CalculateResidualAnalysisConfigMapping
  390. for _, mapping := range mappingList {
  391. if req.IndexType == mapping.IndexType && req.IndexType != 0 {
  392. edbInfoMapping = mapping
  393. }
  394. }
  395. var edbInfoId int64
  396. var edbCode string
  397. // 更新or新增
  398. if edbInfoMapping.EdbInfoId > 0 {
  399. // 查询指标库指标
  400. edbInfo, err := data_manage.GetEdbInfoById(int(edbInfoMapping.EdbInfoId))
  401. if err != nil {
  402. return err
  403. }
  404. if edbInfo == nil {
  405. return fmt.Errorf("指标不存在")
  406. }
  407. edbInfoId = int64(edbInfo.EdbInfoId)
  408. edbCode = edbInfo.EdbCode
  409. if edbInfoByCondition != nil && edbInfoByCondition.EdbInfoId != edbInfo.EdbInfoId {
  410. return fmt.Errorf("指标名称重复")
  411. }
  412. // 须补充更新指标最大值,最小值,数据最新时间,数据最新值
  413. edbInfo.MaxValue = indexMax
  414. edbInfo.MinValue = indexMin
  415. edbInfo.LatestDate = indexLatestDate
  416. edbInfo.LatestValue = indexLatestValue
  417. edbInfo.Unit = req.Unit
  418. edbInfo.Frequency = req.Frequency
  419. edbInfo.ClassifyId = req.ClassifyId
  420. edbInfo.EdbName = req.EdbName
  421. err = edbInfo.Update([]string{"min_value", "max_value", "latest_date", "latest_value", "unit", "frequency", "classify_id", "edb_name"})
  422. if err != nil {
  423. return err
  424. }
  425. // 删除对应得指标数据
  426. err = residual_analysis_model.DeleteResidualAnalysisDataByEdbCode(edbInfo.EdbCode)
  427. if err != nil {
  428. return fmt.Errorf("删除指标数据失败")
  429. }
  430. } else {
  431. if edbInfoByCondition != nil {
  432. return fmt.Errorf("指标名称重复")
  433. }
  434. // 新增指标
  435. edbCode, err = utils.GenerateEdbCode(1, "")
  436. if err != nil {
  437. return err
  438. }
  439. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  440. edbInfoId, err = data_manage.AddEdbInfo(&data_manage.EdbInfo{
  441. EdbCode: edbCode,
  442. UniqueCode: utils.MD5(utils.CHART_PREFIX + "_" + timestamp),
  443. EdbName: req.EdbName,
  444. EdbNameEn: req.EdbNameEn,
  445. ClassifyId: req.ClassifyId,
  446. EdbType: req.EdbType,
  447. Unit: req.Unit,
  448. UnitEn: req.UnitEn,
  449. Frequency: req.Frequency,
  450. Source: req.Source,
  451. SourceName: "残差分析",
  452. Calendar: req.Calendar,
  453. SysUserRealName: sysUser.RealName,
  454. SysUserId: sysUser.AdminId,
  455. LatestDate: indexLatestDate,
  456. LatestValue: indexLatestValue,
  457. MinValue: indexMin,
  458. MaxValue: indexMax,
  459. CreateTime: time.Now(),
  460. ModifyTime: time.Now(),
  461. })
  462. if err != nil {
  463. return err
  464. }
  465. // 新增指标配置关系
  466. _, err = residual_analysis_model.SaveConfigMapping(residual_analysis_model.CalculateResidualAnalysisConfigMapping{
  467. CalculateResidualAnalysisConfigId: req.ConfigId,
  468. EdbInfoId: edbInfoId,
  469. ResidualType: req.ResidualType,
  470. IndexType: req.IndexType,
  471. CreateTime: time.Now(),
  472. ModifyTime: time.Now(),
  473. })
  474. if err != nil {
  475. return err
  476. }
  477. }
  478. // 新增数据
  479. for i := range req.DataList {
  480. req.DataList[i].EdbDataId = 0
  481. req.DataList[i].EdbInfoId = int(edbInfoId)
  482. req.DataList[i].EdbCode = edbCode
  483. }
  484. _, err = residual_analysis_model.AddResidualAnalysisData(req.DataList)
  485. if err != nil {
  486. return err
  487. }
  488. // 新增自变量 因变量与配置得关系 配置中不存在该指标,则新增
  489. var indexMap = make(map[int64]residual_analysis_model.CalculateResidualAnalysisConfigMapping)
  490. for _, mapping := range mappingList {
  491. indexMap[mapping.EdbInfoId] = mapping
  492. }
  493. if _, ok := indexMap[int64(req.EdbInfoIdA)]; !ok {
  494. _, err = residual_analysis_model.SaveConfigMapping(residual_analysis_model.CalculateResidualAnalysisConfigMapping{
  495. CalculateResidualAnalysisConfigId: req.ConfigId,
  496. EdbInfoId: int64(req.EdbInfoIdA),
  497. ResidualType: req.ResidualType,
  498. IndexType: 3,
  499. CreateTime: time.Now(),
  500. ModifyTime: time.Now(),
  501. })
  502. if err != nil {
  503. return err
  504. }
  505. }
  506. if _, ok := indexMap[int64(req.EdbInfoIdB)]; !ok {
  507. _, err = residual_analysis_model.SaveConfigMapping(residual_analysis_model.CalculateResidualAnalysisConfigMapping{
  508. CalculateResidualAnalysisConfigId: req.ConfigId,
  509. EdbInfoId: int64(req.EdbInfoIdB),
  510. ResidualType: req.ResidualType,
  511. IndexType: 4,
  512. CreateTime: time.Now(),
  513. ModifyTime: time.Now(),
  514. })
  515. if err != nil {
  516. return err
  517. }
  518. }
  519. return nil
  520. }
  521. func ResidualAnalysisDetail(edbInfoId int) (residual_analysis_model.ResidualAnalysisDetailResp, error) {
  522. // 通过指标配置映射表 拿到配置id,再获取关联的所有指标信息
  523. var condition string
  524. var pars []interface{}
  525. condition += " and edb_info_id=?"
  526. pars = append(pars, edbInfoId)
  527. mappingList, err := residual_analysis_model.GetConfigMappingListByCondition(condition, pars)
  528. if err != nil {
  529. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  530. }
  531. if len(mappingList) <= 0 {
  532. return residual_analysis_model.ResidualAnalysisDetailResp{}, fmt.Errorf("指标不存在")
  533. }
  534. mapping := mappingList[0]
  535. configMappingList, err := residual_analysis_model.GetConfigMappingListByConfigId(mapping.CalculateResidualAnalysisConfigId)
  536. if err != nil {
  537. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  538. }
  539. var edbInfoIdList []int64
  540. var edbInfoMap = make(map[int64]residual_analysis_model.CalculateResidualAnalysisConfigMapping)
  541. var mappgingFlag = false
  542. var residualFlag = false
  543. for _, v := range configMappingList {
  544. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  545. edbInfoMap[v.EdbInfoId] = v
  546. if v.IndexType == 1 {
  547. mappgingFlag = true
  548. } else if v.IndexType == 2 {
  549. residualFlag = true
  550. }
  551. }
  552. condition = ""
  553. pars = []interface{}{}
  554. condition += ` and edb_info_id in(` + utils.GetOrmInReplace(len(edbInfoIdList)) + `)`
  555. for _, id := range edbInfoIdList {
  556. pars = append(pars, id)
  557. }
  558. edbInfoList, err := data_manage.GetEdbInfoListByCond(condition, pars)
  559. if err != nil {
  560. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  561. }
  562. // 获取配置
  563. configInfo, err := residual_analysis_model.GetResidualAnalysisConfigById(mapping.CalculateResidualAnalysisConfigId)
  564. if err != nil {
  565. return residual_analysis_model.ResidualAnalysisDetailResp{}, err
  566. }
  567. var edbInfoListResp []*residual_analysis_model.DetailEdbInfoList
  568. var dependentEdbInfo residual_analysis_model.DetailEdbInfoList
  569. var independentEdbInfo residual_analysis_model.DetailEdbInfoList
  570. for _, edbInfo := range edbInfoList {
  571. var indexType int
  572. if _, ok := edbInfoMap[int64(edbInfo.EdbInfoId)]; ok {
  573. indexType = edbInfoMap[int64(edbInfo.EdbInfoId)].IndexType
  574. }
  575. info := residual_analysis_model.DetailEdbInfoList{
  576. EdbInfoId: edbInfo.EdbInfoId,
  577. EdbInfoType: edbInfo.EdbInfoType,
  578. IndexType: indexType,
  579. SourceName: edbInfo.SourceName,
  580. Source: edbInfo.Source,
  581. EdbCode: edbInfo.EdbCode,
  582. EdbName: edbInfo.EdbName,
  583. EdbNameEn: edbInfo.EdbNameEn,
  584. Unit: edbInfo.Unit,
  585. UnitEn: edbInfo.UnitEn,
  586. Frequency: edbInfo.Frequency,
  587. FrequencyEn: edbInfo.FrequencyEn,
  588. ClassifyId: edbInfo.ClassifyId,
  589. }
  590. edbInfoListResp = append(edbInfoListResp, &info)
  591. if indexType == 3 {
  592. dependentEdbInfo = info
  593. } else if indexType == 4 {
  594. independentEdbInfo = info
  595. }
  596. }
  597. // 补充表格中 映射指标或者残差指标
  598. if mappgingFlag && !residualFlag {
  599. info := residual_analysis_model.DetailEdbInfoList{
  600. IndexType: 2,
  601. EdbName: independentEdbInfo.EdbName + "映射残差/" + dependentEdbInfo.EdbName,
  602. EdbNameEn: dependentEdbInfo.EdbNameEn,
  603. Unit: dependentEdbInfo.Unit,
  604. UnitEn: dependentEdbInfo.UnitEn,
  605. Frequency: dependentEdbInfo.Frequency,
  606. FrequencyEn: dependentEdbInfo.FrequencyEn,
  607. ClassifyId: dependentEdbInfo.ClassifyId,
  608. }
  609. edbInfoListResp = append(edbInfoListResp, &info)
  610. } else if !mappgingFlag && residualFlag {
  611. info := residual_analysis_model.DetailEdbInfoList{
  612. IndexType: 1,
  613. EdbName: dependentEdbInfo.EdbName + "映射" + independentEdbInfo.EdbName,
  614. EdbNameEn: dependentEdbInfo.EdbNameEn,
  615. Unit: dependentEdbInfo.Unit,
  616. UnitEn: dependentEdbInfo.UnitEn,
  617. Frequency: dependentEdbInfo.Frequency,
  618. FrequencyEn: dependentEdbInfo.FrequencyEn,
  619. ClassifyId: dependentEdbInfo.ClassifyId,
  620. }
  621. edbInfoListResp = append(edbInfoListResp, &info)
  622. }
  623. resp := residual_analysis_model.ResidualAnalysisDetailResp{
  624. ConfigInfo: &configInfo,
  625. EdbInfoList: edbInfoListResp,
  626. ResidualType: mapping.ResidualType,
  627. }
  628. return resp, nil
  629. }
  630. func SaveResidualAnalysisConfig(req residual_analysis_model.ResidualAnalysisReq, sysUser *system.Admin) (int64, error) {
  631. config := residual_analysis_model.ResidualAnalysisConfigVo{
  632. DateType: req.DateType,
  633. StartDate: req.StartDate,
  634. EndDate: req.EndDate,
  635. IsOrder: req.IsOrder,
  636. IndexType: req.IndexType,
  637. LeadValue: req.LeadValue,
  638. LeadFrequency: req.LeadFrequency,
  639. LeftIndexMin: req.LeftIndexMin,
  640. LeftIndexMax: req.LeftIndexMax,
  641. RightIndexMin: req.RightIndexMin,
  642. RightIndexMax: req.RightIndexMax,
  643. ResidualIndexMin: req.ResidualIndexMin,
  644. ResidualIndexMax: req.ResidualIndexMax,
  645. ContrastIndexMin: req.ContrastIndexMin,
  646. ContrastIndexMax: req.ContrastIndexMax,
  647. }
  648. // 转换为json格式
  649. configJson, err := json.Marshal(config)
  650. if err != nil {
  651. return 0, err
  652. }
  653. // 新增or更新
  654. /*
  655. var condition string
  656. var pars []interface{}
  657. if req.EdbInfoId > 0 {
  658. condition += " and edb_info_id=?"
  659. pars = append(pars, req.EdbInfoId)
  660. configMappings, err := residual_analysis_model.GetConfigMappingListByCondition(condition, pars)
  661. if err != nil {
  662. return 0, err
  663. }
  664. if len(configMappings) > 0 {
  665. mapping := configMappings[0]
  666. configInfo, err := residual_analysis_model.GetResidualAnalysisConfigById(mapping.CalculateResidualAnalysisConfigId)
  667. if err != nil {
  668. return 0, err
  669. }
  670. configInfo.Config = string(configJson)
  671. err = residual_analysis_model.UpdateResidualAnalysisConfig(configInfo)
  672. if err != nil {
  673. return 0, err
  674. }
  675. }
  676. }*/
  677. var configId int64
  678. if req.ConfigId > 0 {
  679. configInfo, err := residual_analysis_model.GetResidualAnalysisConfigById(req.ConfigId)
  680. if err != nil {
  681. return 0, err
  682. }
  683. configId = int64(configInfo.CalculateResidualAnalysisConfigId)
  684. configInfo.Config = string(configJson)
  685. err = residual_analysis_model.UpdateResidualAnalysisConfig(configInfo)
  686. if err != nil {
  687. return 0, err
  688. }
  689. }
  690. analysisConfig := residual_analysis_model.CalculateResidualAnalysisConfig{
  691. Config: string(configJson),
  692. SysUserId: sysUser.AdminId,
  693. CreateTime: time.Now(),
  694. ModifyTime: time.Now(),
  695. }
  696. configId, err = residual_analysis_model.SaveResidualAnalysisConfig(analysisConfig)
  697. if err != nil {
  698. return 0, err
  699. }
  700. return configId, nil
  701. }
  702. func CheckResidualAnalysisExist(configId int) (int64, error) {
  703. configMappingList, err := residual_analysis_model.GetConfigMappingListByConfigId(configId)
  704. if err != nil {
  705. return 0, err
  706. }
  707. var configMapping residual_analysis_model.CalculateResidualAnalysisConfigMapping
  708. for _, mapping := range configMappingList {
  709. if mapping.IndexType == 2 {
  710. configMapping = mapping
  711. }
  712. }
  713. return configMapping.EdbInfoId, nil
  714. }