eta_api/models/data_manage/edb_refresh/edb_refresh_config.go

package edb_refresh

import (
	"errors"
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"time"
)

// EdbRefreshConfig
// @Description: 指标的刷新时间配置表
type EdbRefreshConfig struct {
	EdbRefreshConfigId  int       `orm:"column(edb_refresh_config_id);pk" gorm:"primaryKey" `
	RefreshFrequency    string    `description:"刷新频率"`
	RefreshFrequencyDay int       `description:"具体刷新的日期"`
	RefreshTime         string    `description:"刷新时间"`
	RefreshAllData      int       `description:"是否刷新所有数据，0：否，1：刷新所有数据"`
	RefreshDataNum      int       `description:"刷新单元格数"`
	ModifyTime          time.Time `description:"最晚一次的更新时间"`
	CreateTime          time.Time `description:"添加时间"`
}

// Add
// @Description: 添加
// @author: Roc
// @receiver m
// @datetime 2023-12-14 16:11:10
// @param cols []string
// @return err error
func (m *EdbRefreshConfig) Add() (err error) {
	if m.EdbRefreshConfigId > 0 {
		err = errors.New("该配置已存在")
		return
	}
	err = global.DbMap[utils.DbNameIndex].Create(m).Error
	return
}

// Update
// @Description: 更新
// @author: Roc
// @receiver m
// @datetime 2023-12-14 16:11:10
// @param cols []string
// @return err error
func (m *EdbRefreshConfig) Update(cols []string) (err error) {
	err = global.DbMap[utils.DbNameIndex].Select(cols).Updates(m).Error
	return
}

// Delete
// @Description: 删除
// @author: Roc
// @receiver m
// @datetime 2023-12-14 16:11:10
// @return err error
func (m *EdbRefreshConfig) Delete() (err error) {
	sql := ` DELETE FROM edb_refresh_config WHERE edb_refresh_config_id=?`
	err = global.DbMap[utils.DbNameIndex].Exec(sql, m.EdbRefreshConfigId).Error
	return
}

// GetEdbRefreshConfigListByCondition
// @Description: 根据条件获取刷新配置列表
// @author: Roc
// @datetime 2024-01-08 14:07:10
// @param refreshFrequency string
// @param refreshTime string
// @param refreshFrequencyDay int
// @param refreshAllData int
// @param refreshDataNum int
// @return item *EdbRefreshConfig
// @return err error
func GetEdbRefreshConfigListByCondition(refreshFrequency, refreshTime string, refreshFrequencyDay, refreshAllData, refreshDataNum int) (item *EdbRefreshConfig, err error) {
	sql := `SELECT * FROM edb_refresh_config
	     WHERE refresh_frequency = ? AND refresh_frequency_day = ? AND refresh_time = ? AND refresh_all_data = ? AND refresh_data_num = ? ORDER BY edb_refresh_config_id ASC `

	err = global.DbMap[utils.DbNameIndex].Raw(sql, refreshFrequency, refreshFrequencyDay, refreshTime, refreshAllData, refreshDataNum).First(&item).Error
	return
}

// SaveEdbRefreshConfig
// @Description:  保存刷新配置
// @author: Roc
// @datetime 2024-01-08 15:13:23
// @param source int
// @param subSource int
// @param sysUserId int
// @param sysUserRealName string
// @param newConfigList []*EdbRefreshConfig
// @param addMapping []*EdbRefreshMapping
// @param edbIdList []int
// @return err error
func SaveEdbRefreshConfig(source, subSource, sysUserId int, sysUserRealName string, newConfigList []*EdbRefreshConfig, addMapping []*EdbRefreshMapping, edbIdList []int) (err error) {
	num := len(edbIdList)
	if num <= 0 {
		return
	}
	to := global.DbMap[utils.DbNameIndex].Begin()

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

	for _, v := range newConfigList {
		tmpErr := to.Create(v).Error
		if tmpErr != nil {
			err = tmpErr
			return
		}

		for _, edbId := range edbIdList {
			addMapping = append(addMapping, &EdbRefreshMapping{
				EdbRefreshMappingId: 0,
				Source:              source,
				SubSource:           subSource,
				EdbInfoId:           edbId,
				EdbRefreshConfigId:  v.EdbRefreshConfigId,
				SysUserId:           sysUserId,
				SysUserRealName:     sysUserRealName,
				ModifyTime:          time.Now(),
				CreateTime:          time.Now(),
			})
		}
	}

	// 先删除
	sql := `DELETE FROM edb_refresh_mapping WHERE source =? AND sub_source =? AND edb_info_id in (` + utils.GetOrmInReplace(num) + `)`
	err = to.Exec(sql, source, subSource, edbIdList).Error

	// 再写入
	if len(addMapping) > 0 {
		tmpAddDataList := make([]*EdbRefreshMapping, 0)
		i := 0
		for _, v := range addMapping {
			tmpAddDataList = append(tmpAddDataList, v)
			i++
			if i >= 500 {
				err = to.CreateInBatches(tmpAddDataList, utils.MultiAddNum).Error
				if err != nil {
					return
				}

				i = 0
				tmpAddDataList = make([]*EdbRefreshMapping, 0)
			}
		}

		if len(tmpAddDataList) > 0 {
			err = to.CreateInBatches(tmpAddDataList, utils.MultiAddNum).Error
			if err != nil {
				return
			}
		}
	}

	return

}

