eta_api/models/data_manage/base_from_usda_fas.go

package data_manage

import (
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"gorm.io/gorm"
	"time"

	"github.com/rdlucklib/rdluck_tools/paging"
)

type BaseFromUsdaFasIndex struct {
	BaseFromUsdaFasIndexId int `orm:"column(base_from_usda_fas_index_id);pk" gorm:"primaryKey"`
	ClassifyId             int
	IndexCode              string
	IndexName              string
	Frequency              string
	Unit                   string
	Sort                   int
	StartDate              string `description:"开始日期"`
	EndDate                string `description:"结束日期"`
	EndValue               float64
	CreateTime             time.Time
	ModifyTime             time.Time
}

type BaseFromUsdaFasIndexList struct {
	BaseFromUsdaFasIndexId int `orm:"column(base_from_usda_fas_index_id);pk" gorm:"primaryKey"`
	ClassifyId             int
	Interface              string
	EdbInfoId              int
	EdbUniqueCode          string `description:"指标库唯一编码"`
	EdbClassifyId          int    `description:"指标库分类ID"`
	StartDate              string
	EndDate                string
	EndValue               float64
	IndexCode              string
	IndexName              string
	Frequency              string
	Unit                   string
	Sort                   int
	CreateTime             string
	ModifyTime             string
	EdbExist               int                    `description:"指标库是否已添加：0-否；1-是"`
	DataList               []*BaseFromUsdaFasData `gorm:"-"`
	Paging                 *paging.PagingItem     `description:"分页数据" gorm:"-"`
}

func (obj *BaseFromUsdaFasIndexList) AfterFind(db *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		if obj.CreateTime != "" {
			obj.CreateTime = utils.GormDateStrToDateTimeStr(obj.CreateTime)
		}
		if obj.ModifyTime != "" {
			obj.ModifyTime = utils.GormDateStrToDateTimeStr(obj.ModifyTime)
		}
		if obj.StartDate != "" {
			obj.StartDate = utils.GormDateStrToDateStr(obj.StartDate)
		}

		if obj.EndDate != "" {
			obj.EndDate = utils.GormDateStrToDateStr(obj.EndDate)
		}
	}
	return
}

type BaseFromUsdaFasIndexSearchList struct {
	List   []*BaseFromUsdaFasIndexList
	Paging *paging.PagingItem `description:"分页数据"`
}
type UsdaFasSingleDataResp struct {
	BaseFromUsdaFasIndexId int
	ClassifyId             int
	EdbInfoId              int
	IndexCode              string
	IndexName              string
	Frequency              string
	Unit                   string
	StartTime              string
	CreateTime             string
	ModifyTime             string
	EdbExist               int `description:"指标库是否已添加：0-否；1-是"`
	Data                   []*UsdaFasSingleData
}

type UsdaFasSingleData struct {
	Value    string `orm:"column(value)" description:"日期"`
	DataTime string `orm:"column(data_time)" description:"值"`
}

func GetUsdaFasIndexByClassifyId(classifyId int) (items []*BaseFromUsdaFasIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT base_from_usda_fas_index_id, classify_id, index_code, index_name FROM base_from_usda_fas_index WHERE classify_id=? ORDER BY sort ASC, base_from_usda_fas_index_id ASC `
	err = o.Raw(sql, classifyId).Find(&items).Error
	return
}

func GetUsdaFasIndex(condition string, pars []interface{}) (items []*BaseFromUsdaFasIndexList, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_usda_fas_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY sort ASC, base_from_usda_fas_index_id asc`
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

func GetUsdaFasIndexPage(condition string, pars []interface{}, startSize, pageSize int) (items []*BaseFromUsdaFasIndexList, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_usda_fas_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY sort ASC, base_from_usda_fas_index_id asc LIMIT ?,?`
	pars = append(pars, startSize, pageSize)
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

func GetUsdaFasIndexPageCount(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count  FROM base_from_usda_fas_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

func GetUsdaFasIndexDataCount(indexCode string) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count  FROM base_from_usda_fas_data WHERE index_code=? `
	err = o.Raw(sql, indexCode).Scan(&count).Error
	return
}

func GetUsdaFasIndexDataByDataTime(indexCodes []string, startDate, endDate string) (items []*BaseFromUsdaFasData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT *  FROM base_from_usda_fas_data WHERE  index_code in (` + utils.GetOrmInReplace(len(indexCodes)) + `) and data_time >=? and data_time <? ORDER BY data_time DESC `
	err = o.Raw(sql, indexCodes, startDate, endDate).Find(&items).Error
	return
}

func GetUsdaFasIndexDataTimePageByCodes(indexCodes []string, startSize, pageSize int) (dataTimes []string, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT data_time FROM base_from_usda_fas_data WHERE index_code in (` + utils.GetOrmInReplace(len(indexCodes)) + `) GROUP BY data_time ORDER BY data_time DESC LIMIT ?,? `
	err = o.Raw(sql, indexCodes, startSize, pageSize).Scan(&dataTimes).Error
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		for i := range dataTimes {
			dataTimes[i] = utils.GormDateStrToDateStr(dataTimes[i])
		}
	}
	return
}

func GetUsdaFasIndexDataTimePageCount(indexCodes []string) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(DISTINCT data_time) AS count  FROM base_from_usda_fas_data WHERE index_code in (` + utils.GetOrmInReplace(len(indexCodes)) + `) `
	err = o.Raw(sql, indexCodes).Scan(&count).Error
	return
}

func GetUsdaFasIndexDataByCodes(indexCode []string) (items []*BaseFromUsdaFasData, err error) {
	if len(indexCode) == 0 {
		return
	}
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT *  FROM base_from_usda_fas_data WHERE index_code in (` + utils.GetOrmInReplace(len(indexCode)) + `) ORDER BY data_time DESC  `
	err = o.Raw(sql, indexCode).Find(&items).Error
	return
}

