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 `gorm:"column:data_timestamp"` } 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 }