eta_api/models/manual_edb.go

package models

import (
	"eta/eta_api/utils"
	"github.com/beego/beego/v2/client/orm"
	"github.com/rdlucklib/rdluck_tools/paging"
)

// TargetDetailResp 指标数据结构体
type TargetDetailResp struct {
	Detail       *EdbInfoItem
	ClassifyList []*EdbdataClassify
}

// GetTargetByTradeCode
// @Description: 根据手工指标编码获取指标信息
// @author: Roc
// @datetime 2024-07-16 14:18:10
// @param tradeCode string
// @return item *Edbinfo
// @return err error
func GetTargetByTradeCode(tradeCode string) (item *EdbInfoItem, err error) {
	sql := `SELECT * FROM edbinfo WHERE TRADE_CODE = ?  `
	o := orm.NewOrmUsingDB("edb")
	err = o.Raw(sql, tradeCode).QueryRow(&item)
	return
}

// GetCountManualUserClassify
// @Description: 根据用户ID和分类ID获取关系数量
// @author: Roc
// @datetime 2024-07-16 14:27:58
// @param sysUserId int
// @param classifyId int
// @return total int
// @return err error
func GetCountManualUserClassify(sysUserId, classifyId int) (total int, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := `SELECT count(1) ct FROM manual_user_classify WHERE admin_id=? AND classify_id = ? `
	err = o.Raw(sql, sysUserId, classifyId).QueryRow(&total)

	return
}
func GetManualClassifyByClassifyId(classifyId int) (item *EdbdataClassify, err error) {
	o := orm.NewOrmUsingDB("edb")
	sql := ` SELECT classify_id,classify_name,parent_id FROM edbdata_classify WHERE classify_id = ?  `
	err = o.Raw(sql, classifyId).QueryRow(&item)

	return
}

// GetEdbDataListByCode
// @Description: 通过指标ID获取所有数据
// @author: Roc
// @datetime 2024-07-16 15:32:41
// @param tradeCode string
// @return items []*Edbdata
// @return err error
func GetEdbDataListByCode(tradeCode string) (items []*Edbdata, err error) {
	sql := ` SELECT  TRADE_CODE,DT,round(CLOSE,4) CLOSE,modify_time  FROM edbdata WHERE TRADE_CODE = ?  GROUP BY TRADE_CODE,DT ORDER BY DT DESC `
	o := orm.NewOrmUsingDB("edb")
	_, err = o.Raw(sql, tradeCode).QueryRows(&items)
	return
}

// EdbInfoListItem
type EdbInfoListItem struct {
	TradeCode    string `orm:"column(TRADE_CODE);pk" description:"指标code"`
	SecName      string `orm:"column(SEC_NAME);" description:"指标名称"`
	Unit         string `orm:"column(UNIT);" description:"单位"`
	Remark       string `orm:"column(REMARK);" description:"备注"`
	Frequency    string `description:"频度"`
	ClassifyId   int    `description:"分类id"`
	ClassifyName string `description:"分类名称"`
	CreateDate   string `description:"创建时间"`
	UserId       int    `description:"录入用户id"`
	NoticeTime   string `description:"通知时间"`
	Mobile       string `description:"录入者手机号"`
	ModifyDate   string `description:"待更新日期"`
	Status       string `description:"状态：未完成/完成"`
	UniqueCode   string
	IsJoinEdb    int8   `description:"指标库是否已添加：0-否；1-是"`
	UserName     string `description:"录入用户名称"`
}

// EdbListResp 指标数据结构体
type EdbListResp struct {
	List   []*EdbInfoListItem
	Paging *paging.PagingItem `description:"分页数据"`
}

// GetEdbInfoList
// @Description: 根据条件获取指标列表
// @author: Roc
// @datetime 2024-07-17 10:34:05
// @param condition string
// @param pars []interface{}
// @return items []*Edbinfo
// @return err error
func GetEdbInfoList(condition string, pars []interface{}, startSize, pageSize int) (items []*EdbInfoListItem, err error) {
	o := orm.NewOrmUsingDB("edb")
	sql := `SELECT DISTINCT a.* FROM edbinfo AS a  WHERE a.REMARK='手动' `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY a.create_date DESC `

	if pageSize > 0 {
		sql += ` LIMIT ?,? `
		pars = append(pars, startSize, pageSize)
	}

	_, err = o.Raw(sql, pars).QueryRows(&items)

	return
}

// GetCountEdbInfoList
// @Description: 根据条件获取指标数量
// @author: Roc
// @datetime 2024-07-17 14:51:00
// @param condition string
// @param pars []interface{}
// @return total int
// @return err error
func GetCountEdbInfoList(condition string, pars []interface{}) (total int, err error) {
	o := orm.NewOrmUsingDB("edb")
	sql := `SELECT COUNT(1) ct FROM edbinfo AS a  WHERE a.REMARK='手动' `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY a.create_date DESC `

	err = o.Raw(sql, pars).QueryRow(&total)

	return
}

// OnlyMultiAddEdbdata
// @Description: 批量添加指标数据
// @author: Roc
// @datetime 2024-07-18 16:48:55
// @param items []*Edbdata
// @return err error
func OnlyMultiAddEdbdata(items []*Edbdata) (err error) {
	o := orm.NewOrmUsingDB("edb")
	_, err = o.InsertMulti(utils.MultiAddNum, items)
	return
}

// DelEdbdataByCodeAndDateList
// @Description: 根据指标编码和日期列表批量删除指标的明细数据
// @author: Roc
// @datetime 2024-07-18 16:54:27
// @param tradeCode string
// @param dateList []string
// @return err error
func DelEdbdataByCodeAndDateList(tradeCode string, dateList []string) (err error) {
	num := len(dateList)
	if num <= 0 {
		return
	}

	o := orm.NewOrmUsingDB("edb")
	sql := `delete from  edbdata AS a  WHERE a.TRADE_CODE=? and DT in (` + utils.GetOrmInReplace(num) + `) `
	_, err = o.Raw(sql, tradeCode, dateList).Exec()

	return
}

// UpdateManualIsJoinEdbStatus
// @Description: 根据手工数据加入指标的状态
// @author: Roc
// @datetime 2024-07-22 11:29:13
// @param edbCode string
// @param isJoinEdb int 是否加入指标库
// @return err error
func UpdateManualIsJoinEdbStatus(edbCode string, isJoinEdb int8) (err error) {
	o := orm.NewOrmUsingDB("edb")
	sql := ` UPDATE edbinfo SET is_join_edb = ? WHERE TRADE_CODE =? `
	_, err = o.Raw(sql, isJoinEdb, edbCode).Exec()
	return
}

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