eta_index_lib/models/base_from_ly_data.go

// @Author gmy 2024/8/7 9:50:00
package models

import (
	"eta/eta_index_lib/global"
	"eta/eta_index_lib/utils"
	"github.com/beego/beego/v2/client/orm"
)

type BaseFromLyData struct {
	BaseFromLyDataId int `gorm:"column:base_from_ly_data_id;primaryKey"` // 数据ID
	//BaseFromLyDataId  int     `orm:"column(base_from_ly_data_id);pk"` // 数据ID
	CreateTime        string  `orm:"column(create_time)"`           // 创建时间
	ModifyTime        string  `orm:"column(modify_time)"`           // 修改时间
	BaseFromLyIndexId int     `orm:"column(base_from_ly_index_id)"` // 指标id
	IndexCode         string  `orm:"column(index_code)"`            // 指标编码
	DataTime          string  `orm:"column(data_time)"`             // 数据日期
	Value             float64 `orm:"column(value)"`                 // 数据值
}

func init() {
	orm.RegisterModel(new(BaseFromLyData))
}

// AddLyDataList 批量插入数据记录列表
func AddLyDataList(items []BaseFromLyData) (err error) {
	//o := orm.NewOrm()
	//_, err = o.InsertMulti(len(items), items)
	err = global.DEFAULT_DB.CreateInBatches(items, len(items)).Error
	return
}

// GetLyDataByIndexIdAndDataTime 根据指标id和数据日期查询数据
func GetLyDataByIndexIdAndDataTime(indexId int, dataTime string) (items []BaseFromLyData, err error) {
	//o := orm.NewOrm()
	sql := `SELECT * FROM base_from_ly_data WHERE base_from_ly_index_id=? AND data_time=?`
	//_, err = o.Raw(sql, indexId, dataTime).QueryRows(&items)
	err = global.DEFAULT_DB.Raw(sql, indexId, dataTime).Find(&items).Error
	if utils.IsErrNoRow(err) {
		return nil, nil
	}
	return
}

// GetLyDataByIndexIdAndDataTimeYM 根据指标id和数据日期的年月查询数据
func GetLyDataByIndexIdAndDataTimeYM(indexId int, dataTime string) (items []BaseFromLyData, err error) {
	//o := orm.NewOrm()
	sql := `SELECT * FROM base_from_ly_data WHERE base_from_ly_index_id=? AND data_time like ?`
	//_, err = o.Raw(sql, indexId, dataTime+"%").QueryRows(&items)
	err = global.DEFAULT_DB.Raw(sql, indexId, dataTime+"%").Find(&items).Error
	if utils.IsErrNoRow(err) {
		return nil, nil
	}
	return
}

// UpdateLyDataById 根据主键id更新数据
func UpdateLyDataById(dataId int, value float64) (err error) {
	//o := orm.NewOrm()
	sql := `UPDATE base_from_ly_data SET value=? WHERE base_from_ly_data_id=?`
	//_, err = o.Raw(sql, value, dataId).Exec()
	err = global.DEFAULT_DB.Exec(sql, value, dataId).Error
	return
}

// GetBaseFromLyDataByIndexCode 根据指标编码查询
func GetBaseFromLyDataByIndexCode(condition string, pars []interface{}) (items []BaseFromLyData, err error) {
	sql := `SELECT * FROM base_from_ly_data WHERE 1=1 `
	//o := orm.NewOrm()
	if condition != "" {
		sql += condition
	}
	//_, err = o.Raw(sql, pars...).QueryRows(&items)
	err = global.DEFAULT_DB.Raw(sql, pars...).Find(&items).Error
	if utils.IsErrNoRow(err) {
		return nil, nil
	}
	return
}