eta_api/models/data_manage/line_equation/line_equation.go

package line_equation

import (
	"eta_gn/eta_api/global"
	"eta_gn/eta_api/models/data_manage"
	"eta_gn/eta_api/utils"
)

type BatchAddChart struct {
	ChartInfo                       *data_manage.ChartInfo
	EdbMappingList                  []*data_manage.ChartEdbMapping
	MultipleGraphConfigChartMapping *data_manage.MultipleGraphConfigChartMapping
}

func CreateLineEquationChartAndEdb(multipleGraphConfig *data_manage.MultipleGraphConfig, batchAddChartList []BatchAddChart) (err error) {
	tx := global.DmSQL["data"].Begin()
	defer func() {
		if err != nil {
			_ = tx.Rollback()
		} else {
			_ = tx.Commit()
		}
	}()
	err = tx.Create(multipleGraphConfig).Error
	if err != nil {
		return
	}

	for k, v := range batchAddChartList {
		chartInfo := v.ChartInfo
		err = tx.Create(chartInfo).Error
		if err != nil {
			return
		}
		batchAddChartList[k].ChartInfo = chartInfo

		edbMappingList := v.EdbMappingList
		if len(edbMappingList) > 0 {
			for i := range edbMappingList {
				edbMappingList[i].ChartInfoId = chartInfo.ChartInfoId
			}
			err = tx.CreateInBatches(edbMappingList, utils.MultiAddNum).Error
			if err != nil {
				return
			}
		}
		batchAddChartList[k].EdbMappingList = edbMappingList

		multipleGraphConfigChartMapping := v.MultipleGraphConfigChartMapping
		multipleGraphConfigChartMapping.ChartInfoId = chartInfo.ChartInfoId
		multipleGraphConfigChartMapping.MultipleGraphConfigId = multipleGraphConfig.MultipleGraphConfigId
		err = tx.Create(v.MultipleGraphConfigChartMapping).Error
		if err != nil {
			return
		}
	}

	return
}

func CreateAndEditLineEquationChartAndEdb(multipleGraphConfig *data_manage.MultipleGraphConfig, batchAddChartList []BatchAddChart, updateMultipleGraphConfigCols, updateChartCols []string) (err error) {
	tx := global.DmSQL["data"].Begin()
	defer func() {
		if err != nil {
			_ = tx.Rollback()
		} else {
			_ = tx.Commit()
		}
	}()
	err = tx.Select(updateMultipleGraphConfigCols).Updates(multipleGraphConfig).Error
	if err != nil {
		return
	}

	for k, v := range batchAddChartList {
		chartInfo := v.ChartInfo
		var isDelMapping bool
		{
			if chartInfo.ChartInfoId <= 0 {
				err = tx.Create(chartInfo).Error
				if err != nil {
					return
				}
				batchAddChartList[k].ChartInfo = chartInfo
			} else {
				err = tx.Select(updateChartCols).Updates(chartInfo).Error
				if err != nil {
					return
				}
				isDelMapping = true
			}
		}

		{
			if isDelMapping {
				sql := ` DELETE FROM  chart_edb_mapping WHERE chart_info_id=? `
				err = tx.Exec(sql, chartInfo.ChartInfoId).Error
				if err != nil {
					return
				}
			}
			edbMappingList := v.EdbMappingList
			if len(edbMappingList) > 0 {
				for i := range edbMappingList {
					edbMappingList[i].ChartInfoId = chartInfo.ChartInfoId
				}
				err = tx.CreateInBatches(edbMappingList, utils.MultiAddNum).Error
				if err != nil {
					return
				}
			}
			batchAddChartList[k].EdbMappingList = edbMappingList
		}

		{
			multipleGraphConfigChartMapping := v.MultipleGraphConfigChartMapping
			if multipleGraphConfigChartMapping.Id <= 0 {
				multipleGraphConfigChartMapping.ChartInfoId = chartInfo.ChartInfoId
				multipleGraphConfigChartMapping.MultipleGraphConfigId = multipleGraphConfig.MultipleGraphConfigId
				err = tx.Create(v.MultipleGraphConfigChartMapping).Error
				if err != nil {
					return
				}
			} else {
				if multipleGraphConfigChartMapping.ChartInfoId != chartInfo.ChartInfoId {
					multipleGraphConfigChartMapping.ChartInfoId = chartInfo.ChartInfoId
					err = tx.Select("ChartInfoId").Updates(multipleGraphConfigChartMapping).Error
					if err != nil {
						return
					}
				}
			}
			batchAddChartList[k].MultipleGraphConfigChartMapping = multipleGraphConfigChartMapping
		}
	}

	return
}