残差分析 生成指标存es-指标溯源

services/residual_analysis_service/residual_analysis_service.go

@@ -5,6 +5,7 @@ import (
 	"eta/eta_api/models/data_manage"
 	"eta/eta_api/models/residual_analysis_model"
 	"eta/eta_api/models/system"
+	"eta/eta_api/services/data"
 	"eta/eta_api/utils"
 	"fmt"
 	"math"
@@ -815,6 +816,37 @@ func SaveResidualAnalysis(req residual_analysis_model.ResidualAnalysisIndexSaveR
 		}
 	}
 
+	// 保存溯源信息
+	{
+		fromEdbInfo, err := data_manage.GetEdbInfoById(int(edbInfoId))
+		if err != nil {
+			if err.Error() == utils.ErrNoRow() {
+				return fmt.Errorf("指标不存在,err:" + err.Error())
+			}
+			return fmt.Errorf("获取失败")
+		}
+
+		edbCalculateMappingInfo := new(data_manage.EdbInfoCalculateMapping)
+		edbCalculateMappingInfo.EdbInfoId = int(edbInfoId)
+		edbCalculateMappingInfo.Source = req.Source
+		edbCalculateMappingInfo.SourceName = "残差分析"
+		edbCalculateMappingInfo.EdbCode = edbCode
+		edbCalculateMappingInfo.FromEdbInfoId = fromEdbInfo.EdbInfoId
+		edbCalculateMappingInfo.FromEdbCode = fromEdbInfo.EdbCode
+		edbCalculateMappingInfo.FromEdbName = fromEdbInfo.EdbName
+		edbCalculateMappingInfo.FromSource = fromEdbInfo.Source
+		edbCalculateMappingInfo.FromSourceName = fromEdbInfo.SourceName
+		edbCalculateMappingInfo.CreateTime = time.Now()
+		edbCalculateMappingInfo.ModifyTime = time.Now()
+		err = edbCalculateMappingInfo.Insert()
+		if err != nil {
+			return fmt.Errorf("保存溯源信息失败，Err:" + err.Error())
+		}
+	}
+
+	//添加es
+	data.AddOrEditEdbInfoToEs(int(edbInfoId))
+
 	return nil
 }