eta_api/models/data_manage/mysteel_chemical_index.go

package data_manage

import (
	"eta_gn/eta_api/global"
	"eta_gn/eta_api/utils"
	"fmt"
	"strings"
	"time"

	"github.com/rdlucklib/rdluck_tools/paging"
)

// BaseFromMysteelChemicalIndex 钢联化工指标表
type BaseFromMysteelChemicalIndex struct {
	BaseFromMysteelChemicalIndexId    int       `orm:"column(base_from_mysteel_chemical_index_id);pk" gorm:"primaryKey" `
	BaseFromMysteelChemicalClassifyId int       `orm:"column(base_from_mysteel_chemical_classify_id)" description:"钢联化工指标分类id"`
	IndexCode                         string    `description:"指标编码"`
	IndexName                         string    `description:"指标名称"`
	Unit                              string    `description:"单位"`
	Source                            string    `description:"数据来源"`
	Frequency                         string    `description:"频度"`
	StartDate                         string    `description:"开始日期"`
	EndDate                           string    `description:"结束日期"`
	Describe                          string    `description:"指标描述"`
	UpdateWeek                        string    `description:"更新周期"`
	UpdateTime                        string    `description:"更新时间1"`
	UpdateTime2                       string    `description:"更新时间2"`
	SysUserId                         int       `description:"创建人id"`
	SysUserRealName                   string    `description:"创建人姓名"`
	ModifyTime                        time.Time `description:"修改时间"`
	CreateTime                        time.Time `description:"创建时间"`
	Sort                              int       `description:"排序字段"`
	MergeFilePath                     string    `description:"合并文件"`
	TerminalCode                      string    `description:"终端编码"`
	IsStop                            int       `description:"是否停更:1:停更,0:未停更"`
	EndValue                          float64   `description:"指标的最新值"`
	IsSupplierStop                    int       `description:"是否供应商停更:1:停更,0:未停更"`
}

type BaseFromMysteelChemicalIndexItem struct {
	BaseFromMysteelChemicalIndexId int64  `orm:"column(base_from_mysteel_chemical_index_id);pk" gorm:"primaryKey" `
	IndexCode                      string `description:"指标编码"`
	IndexName                      string `description:"指标名称"`
	Unit                           string `description:"单位"`
	Frequency                      string `description:"频度"`
	Source                         int    `description:"数据来源"`
	StartDate                      string `description:"开始日期"`
	EndDate                        string `description:"结束日期"`
	CreateTime                     string `description:"创建时间"`
	ModifyTime                     string `description:"修改时间"`
	EdbInfoId                      int    `description:"eta指标库的id"`
	EdbUniqueCode                  string `description:"指标库唯一编码"`
	EdbClassifyId                  int    `description:"指标库分类ID"`
	IsStop                         int    `description:"是否停更:1:停更,0:未停更"`
	EdbExist                       int    `description:"指标库是否已添加：0-否；1-是"`
}

var BaseFromMysteelChemicalIndexCols = struct {
	BaseFromMysteelChemicalIndexId string
	IndexCode                      string
	IndexName                      string
	Unit                           string
	Frequency                      string
	Source                         string
	StartDate                      string
	EndDate                        string
	CreateTime                     string
	ModifyTime                     string
}{
	BaseFromMysteelChemicalIndexId: "base_from_mysteel_chemical_index_id",
	IndexCode:                      "index_code",
	IndexName:                      "index_name",
	Unit:                           "unit",
	Frequency:                      "frequency",
	Source:                         "source",
	StartDate:                      "start_date",
	EndDate:                        "end_date",
	CreateTime:                     "create_time",
	ModifyTime:                     "modify_time",
}

