eta_task/models/data_manage/base_from_mysteel_chemical_index.go

package data_manage

import (
	"eta/eta_task/global"
	"eta/eta_task/utils"
	"gorm.io/gorm"
)

type BaseFromMysteelChemicalIndexItem struct {
	BaseFromMysteelChemicalIndexId    int32   `gorm:"column:base_from_mysteel_chemical_index_id;primaryKey;autoIncrement" json:"base_from_mysteel_chemical_index_id"`
	BaseFromMysteelChemicalClassifyId int32   `json:"base_from_mysteel_chemical_classify_id"`  // 钢联化工指标分类id
	IndexCode                         string  `json:"index_code"`                              // 指标编码
	IndexName                         string  `json:"index_name"`                              // 指标名称
	Unit                              string  `json:"unit"`                                    // 单位
	Source                            string  `json:"source"`                                  // 数据来源
	Frequency                         string  `json:"frequency"`                               // 频度
	StartDate                         string  `json:"start_date"`                              // 开始日期
	EndDate                           string  `json:"end_date"`                                // 结束日期
	Describe                          string  `json:"describe"`                                // 指标描述
	UpdateWeek                        string  `json:"update_week"`                             // 更新周期
	UpdateTime                        string  `json:"update_time"`                             // 更新时间，多个时间点用英文,隔开
	UpdateTime2                       string  `gorm:"column:update_time2" json:"update_time2"` // 更新时间2
	SysUserId                         int32   `json:"sys_user_id"`                             // 创建人id
	SysUserRealName                   string  `json:"sys_user_real_name"`                      // 创建人姓名
	CreateTime                        string  `json:"create_time"`                             // 创建时间
	ModifyTime                        string  `json:"modify_time"`                             // 修改时间
	FilePath                          string  `json:"file_path"`                               // 文件存储路径
	Sort                              int32   `json:"sort"`                                    // 排序
	MergeFilePath                     string  `json:"merge_file_path"`
	FileIndex                         int32   `json:"file_index"`
	MergeUpdateWeek                   string  `json:"merge_update_week"`    // 合并文件的更新周
	MergeFilePathWeek                 string  `json:"merge_file_path_week"` // 更新文件
	UpdateDate                        string  `json:"update_date"`          // 更新日期
	IsRefresh                         int32   `json:"is_refresh"`           // 0:不需要及时刷新,1:需要及时刷新
	IsStop                            int32   `json:"is_stop"`              // 是否停更:1:停更,0:未停更
	TerminalCode                      string  `json:"terminal_code"`        // 所属终端编码
	EndValue                          float64 `json:"end_value"`            // 指标的最新值
	EdbInfoId                         int     `json:"edb_info_id"`
}

func (m *BaseFromMysteelChemicalIndexItem) AfterFind(db *gorm.DB) (err error) {
	m.StartDate = utils.GormDateStrToDateStr(m.StartDate)
	m.EndDate = utils.GormDateStrToDateStr(m.EndDate)
	m.CreateTime = utils.GormDateStrToDateTimeStr(m.CreateTime)
	m.ModifyTime = utils.GormDateStrToDateTimeStr(m.ModifyTime)
	return
}

func (m *BaseFromMysteelChemicalIndexItem) ConvertTimeStr() {
	m.StartDate = utils.GormDateStrToDateStr(m.StartDate)
	m.EndDate = utils.GormDateStrToDateStr(m.EndDate)
	m.CreateTime = utils.GormDateStrToDateTimeStr(m.CreateTime)
	m.ModifyTime = utils.GormDateStrToDateTimeStr(m.ModifyTime)
	return
}

// GetBaseFromMysteelChemicalIndexItemByCode
// @Description: 根据指标编码获取钢联指标详情
// @author: Roc
// @datetime 2024-03-11 16:42:56
// @param edbCode string
// @return item *BaseFromMysteelChemicalIndexItem
// @return err error
func GetBaseFromMysteelChemicalIndexItemByCode(edbCode string) (item *BaseFromMysteelChemicalIndexItem, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE index_code = ?`
	err = o.Raw(sql, edbCode).First(&item).Error
	return
}

func GetBaseFromMysteelChemicalIndexItems(frequencyList []string) (items []*BaseFromMysteelChemicalIndexItem, err error) {
	num := len(frequencyList)
	if num == 0 {
		return
	}
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT b.*, e.edb_info_id FROM base_from_mysteel_chemical_index AS b
	LEFT JOIN edb_info AS e
	ON b.index_code = e.edb_code
	WHERE b.source='api' AND b.frequency IN (` + utils.GetOrmInReplace(len(frequencyList)) + `)`
	err = o.Raw(sql, frequencyList).Find(&items).Error
	return
}

// GetRefreshBaseFromMysteelChemicalIndexItemByCreateTime 获取正常刷新的钢联化工指标
func GetRefreshBaseFromMysteelChemicalIndexItemByCreateTime(endDate string, startPage, pageSize int) (items []*BaseFromMysteelChemicalIndexItem, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE is_stop = 0 and create_time < ? Limit ?,?`
	err = o.Raw(sql, endDate, startPage, pageSize).Find(&items).Error
	return
}

// GetCountRefreshBaseFromMysteelChemicalIndexItemByCreateTime 获取正常刷新的钢联化工指标
func GetCountRefreshBaseFromMysteelChemicalIndexItemByCreateTime(endDate string) (total int64, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT count(*) FROM base_from_mysteel_chemical_index WHERE is_stop = 0 and create_time < ?`
	err = o.Raw(sql, endDate).Scan(&total).Error
	return
}

func SetStopRefreshMysteelChemicalIndex(ids []int32) (err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` UPDATE base_from_mysteel_chemical_index SET is_stop = 1 WHERE base_from_mysteel_chemical_index_id IN (` + utils.GetOrmInReplace(len(ids)) + `) and is_stop=0`
	err = o.Exec(sql, ids).Error
	return
}