// EdbRefreshConfigItem
// @Description: 指标的刷新时间配置项
type EdbRefreshConfigItem struct {
	RefreshFrequency    string `description:"刷新频率"`
	RefreshFrequencyDay int    `description:"具体刷新的日期"`
	RefreshTime         string `description:"刷新时间"`
	RefreshAllData      int    `description:"是否刷新所有数据，0：否，1：刷新所有数据"`
	RefreshDataNum      int    `description:"刷新单元格数"`
}

// GetEdbRefreshConfigListBySourceAndeEdbInfoId
// @Description: 根据来源和指标获取其配置列表
// @author: Roc
// @datetime 2024-01-10 14:18:21
// @param source int
// @param subSource int
// @param edbInfoId int
// @return list []*EdbRefreshDefaultConfig
// @return err error
func GetEdbRefreshConfigListBySourceAndeEdbInfoId(source, subSource, edbInfoId int) (list []*EdbRefreshConfigItem, err error) {
	sql := `SELECT a.* FROM edb_refresh_config a 
	    JOIN  edb_refresh_mapping b ON a.edb_refresh_config_id = b.edb_refresh_config_id
	    WHERE b.source = ? AND b.sub_source = ? AND b.edb_info_id = ? ORDER BY a.edb_refresh_config_id ASC `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, source, subSource, edbInfoId).Find(&list).Error

	return
}

// EdbRefreshConfigAndEdbItem
// @Description: 指标的刷新时间配置项
type EdbRefreshConfigAndEdbItem struct {
	RefreshFrequency    string `description:"刷新频率"`
	RefreshFrequencyDay int    `description:"具体刷新的日期"`
	RefreshTime         string `description:"刷新时间"`
	RefreshAllData      int    `description:"是否刷新所有数据，0：否，1：刷新所有数据"`
	RefreshDataNum      int    `description:"刷新单元格数"`
	EdbInfoId           int    `description:"指标id，如果是数据源（钢联、有色金属）的，那么就是数据源里面的id"`
	Source              int    `description:"来源"`
	SubSource           int    `description:"子来源"`
}

// GetEdbRefreshConfigAndEdbListBySourceAndeEdbInfoId
// @Description: 根据来源和指标获取其配置列表
// @author: Roc
// @datetime 2024-01-10 14:18:21
// @param source int
// @param subSource int
// @param edbInfoId int
// @return list []*EdbRefreshDefaultConfig
// @return err error
func GetEdbRefreshConfigAndEdbListBySourceAndeEdbInfoId(source, subSource int, edbInfoIdList []int) (list []*EdbRefreshConfigAndEdbItem, err error) {
	num := len(edbInfoIdList)
	if num <= 0 {
		return
	}
	sql := `SELECT a.*,b.source,b.sub_source,b.edb_info_id FROM edb_refresh_config a 
	    JOIN  edb_refresh_mapping b ON a.edb_refresh_config_id = b.edb_refresh_config_id
	    WHERE b.source = ? AND b.sub_source = ? AND b.edb_info_id in (` + utils.GetOrmInReplace(num) + `)  ORDER BY a.edb_refresh_config_id ASC `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, source, subSource, edbInfoIdList).Find(&list).Error
	return
}