// Update 更新钢联化工指标基础信息
func (item *BaseFromMysteelChemicalIndex) Update(cols []string) (err error) {
	err = global.DmSQL["data"].Select(cols).Updates(item).Error
	return
}
func (m *BaseFromMysteelChemicalIndex) GeItemsByCondition(condition string, pars []interface{}, limitSize int) (items []*BaseFromMysteelChemicalIndexItem, err error) {
	if pars == nil {
		pars = make([]interface{}, 0)
	}

	sql := `select a.*,b.edb_info_id,b.unique_code as edb_unique_code,b.classify_id as edb_classify_id FROM base_from_mysteel_chemical_index AS a LEFT JOIN edb_info b on a.index_code=b.edb_code AND b.source=? WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	sql += ` order by a.base_from_mysteel_chemical_index_id desc `

	if limitSize > 0 {
		sql += " limit ? "
		pars = append(pars, limitSize)
	}
	err = global.DmSQL["data"].Raw(sql, utils.ForwardPars(pars, utils.DATA_SOURCE_MYSTEEL_CHEMICAL)...).Find(&items).Error
	return
}

// AddBaseFromMysteelChemicalIndex 添加钢联化工指标
func AddBaseFromMysteelChemicalIndex(item *BaseFromMysteelChemicalIndex) (lastId int64, err error) {
	err = global.DmSQL["data"].Create(item).Error
	return
}

// AddBaseFromMysteelChemicalIndex 添加钢联化工指标
func BatchAddBaseFromMysteelChemicalIndex(items []*BaseFromMysteelChemicalIndex) (lastId int64, err error) {
	err = global.DmSQL["data"].CreateInBatches(items, utils.MultiAddNum).Error
	return
}

// BaseFromMysteelChemicalData 钢联化工指标数据表
type BaseFromMysteelChemicalData struct {
	BaseFromMysteelChemicalDataId  int       `orm:"column(base_from_mysteel_chemical_data_id);pk" gorm:"primaryKey" `
	BaseFromMysteelChemicalIndexId int       `orm:"column(base_from_mysteel_chemical_index_id)" description:"钢联化工指标id"`
	IndexCode                      string    `description:"指标编码"`
	DataTime                       time.Time `description:"数据日期"`
	Value                          float64   `description:"数据值"`
	ModifyTime                     time.Time `description:"修改时间"`
	CreateTime                     time.Time `description:"创建时间"`
}

// MysteelChemicalFrequency 钢联化工频度
type MysteelChemicalFrequency struct {
	Frequency string `description:"频度:1-日度 2-周度 3-月度 4-季度 5-年度 99-无固定频率"`
}

// GetMysteelChemicalIndexByClassifyId 用于分类展示
func GetMysteelChemicalIndexByClassifyId(classifyId int) (items []*BaseFromMysteelChemicalClassifyItems, err error) {
	sql := ` SELECT base_from_mysteel_chemical_index_id,base_from_mysteel_chemical_classify_id,index_name AS classify_name,
	        sys_user_id,sys_user_real_name,sort,index_code
	       FROM base_from_mysteel_chemical_index where base_from_mysteel_chemical_classify_id=? ORDER BY sort asc, create_time ASC `
	err = global.DmSQL["data"].Raw(sql, classifyId).Find(&items).Error
	return
}

// GetMysteelChemicalFrequency 获取钢联化工频度数据列表
func GetMysteelChemicalFrequency(condition string, pars []interface{}) (items []*MysteelChemicalFrequency, err error) {
	sql := ` SELECT frequency FROM base_from_mysteel_chemical_index WHERE 1=1 AND frequency != "" `
	if condition != "" {
		sql += condition
	}
	sql += ` GROUP BY frequency ORDER BY frequency ASC `
	err = global.DmSQL["data"].Raw(sql, pars...).Find(&items).Error
	return
}

// MysteelChemicalList 钢联化工指标列表
type MysteelChemicalList struct {
	Id                                int                    `orm:"column(base_from_mysteel_chemical_index_id)"`
	BaseFromMysteelChemicalClassifyId int                    `orm:"column(base_from_mysteel_chemical_classify_id)" description:"钢联化工指标分类id"`
	ParentClassifyId                  int                    `description:"钢联化工指标父级分类id"`
	IndexCode                         string                 `description:"指标编码"`
	IndexName                         string                 `description:"指标名称"`
	UnitName                          string                 `orm:"column(unit)"`
	UniqueCode                        string                 `description:"唯一编码"`
	FrequencyName                     string                 `orm:"column(frequency)"`
	EdbInfoId                         int                    `description:"指标库的id"`
	UpdateTime                        string                 `orm:"column(modify_time)"`
	IsStop                            int                    `description:"是否停更:1:停更,0:未停更"`
	IsSupplierStop                    int                    `description:"是否供应商停更:1:停更,0:未停更"`
	Paging                            *paging.PagingItem     `description:"分页数据"`
	DataList                          []*MysteelChemicalData `gorm:"-"`
}

// MysteelChemicalData 钢联化工数据列表
type MysteelChemicalData struct {
	InputValue string `orm:"column(value)" description:"值"`
	DataTime   string `orm:"column(data_time)" description:"日期"`
}

// GetMysteelChemicalIndex 根据分类id获取钢联化工频度数据列表
func GetMysteelChemicalIndex(condition string, pars []interface{}) (items []*MysteelChemicalList, err error) {
	sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE 1=1 `

	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY base_from_mysteel_chemical_index_id ASC `
	err = global.DmSQL["data"].Raw(sql, pars...).Find(&items).Error
	return
}

// GetMysteelChemicalIndexData 根据指标code获取钢联化工数据列表
func GetMysteelChemicalIndexData(indexCode string, startSize, pageSize int) (items []*MysteelChemicalData, err error) {
	sql := ` SELECT * FROM (
	SELECT DISTINCT a.index_code,a.value,a.data_time FROM base_from_mysteel_chemical_data AS a WHERE index_code=? 
	ORDER BY data_time DESC
	)AS t
	GROUP BY t.data_time
	ORDER BY t.data_time DESC LIMIT ?,? `
	err = global.DmSQL["data"].Raw(sql, indexCode, startSize, pageSize).Find(&items).Error
	return
}

// GetMysteelChemicalIndexDataCount 根据指标code获取钢联化工数据列表 获取钢联数据总数
func GetMysteelChemicalIndexDataCount(indexCode string) (count int, err error) {
	sql := `SELECT COUNT(1) AS count FROM (
			SELECT * FROM (
	SELECT DISTINCT a.index_code,a.value,a.data_time FROM base_from_mysteel_chemical_data AS a WHERE index_code=? 
	ORDER BY data_time DESC
	)AS t
	GROUP BY t.data_time
	ORDER BY t.data_time DESC
			)AS n `
	err = global.DmSQL["data"].Raw(sql, indexCode).Scan(&count).Error
	return
}

// GetBaseFromMysteelChemicalIndexByIndexId 根据指标id获取指标信息
func GetBaseFromMysteelChemicalIndexByIndexId(indexId int) (item *BaseFromMysteelChemicalIndex, err error) {
	sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_index_id=? `
	err = global.DmSQL["data"].Raw(sql, indexId).First(&item).Error
	return
}

