package models

import (
	"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
	EdbExist     int `description:"指标库是否已添加：0-否；1-是"`
}

// 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
}