eta_api/models/data_manage/base_from_ly_data.go

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

import (
	"errors"
	"github.com/beego/beego/v2/client/orm"
	"github.com/rdlucklib/rdluck_tools/paging"
)

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

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

type BaseFromLyDataPage struct {
	List   []*BaseFromLyData  `description:"指标数据列表"`
	Paging *paging.PagingItem `description:"分页数据"`
}

// GetLyDataCountByIndexId 获取指标数据总数
func GetLyDataCountByIndexId(indexId int) (count int, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := `SELECT count(*) FROM base_from_ly_data WHERE base_from_ly_index_id=?`
	err = o.Raw(sql, indexId).QueryRow(&count)
	return
}

// GetLyDataPageByIndexId 获取指标数据分页列表
func GetLyDataPageByIndexId(indexId int, startSize, pageSize int) (items []*BaseFromLyData, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := `SELECT * FROM base_from_ly_data WHERE base_from_ly_index_id=? ORDER BY data_time desc LIMIT ?,?`
	_, err = o.Raw(sql, indexId, startSize, pageSize).QueryRows(&items)
	return
}

// GetBaseFromLyDataByIndexCode 根据指标编码查询
func GetBaseFromLyDataByIndexCode(indexCode string) (items []*BaseFromLyData, err error) {
	sql := `SELECT * FROM base_from_ly_data WHERE index_code=?`
	o := orm.NewOrmUsingDB("data")
	_, err = o.Raw(sql, indexCode).QueryRows(&items)
	return
}

// GetLyDataLastByIndexCode 根据指标编码查询 返回ModifyTime最后一条数据
func GetLyDataLastByIndexCode(indexCodes []string) (item []*BaseFromLyData, err error) {
	o := orm.NewOrmUsingDB("data")
	qs := o.QueryTable("base_from_ly_data")

	// 使用 Filter 来构造 IN 查询
	qs = qs.Filter("index_code__in", indexCodes)

	// 使用 OrderBy 和 Limit 来获取最新的一条数据
	err = qs.OrderBy("-modify_time").Limit(1).One(&item)
	if err != nil {
		if errors.Is(err, orm.ErrNoRows) {
			return nil, nil // 返回 nil 表示没有找到数据
		}
		return nil, err
	}
	return
}