// GetBaseFromMysteelChemicalIndexByCode 根据指标code获取指标信息
func GetBaseFromMysteelChemicalIndexByCode(indexCode string) (item *BaseFromMysteelChemicalIndex, err error) {
	sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE index_code=? `
	err = global.DmSQL["data"].Raw(sql, indexCode).First(&item).Error
	return
}

// GetBaseFromMysteelChemicalIndexByCodeList 根据指标code获取指标信息
func GetBaseFromMysteelChemicalIndexByCodeList(indexCodeList []string) (items []*BaseFromMysteelChemicalIndex, err error) {
	if len(indexCodeList) <= 0 {
		return
	}
	sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE index_code IN (%s) `
	holder := make([]string, 0)
	for range indexCodeList {
		holder = append(holder, "?")
	}
	sql = fmt.Sprintf(sql, strings.Join(holder, ","))
	err = global.DmSQL["data"].Raw(sql, indexCodeList).Find(&items).Error
	return
}

// GetBaseFromMysteelChemicalIndexListByClassifyIdList 根据指标分类id集合获取指标列表信息
func GetBaseFromMysteelChemicalIndexListByClassifyIdList(classifyIdList []int) (items []*BaseFromMysteelChemicalIndex, err error) {
	num := len(classifyIdList)
	if num <= 0 {
		return
	}
	sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_classify_id IN (` + utils.GetOrmInReplace(num) + `) `
	err = global.DmSQL["data"].Raw(sql, classifyIdList).Find(&items).Error
	return
}

// GetBaseFromMysteelChemicalDataMaxCount 获取分类下指标的最大数量
func GetBaseFromMysteelChemicalDataMaxCount(classifyId int) (count int, err error) {
	sql := `SELECT MAX(t.num) AS count FROM (
				SELECT COUNT(1) AS num  FROM base_from_mysteel_chemical_index AS a
				INNER JOIN base_from_mysteel_chemical_data AS b ON a.base_from_mysteel_chemical_index_id=b.base_from_mysteel_chemical_index_id
				WHERE a.base_from_mysteel_chemical_classify_id=?
				GROUP BY a.base_from_mysteel_chemical_index_id
			)AS t `
	err = global.DmSQL["data"].Raw(sql, classifyId).Scan(&count).Error
	return
}

// GetMysteelChemicalIndexDataByCode 通过钢联化工指标code获取所有数据列表
func GetMysteelChemicalIndexDataByCode(indexCode string) (items []*MysteelChemicalData, err error) {
	sql := ` SELECT * FROM (
	SELECT DISTINCT a.index_code,a.value,a.data_time FROM base_from_mysteel_chemical_data AS a WHERE index_code=? 
	ORDER BY data_time DESC
	)AS t
	GROUP BY t.data_time
	ORDER BY t.data_time DESC `
	err = global.DmSQL["data"].Raw(sql, indexCode).Find(&items).Error
	return
}

// MoveBaseFromMysteelChemicalIndex 移动钢联化工指标分类
func MoveBaseFromMysteelChemicalIndex(chartInfoId, classifyId int) (err error) {
	sql := ` UPDATE base_from_mysteel_chemical_index
			SET
			  base_from_mysteel_chemical_classify_id = ?
			WHERE base_from_mysteel_chemical_index_id = ?`
	err = global.DmSQL["data"].Exec(sql, classifyId, chartInfoId).Error
	return
}

// UpdateBaseFromMysteelChemicalIndexByClassifyId 根据指标id更新排序
func UpdateBaseFromMysteelChemicalIndexByClassifyId(classifyId, nowSort, prevIndexInfoId int, updateSort string) (err error) {
	sql := ` update base_from_mysteel_chemical_index set sort = ` + updateSort + ` WHERE base_from_mysteel_chemical_classify_id=? AND  `
	if prevIndexInfoId > 0 {
		sql += ` ( sort > ? or (base_from_mysteel_chemical_index_id > ` + fmt.Sprint(prevIndexInfoId) + ` and sort = ` + fmt.Sprint(nowSort) + `))`
	}
	err = global.DmSQL["data"].Exec(sql, classifyId, nowSort).Error
	return
}

// GetFirstBaseFromMysteelChemicalIndexByClassifyId 获取当前分类下，且排序数相同 的排序第一条的数据
func GetFirstBaseFromMysteelChemicalIndexByClassifyId(classifyId int) (item *BaseFromMysteelChemicalIndex, err error) {
	sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_classify_id=? order by sort asc,base_from_mysteel_chemical_index_id asc limit 1`
	err = global.DmSQL["data"].Raw(sql, classifyId).First(&item).Error
	return
}