// GetUsdaFasByConditionAndFrequency 根据条件获取涌益咨询指标列表
func GetUsdaFasByConditionAndFrequency(condition, frequency string, pars []interface{}) (items []*BaseFromUsdaFasIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_usda_fas_index WHERE 1=1 `

	if condition != "" {
		sql += condition
	}
	sql += ` AND frequency=?`
	sql += ` ORDER BY sort ASC, base_from_usda_fas_index_id ASC`
	err = o.Raw(sql, pars, frequency).Find(&items).Error
	return
}

func GetUsdaFasFrequencyByCondition(condition string, pars []interface{}) (items []string, err error) {
	sql := `SELECT DISTINCT frequency FROM base_from_usda_fas_index WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY FIELD(frequency,'日度','周度','旬度','月度','季度','半年度','年度') `
	o := global.DbMap[utils.DbNameIndex]
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

// GetUsdaFasDataDataTimeByIndexId 根据指标id获取指标数据的日期列表
func GetUsdaFasDataDataTimeByIndexId(indexIdList []int) (items []string, err error) {
	if len(indexIdList) == 0 {
		return
	}
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT DISTINCT data_time FROM base_from_usda_fas_data WHERE base_from_usda_fas_index_id IN (` + utils.GetOrmInReplace(len(indexIdList)) + `) ORDER BY data_time DESC`
	err = o.Raw(sql, indexIdList).Find(&items).Error
	if err != nil {
		return
	}
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		for i := range items {
			items[i] = utils.GormDateStrToDateStr(items[i])
		}
	}
	return
}

type BaseFromUsdaFasData struct {
	BaseFromUsdaFasDataId  int `orm:"column(base_from_usda_fas_data_id);pk" gorm:"primaryKey"`
	BaseFromUsdaFasIndexId int
	IndexCode              string
	DataTime               string
	Value                  string
	CreateTime             string
	ModifyTime             string
	DataTimestamp          int64
}

func (obj *BaseFromUsdaFasData) AfterFind(tx *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		if obj.CreateTime != "" {
			obj.CreateTime = utils.GormDateStrToDateTimeStr(obj.CreateTime)
		}
		if obj.ModifyTime != "" {
			obj.ModifyTime = utils.GormDateStrToDateTimeStr(obj.ModifyTime)
		}
		if obj.DataTime != "" {
			obj.DataTime = utils.GormDateStrToDateStr(obj.DataTime)
		}
	}
	return
}

type BaseFromUsdaFasIndexSearchItem struct {
	BaseFromUsdaFasIndexId int `orm:"column(base_from_usda_fas_index_id);pk" gorm:"primaryKey"`
	ClassifyId             int
	ParentClassifyId       int
	IndexCode              string
	IndexName              string
}

// BatchCheckUsdaFasEdbReq 指标数据结构体
type BatchCheckUsdaFasEdbReq struct {
	//IsJoinEdb      int      `form:"IsJoinEdb" description:"是否加到指标库，0:未加到指标库"`
	Frequencies   string `description:"频度；枚举值：日度、周度、月度、季度、半年度、年度"`
	Keyword       string `description:"关键字"`
	ClassifyIds   string `description:"所选品种id列表"`
	ListAll       bool   `form:"ListAll" json:"ListAll" description:"列表全选"`
	TradeCodeList string `form:"TradeCodeList" json:"TradeCodeList" description:"全选为false时, 该数组为选中; 全选为true时, 该数组为不选的指标"`
}

// GetUsdaFasItemList 模糊查询UsdaFas数据库指标列表
func GetUsdaFasItemList(condition string) (items []*BaseFromUsdaFasIndexSearchItem, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := "SELECT * FROM base_from_usda_fas_index  WHERE 1=1"
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql).Find(&items).Error
	return
}

func GetUsdaFasIndexDataByCode(indexCode string) (list []*BaseFromUsdaFasData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := `SELECT * FROM base_from_usda_fas_data WHERE index_code=? `
	err = o.Raw(sql, indexCode).Find(&list).Error
	return
}

func GetBaseFromUsdaFasIndexByIndexCode(indexCode string) (list *BaseFromUsdaFasIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_usda_fas_index WHERE index_code=? `
	err = o.Raw(sql, indexCode).First(&list).Error
	return
}

type BaseFromUsdaFasIndexType struct {
	Type2 string `gorm:"column:type_2"`
	Type3 string `gorm:"column:type_3"`
}

// Update 更新UsdaFas指标基础信息
func (item *BaseFromUsdaFasIndex) Update(cols []string) (err error) {
	o := global.DbMap[utils.DbNameIndex]
	err = o.Select(cols).Updates(item).Error
	return
}

// EditUsdaFasIndexInfoResp 新增指标的返回
type EditUsdaFasIndexInfoResp struct {
	BaseFromUsdaFasIndexId int    `description:"指标ID"`
	IndexCode              string `description:"指标code"`
}

type UsdaFasIndexSource2EdbReq struct {
	EdbCode       string
	EdbName       string
	Frequency     string
	Unit          string
	ClassifyId    int
	AdminId       int
	AdminRealName string
}

func GetUsdaFasFrequencyByClassifyId(classifyId int) (items []*GlFrequency, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT frequency FROM base_from_usda_fas_index WHERE classify_id = ? `
	sql += ` GROUP BY frequency ORDER BY frequency ASC `
	err = o.Raw(sql, classifyId).Find(&items).Error
	return
}