package data_manage

import (
	"eta/eta_hub/global"
	"eta/eta_hub/utils"
)

func EdbInfoUpdateStatusByEdbInfoId(edbInfoIds []int, isStop int, calculateEdbInfoIds []int) (err error) {
	//o, err := orm.NewOrmUsingDB("data").Begin()
	//if err != nil {
	//	return
	//}
	//defer func() {
	//	if err != nil {
	//		_ = o.Rollback()
	//		return
	//	}
	//	_ = o.Commit()
	//}()

	tx := global.DbMap[utils.DbNameIndex].Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()

	// 更改指标的更新状态
	if len(edbInfoIds) == 1 {
		sql := ` UPDATE edb_info SET no_update = ? WHERE edb_info_id=? `
		err = tx.Exec(sql, isStop, edbInfoIds[0]).Error
		if err != nil {
			return
		}
	} else {
		sql := ` UPDATE edb_info SET no_update = ? WHERE edb_info_id IN (` + utils.GetOrmInReplace(len(edbInfoIds)) + `) `
		err = tx.Exec(sql, isStop, edbInfoIds).Error
		if err != nil {
			return
		}
	}

	if len(calculateEdbInfoIds) > 0 {
		// 批量更新相关联的指标ID
		sql := ` UPDATE edb_info SET no_update = ? WHERE edb_info_id IN (` + utils.GetOrmInReplace(len(calculateEdbInfoIds)) + `) `
		err = tx.Exec(sql, isStop, calculateEdbInfoIds).Error
		if err != nil {
			return
		}
	}

	return
}