// GetMysteelChemicalIndexCount 根据条件获取钢联化工数据
func GetMysteelChemicalIndexCount(condition string, pars []interface{}) (count int, err error) {
	sql := ` SELECT COUNT(1) AS count FROM base_from_mysteel_chemical_index WHERE 1=1 `

	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY base_from_mysteel_chemical_index_id ASC `
	err = global.DmSQL["data"].Raw(sql, pars...).Scan(&count).Error
	return
}

// GetMysteelChemicalIndexList 根据分类id获取钢联化工频度数据列表
func GetMysteelChemicalIndexList(condition string, pars []interface{}, startSize, pageSize int, orderDesc string) (items []*BaseFromMysteelChemicalIndex, err error) {
	sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE 1=1 `

	if condition != "" {
		sql += condition
	}

	if orderDesc == `` {
		orderDesc = ` ASC `
	}
	sql += ` ORDER BY base_from_mysteel_chemical_index_id `
	sql += orderDesc
	sql += ` LIMIT ?,? `

	pars = append(pars, startSize)
	pars = append(pars, pageSize)
	err = global.DmSQL["data"].Raw(sql, pars...).Find(&items).Error
	return
}

// GetMysteelChemicalIndexListByIndexId
// @Description: 根据指标id列表获取列表信息
// @param edbIdList
// @return items
// @return err
func GetMysteelChemicalIndexListByIndexId(edbIdList []string) (items []*BaseFromMysteelChemicalIndex, err error) {
	num := len(edbIdList)
	if num <= 0 {
		return
	}
	sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_index_id in (` + utils.GetOrmInReplace(num) + `) `
	err = global.DmSQL["data"].Raw(sql, edbIdList).Find(&items).Error
	return
}

// GetMysteelChemicalIndexListByUserId
// @Description: 根据用户id列表获取列表信息
// @param userIdList
// @return items
// @return err
func GetMysteelChemicalIndexListByUserId(userIdList []int) (items []*BaseFromMysteelChemicalIndex, err error) {
	num := len(userIdList)
	if num <= 0 {
		return
	}
	sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE sys_user_id in (` + utils.GetOrmInReplace(num) + `) `
	err = global.DmSQL["data"].Raw(sql, userIdList).Find(&items).Error
	return
}

