eta_api/models/data_manage/edb_data_wind.go

package data_manage

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

type EdbDataWind struct {
	//EdbDataId     int `orm:"column(edb_data_id);pk"`
	EdbDataId     int `gorm:"column:edb_data_id;primaryKey"`
	EdbInfoId     int
	EdbCode       string
	DataTime      string
	Value         float64
	Status        int
	CreateTime    time.Time
	ModifyTime    time.Time
	DataTimestamp int64
}

func AddEdbDataWindBySql(sqlStr string) (err error) {
	//o := orm.NewOrmUsingDB("data")
	//_, err = o.Raw(sqlStr).Exec()
	err = global.DbMap[utils.DbNameIndex].Exec(sqlStr).Error
	return
}

func DeleteEdbDataWind(edbCode string) (err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := `DELETE FROM edb_data_wind WHERE edb_code=? `
	//_, err = o.Raw(sql, edbCode).Exec()
	err = global.DbMap[utils.DbNameIndex].Exec(sql, edbCode).Error
	return
}

func ModifyEdbDataWind(edbInfoId int64, dataTime string, value float64) (err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := ` UPDATE edb_data_wind SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
	//_, err = o.Raw(sql, value, edbInfoId, dataTime).Exec()
	err = global.DbMap[utils.DbNameIndex].Exec(sql, value, edbInfoId, dataTime).Error
	return
}

func GetEdbDataWindByCode(edbCode string, size int) (items []*EdbInfoSearchData, err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := ` SELECT * FROM edb_data_wind WHERE edb_code=? ORDER BY data_time DESC LIMIT ? `
	//_, err = o.Raw(sql, edbCode, size).QueryRows(&items)
	err = global.DbMap[utils.DbNameIndex].Raw(sql, edbCode, size).Find(&items).Error
	return
}

func GetEdbDataWindMaxOrMinDate(edbCode string) (min_date, max_date string, err error) {
	//o := orm.NewOrmUsingDB("data")
	//sql := ` SELECT MIN(data_time) AS min_date,MAX(data_time) AS max_date FROM edb_data_wind WHERE edb_code=? `
	//err = o.Raw(sql, edbCode).QueryRow(&min_date, &max_date)
	sql := ` SELECT MIN(data_time) AS minDate,MAX(data_time) AS maxDate FROM edb_data_wind WHERE edb_code=? `
	var maxAndMinDate MaxAndMinDate
	err = global.DbMap[utils.DbNameIndex].Raw(sql, edbCode).First(&maxAndMinDate).Error
	if err != nil {
		return
	}
	min_date = maxAndMinDate.MinDate.Format(utils.FormatDate)
	max_date = maxAndMinDate.MaxDate.Format(utils.FormatDate)
	return
}

func GetEdbDataWindByCodeAndDate(edbCode string, startDate string) (count int, err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := ` SELECT COUNT(1) AS count FROM edb_data_wind WHERE edb_code=? AND data_time=? `
	//err = o.Raw(sql, edbCode, startDate).QueryRow(&count)
	err = global.DbMap[utils.DbNameIndex].Raw(sql, edbCode, startDate).First(&count).Error
	return
}

type EdbDataFromWind struct {
	Close  map[string]float64 `json:"CLOSE"`
	Dt     map[string]int64   `json:"DT"`
	ErrMsg string
}

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

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

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

	return
}