// ModifyMysteelChemicalIndexUserIdByCodeList 根据指标code列表修改创建人
func ModifyMysteelChemicalIndexUserIdByCodeList(edbIdList []string, userId int, userName string) (err error) {
	num := len(edbIdList)
	if num <= 0 {
		return
	}
	sql := `UPDATE base_from_mysteel_chemical_index SET sys_user_id=?,sys_user_real_name=? WHERE base_from_mysteel_chemical_index_id in (` + utils.GetOrmInReplace(num) + `) `
	err = global.DmSQL["data"].Exec(sql, userId, userName, edbIdList).Error
	return
}

// ModifyMysteelChemicalIndexUserIdByOldUserId
// @Description: 根据旧用户id修改新用户id
// @author: Roc
// @datetime 2024-03-25 17:59:08
// @param oldUserId int
// @param userId int
// @param userName string
// @return err error
func ModifyMysteelChemicalIndexUserIdByOldUserId(oldUserIdList []int, userId int, userName string) (err error) {
	num := len(oldUserIdList)
	if num <= 0 {
		return
	}
	sql := `UPDATE base_from_mysteel_chemical_index SET sys_user_id=?,sys_user_real_name=? WHERE sys_user_id in (` + utils.GetOrmInReplace(num) + `)  `
	err = global.DmSQL["data"].Exec(sql, userId, userName, oldUserIdList).Error
	return
}

// GetMysteelChemicalIndexAdminList 获取所有指标创建人
func GetMysteelChemicalIndexAdminList() (list []int, err error) {
	sql := ` SELECT sys_user_id FROM base_from_mysteel_chemical_index GROUP BY sys_user_id `
	err = global.DmSQL["data"].Raw(sql).Find(&list).Error
	return
}

type TerminalNum struct {
	TerminalCode string `description:"terminal_code"`
	Num          int    `description:"num"`
}

// GetMysteelChemicalGroupTerminalNum 获取钢联化工指标的终端分布
func GetMysteelChemicalGroupTerminalNum() (items []*TerminalNum, err error) {
	sql := ` SELECT terminal_code,count(1) num FROM base_from_mysteel_chemical_index GROUP BY terminal_code ORDER BY num ASC `
	err = global.DmSQL["data"].Raw(sql).Find(&items).Error
	return
}

// BaseRefreshEdbInfo
// @Description: 刷新配置的基础指标信息结构体
type BaseRefreshEdbInfo struct {
	EdbInfoId       int
	ClassifyId      int    `description:"钢联化工指标分类id"`
	IndexCode       string `description:"指标编码"`
	IndexName       string `description:"指标名称"`
	EndDate         string `description:"最新日期"`
	EndValue        string `description:"最新值"`
	SysUserId       int    `description:"创建人id"`
	SysUserRealName string `description:"创建人姓名"`
	Frequency       string `description:"频度"`
	IsStop          int    `description:"是否停更:1:停更,0:未停更"`
	TerminalCode    string `description:"终端编码"`
	RefreshTime     string `description:"刷新时间"`
}

// RefreshBaseEdbInfoResp
// @Description: 刷新数据源的数据返回
type RefreshBaseEdbInfoResp struct {
	Paging *paging.PagingItem
	List   []*BaseRefreshEdbInfo
}

// GetMysteelChemicalBaseInfoList
// @Description: 获取钢联化工数据列表
// @author: Roc
// @datetime 2024-01-10 14:28:35
// @param condition string
// @param pars []interface{}
// @param orderBy string
// @param startSize int
// @param pageSize int
// @return total int
// @return items []*BaseRefreshEdbInfo
// @return err error
func GetMysteelChemicalBaseInfoList(condition string, pars []interface{}, orderBy string, startSize, pageSize int) (total int, items []*BaseRefreshEdbInfo, err error) {
	totalSql := ` SELECT count(1) FROM base_from_mysteel_chemical_index WHERE 1=1 `
	if condition != "" {
		totalSql += condition
	}
	err = global.DmSQL["data"].Raw(totalSql, pars...).Scan(&total).Error
	if err != nil {
		return
	}
	sql := ` SELECT base_from_mysteel_chemical_index_id as edb_info_id, base_from_mysteel_chemical_classify_id as classify_id,index_code,index_name,end_date,end_value,sys_user_id,sys_user_real_name,frequency,is_stop,terminal_code FROM base_from_mysteel_chemical_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}

	if orderBy != "" {
		sql += ` ORDER BY ` + orderBy
	} else {
		sql += ` ORDER BY base_from_mysteel_chemical_index_id ASC `
	}
	sql += `  LIMIT ?,? `

	pars = append(pars, startSize)
	pars = append(pars, pageSize)
	err = global.DmSQL["data"].Raw(sql, pars...).Find(&items).Error
	return
}

// ModifyMysteelChemicalUpdateStatus
// @Description:  修改钢联化工数据停更状态
// @author: Roc
// @datetime 2024-01-08 16:23:31
// @param edbIdList []int
// @param indexCodeList []string
// @param isStop int
// @return err error
func ModifyMysteelChemicalUpdateStatus(edbIdList []int, indexCodeList []string, isStop int) (err error) {
	idNum := len(edbIdList)
	if idNum <= 0 {
		return
	}

	sql := ` UPDATE base_from_mysteel_chemical_index SET is_stop = ? WHERE base_from_mysteel_chemical_index_id IN (` + utils.GetOrmInReplace(idNum) + `) `
	err = global.DmSQL["data"].Exec(sql, isStop, edbIdList).Error
	if err != nil {
		return
	}

	codeNum := len(indexCodeList)
	if codeNum <= 0 {
		// 需要通过指标id列表查找code列表
		sql := ` SELECT index_code FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_index_id IN (` + utils.GetOrmInReplace(idNum) + `) `
		err = global.DmSQL["data"].Raw(sql, edbIdList).Find(&indexCodeList).Error
		if err != nil {
			return
		}

	}

	codeNum = len(indexCodeList)
	// 查出来的编码是空的话，那么就直接返回了
	if codeNum <= 0 {
		return
	}

	// 更改指标的更新状态
	sql = ` UPDATE edb_info SET no_update = ? WHERE source = ? AND sub_source= ? AND edb_code IN (` + utils.GetOrmInReplace(codeNum) + `) `
	err = global.DmSQL["data"].Exec(sql, isStop, utils.DATA_SOURCE_MYSTEEL_CHEMICAL, 0, indexCodeList).Error
	if err != nil {
		return
	}
	return
}

// ModifyMysteelChemicalUpdateStatusByEdbInfoId
// @Description:  修改单个钢联化工指标停更状态，同时停更依赖于该指标的计算指标
// @author: Roc
// @datetime 2024-01-08 16:23:31
// @param edbIdList []int
// @param indexCodeList []string
// @param isStop int
// @return err error
func ModifyMysteelChemicalUpdateStatusByEdbInfoId(edbInfoId, isStop int, edbCode string, calculateEdbInfoIds []int) (err error) {
	to := global.DmSQL["data"].Begin()

	defer func() {
		if err != nil {
			_ = to.Rollback()
		} else {
			_ = to.Commit()
		}
	}()

	// 更改数据源的更新状态
	sql := ` UPDATE base_from_mysteel_chemical_index SET is_stop = ? WHERE index_code = ? `
	err = to.Exec(sql, isStop, edbCode).Error
	if err != nil {
		return
	}

	// 更改指标的更新状态
	sql = ` UPDATE edb_info SET no_update = ? WHERE source = ? AND sub_source= ? AND edb_info_id=? `
	err = to.Exec(sql, isStop, utils.DATA_SOURCE_MYSTEEL_CHEMICAL, 0, edbInfoId).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 = to.Exec(sql, isStop, calculateEdbInfoIds).Error
		if err != nil {
			return
		}
	}

	return
}

func ModifyMysteelChemicalUpdateStatusByEdbInfoIds(edbInfoIds []int, isStop int, edbCodes []string, calculateEdbInfoIds []int) (err error) {
	to := global.DmSQL["data"].Begin()

	defer func() {
		if err != nil {
			_ = to.Rollback()
		} else {
			_ = to.Commit()
		}
	}()

	// 更改数据源的更新状态
	sql := ` UPDATE base_from_mysteel_chemical_index SET is_stop = ? WHERE index_code IN (` + utils.GetOrmInReplace(len(edbCodes)) + `) `
	err = to.Exec(sql, isStop, edbCodes).Error
	if err != nil {
		return
	}

	// 更改指标的更新状态
	sql = ` UPDATE edb_info SET no_update = ? WHERE source = ? AND edb_info_id IN (` + utils.GetOrmInReplace(len(edbInfoIds)) + `) `
	err = to.Exec(sql, isStop, utils.DATA_SOURCE_MYSTEEL_CHEMICAL, 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 = to.Exec(sql, isStop, calculateEdbInfoIds).Error
		if err != nil {
			return
		}
	}

	return
}

// GetNotIsSupplierStopIndexByCodeList
// @Description: 获取未被供应商停更的指标
// @author: Roc
// @datetime 2024-08-28 18:15:03
// @param codeList []string
// @param isStop int
// @return items []*BaseFromMysteelChemicalIndex
// @return err error
func GetNotIsSupplierStopIndexByCodeList(codeList []string, isStop int) (items []*BaseFromMysteelChemicalIndex, err error) {
	num := len(codeList)
	if num <= 0 {
		return
	}

	sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE is_supplier_stop = ? AND index_code in (` + utils.GetOrmInReplace(num) + `) `
	err = global.DmSQL["data"].Raw(sql, isStop, codeList).Find(&items).Error
	return
}