eta_index_lib/models/edb_info.go

package models

import (
	"encoding/json"
	"errors"
	"eta_gn/eta_index_lib/global"
	"eta_gn/eta_index_lib/models/mgo"
	"eta_gn/eta_index_lib/utils"
	"fmt"
	"gorm.io/gorm"
	"strconv"
	"time"

	"go.mongodb.org/mongo-driver/bson"

	"github.com/beego/beego/v2/client/orm"
	"github.com/shopspring/decimal"
)

type EdbInfo struct {
	EdbInfoId        int       `gorm:"primaryKey;autoIncrement;column:edb_info_id" description:"唯一标识"`
	EdbInfoType      int       `gorm:"column:edb_info_type" description:"指标类型，0：普通指标，1：预测指标"`
	SourceName       string    `gorm:"column:source_name" description:"来源名称"`
	Source           int       `gorm:"column:source" description:"来源id"`
	EdbCode          string    `gorm:"column:edb_code" description:"指标编码"`
	EdbName          string    `gorm:"column:edb_name" description:"指标名称"`
	EdbNameEn        string    `gorm:"column:edb_name_en" description:"英文指标名称"`
	EdbNameSource    string    `gorm:"column:edb_name_source" description:"指标名称来源"`
	Frequency        string    `gorm:"column:frequency" description:"频率"`
	Unit             string    `gorm:"column:unit" description:"单位"`
	UnitEn           string    `gorm:"column:unit_en" description:"英文单位"`
	StartDate        string    `gorm:"column:start_date" description:"起始日期"`
	EndDate          string    `gorm:"column:end_date" description:"终止日期"`
	ClassifyId       int       `gorm:"column:classify_id" description:"分类id"`
	SysUserId        int       `gorm:"column:sys_user_id" description:"系统用户ID"`
	SysUserRealName  string    `gorm:"column:sys_user_real_name" description:"系统用户真实姓名"`
	UniqueCode       string    `gorm:"column:unique_code" description:"指标唯一编码"`
	CreateTime       time.Time `gorm:"column:create_time" description:"创建时间"`
	ModifyTime       time.Time `gorm:"column:modify_time" description:"修改时间"`
	BaseModifyTime   time.Time `gorm:"column:base_modify_time" description:"基础修改时间"`
	MinValue         float64   `gorm:"column:min_value" description:"指标最小值"`
	MaxValue         float64   `gorm:"column:max_value" description:"指标最大值"`
	CalculateFormula string    `gorm:"column:calculate_formula" description:"计算公式"`
	EdbType          int       `gorm:"column:edb_type" description:"指标类型:1:基础指标,2:计算指标"`
	IsUpdate         int       `gorm:"column:is_update" description:"当天是否已更新，1:未更新,2:已更新"`
	Sort             int       `gorm:"column:sort" description:"排序字段"`
	LatestDate       string    `gorm:"column:latest_date" description:"数据最新日期（实际日期）"`
	LatestValue      float64   `gorm:"column:latest_value" description:"数据最新值（实际值）"`
	EndValue         float64   `gorm:"column:end_value" description:"数据的最新值（预测日期的最新值）"`
	MoveType         int       `gorm:"column:move_type" description:"移动方式:1:领先(默认),2:滞后"`
	MoveFrequency    string    `gorm:"column:move_frequency" description:"移动频度"`
	NoUpdate         int8      `gorm:"column:no_update" description:"是否停止更新，0：继续更新；1：停止更新"`
	ServerUrl        string    `gorm:"column:server_url" description:"服务器地址"`
	ChartImage       string    `gorm:"column:chart_image" description:"图表图片"`
	Calendar         string    `gorm:"column:calendar;default:公历" description:"公历/农历"`
	EmptyType        int       `gorm:"column:empty_type" description:"空值处理类型（0查找前后35天，1不计算，2前值填充，3后值填充，4等于0）"`
	MaxEmptyType     int       `gorm:"column:max_empty_type" description:"MAX、MIN公式空值处理类型（1、等于0；2、跳过空值）"`
	DataDateType     string    `gorm:"column:data_date_type;size:255;default:交易日" description:"数据日期类型"`
	ManualSave       int       `gorm:"column:manual_save" description:"是否有手动保存过上下限: 0-否; 1-是"`
	TerminalCode     string    `gorm:"column:terminal_code" description:"终端编码，用于配置在机器上"`
	DataUpdateTime   string    `gorm:"column:data_update_time" description:"最近一次数据发生变化的时间"`
	ErDataUpdateDate string    `gorm:"column:er_data_update_date" description:"本次更新，数据发生变化的最早日期"`
	SourceIndexName  string    `gorm:"column:source_index_name" description:"数据源中的指标名称"`
	SubSource        int       `gorm:"column:sub_source" description:"子数据来源:0:经济数据库,1:日期序列"`
	SubSourceName    string    `gorm:"column:sub_source_name" description:"子数据来源名称"`
	IndicatorCode    string    `gorm:"column:indicator_code" description:"指标代码"`
	StockCode        string    `gorm:"column:stock_code" description:"证券代码"`
	Extra            string    `gorm:"column:extra" description:"指标的额外配置"`
	OriginalEdbCode  string    `description:"指标原始编码"`
}

func (e *EdbInfo) Add() (err error) {

	err = global.DEFAULT_DmSQL.Create(e).Error
	return
}

func (m *EdbInfo) AfterFind(db *gorm.DB) (err error) {
	if m.StartDate != "" {
		minDateTmp, e := time.ParseInLocation(utils.FormatDateWallWithLoc, m.StartDate, time.Local)
		if e != nil {
			err = e
			return
		}
		m.StartDate = minDateTmp.Format(utils.FormatDate)
	}
	if m.EndDate != "" {
		maxDateTmp, e := time.ParseInLocation(utils.FormatDateWallWithLoc, m.EndDate, time.Local)
		if e != nil {
			err = e
			return
		}
		m.EndDate = maxDateTmp.Format(utils.FormatDate)
	}
	if m.LatestDate != "" {
		maxDateTmp, e := time.ParseInLocation(utils.FormatDateWallWithLoc, m.LatestDate, time.Local)
		if e != nil {
			err = e
			return
		}
		m.LatestDate = maxDateTmp.Format(utils.FormatDate)
	}
	if m.DataUpdateTime != "" {
		maxDateTmp, e := time.ParseInLocation(utils.FormatDateWallWithLoc, m.DataUpdateTime, time.Local)
		if e != nil {
			err = e
			return
		}
		m.DataUpdateTime = maxDateTmp.Format(utils.FormatDate)
	}

	return
}

func (e *EdbInfo) ConvertDate() {
	e.StartDate = utils.GormDateStrToDateStr(e.StartDate)
	e.EndDate = utils.GormDateStrToDateStr(e.EndDate)
	e.LatestDate = utils.GormDateStrToDateStr(e.LatestDate)
	e.DataUpdateTime = utils.GormDateStrToDateStr(e.DataUpdateTime)

	return
}

func AddEdbInfo(item *EdbInfo) (lastId int64, err error) {

	err = global.DEFAULT_DmSQL.Create(item).Error
	return
}

type EdbInfoList struct {
	FrequencyEn     string                 `gorm:"-" description:"英文频率"`
	EdbNameAlias    string                 `gorm:"-" json:"-" description:"指标名称,别名"`
	RuleType        int                    `gorm:"-" description:"预测规则，1：最新，2：固定值"`
	FixedValue      float64                `gorm:"-" description:"固定值"`
	DataList        []*EdbData             `gorm:"-" description:"实际指标数据"`
	PredictDataList []*EdbData             `gorm:"-" description:"预测指标数据"`
	Button          EdbClassifyItemsButton `gorm:"-" description:"操作权限"`
	IsEnEdb         bool                   `gorm:"-" description:"是否展示英文标识"`

	EdbInfoId       int     `gorm:"primaryKey;autoIncrement;column:edb_info_id" description:"唯一标识"`
	EdbInfoType     int     `gorm:"column:edb_info_type" description:"指标类型，0：普通指标，1：预测指标"`
	SourceName      string  `gorm:"column:source_name" description:"来源名称"`
	Source          int     `gorm:"column:source" description:"来源id"`
	EdbCode         string  `gorm:"column:edb_code" description:"指标编码"`
	EdbName         string  `gorm:"column:edb_name" description:"指标名称"`
	EdbNameEn       string  `gorm:"column:edb_name_en" description:"英文指标名称"`
	Frequency       string  `gorm:"column:frequency" description:"频率"`
	Unit            string  `gorm:"column:unit" description:"单位"`
	UnitEn          string  `gorm:"column:unit_en" description:"英文单位"`
	StartDate       string  `gorm:"column:start_date" description:"起始日期"`
	EndDate         string  `gorm:"column:end_date" description:"终止日期"`
	ClassifyId      int     `gorm:"column:classify_id" description:"分类id"`
	SysUserId       int     `gorm:"column:sys_user_id" description:"系统用户ID"`
	SysUserRealName string  `gorm:"column:sys_user_real_name" description:"系统用户真实姓名"`
	UniqueCode      string  `gorm:"column:unique_code" description:"指标唯一编码"`
	CreateTime      string  `gorm:"column:create_time" description:"创建时间"`
	ModifyTime      string  `gorm:"column:modify_time" description:"修改时间"`
	EdbType         int     `gorm:"column:edb_type" description:"指标类型:1:基础指标,2:计算指标"`
	LatestDate      string  `gorm:"column:latest_date" description:"数据最新日期（实际日期）"`
	LatestValue     float64 `gorm:"column:latest_value" description:"数据最新值（实际值）"`
	EndValue        float64 `gorm:"column:end_value" description:"数据的最新值（预测日期的最新值）"`
	NoUpdate        int8    `gorm:"column:no_update" description:"是否停止更新，0：继续更新；1：停止更新"`
	ChartImage      string  `gorm:"column:chart_image" description:"图表图片"`
	EmptyType       int     `gorm:"column:empty_type" description:"空值处理类型（0查找前后35天，1不计算，2前值填充，3后值填充，4等于0）"`
	MaxEmptyType    int     `gorm:"column:max_empty_type" description:"MAX、MIN公式空值处理类型（1、等于0；2、跳过空值）"`
	DataDateType    string  `gorm:"column:data_date_type;size:255;default:交易日" description:"数据日期类型"`
	SubSource       int     `gorm:"column:sub_source" description:"子数据来源:0:经济数据库,1:日期序列"`
	SubSourceName   string  `gorm:"column:sub_source_name" description:"子数据来源名称"`
	IndicatorCode   string  `gorm:"column:indicator_code" description:"指标代码"`
	StockCode       string  `gorm:"column:stock_code" description:"证券代码"`
}

func (e *EdbInfoList) ConvertDate() {
	e.CreateTime = utils.GormDateStrToDateTimeStr(e.CreateTime)
	e.ModifyTime = utils.GormDateStrToDateTimeStr(e.ModifyTime)
	e.StartDate = utils.GormDateStrToDateStr(e.StartDate)
	e.EndDate = utils.GormDateStrToDateStr(e.EndDate)
	e.LatestDate = utils.GormDateStrToDateStr(e.LatestDate)

	return
}

type EdbClassifyItemsButton struct {
	AddButton         bool `description:"是否可添加"`
	OpButton          bool `description:"是否可编辑"`
	DeleteButton      bool `description:"是否可删除"`
	MoveButton        bool `description:"是否可移动"`
	ShowEdbRelation   bool `description:"是否展示关联指标"`
	ShowChartRelation bool `description:"是否展示关联图表"`
}

func GetEdbInfoByName(edbName string) (items []*EdbInfoList, err error) {
	sql := ` SELECT * FROM edb_info WHERE edb_name=? `
	err = global.DEFAULT_DmSQL.Raw(sql, edbName).Scan(&items).Error

	if err != nil {
		return
	}
	for _, edbInfoItem := range items {
		edbInfoItem.ConvertDate()
	}

	return
}

func ModifyEdbInfoNameSource(edbNameSource string, edbInfoId int) (err error) {

	sql := ` UPDATE  edb_info SET edb_name_source=? WHERE edb_info_id = ? `
	err = global.DEFAULT_DmSQL.Exec(sql, edbNameSource, edbInfoId).Error
	return
}

func GetEdbInfoById(edbInfoId int) (item *EdbInfo, err error) {
	sql := ` SELECT * FROM edb_info WHERE edb_info_id = ? `
	err = global.DEFAULT_DmSQL.Raw(sql, edbInfoId).First(&item).Error
	if err != nil {
		return
	}

	item.ConvertDate()

	return
}

func GetEdbInfoByUniqueCode(uniqueCode string) (item *EdbInfo, err error) {

	sql := ` SELECT * FROM edb_info WHERE unique_code=? `
	err = global.DEFAULT_DmSQL.Raw(sql, uniqueCode).First(&item).Error
	if err != nil {
		return
	}

	item.ConvertDate()
	return
}

func GetEdbInfoByIdList(edbInfoIdList []int) (items []*EdbInfo, err error) {
	num := len(edbInfoIdList)
	if num <= 0 {
		return
	}

	sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
	err = global.DEFAULT_DmSQL.Raw(sql, edbInfoIdList).Scan(&items).Error
	return
}

func (edbInfo *EdbInfo) Update(cols []string) (err error) {

	err = global.DEFAULT_DmSQL.Model(edbInfo).Select(cols).Updates(edbInfo).Error
	return
}

type EdbInfoSearchData struct {
	EdbDataId     int     `description:"数据ID"`
	EdbInfoId     int     `description:"指标ID"`
	DataTime      string  `description:"数据日期"`
	Value         float64 `description:"数据"`
	EdbCode       string  `description:"指标编码"`
	DataTimestamp int64   `description:"时间戳"`
}

func (m *EdbInfoSearchData) ConvertTimeStr() {
	m.DataTime = utils.GormDateStrToDateStr(m.DataTime)

	return
}

type FindEdbDataListAllCond struct {
	EdbInfoId         int
	StartDataTime     string
	StartDataTimeCond string
	EndDataTime       string
	EndDataTimeCond   string
}

func GetEdbDataListAll(source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (item []*EdbInfoSearchData, err error) {
	if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo {
		return GetEdbDataListAllByMongo(source, subSource, findEdbDataListAllCond, order)
	}

	return GetEdbDataListAllByMysql(source, subSource, findEdbDataListAllCond, order)
}

func GetEdbDataListAllByMysql(source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (items []*EdbInfoSearchData, err error) {
	if findEdbDataListAllCond.EdbInfoId <= 0 {
		return
	}

	sql := ``
	tableName := GetEdbDataTableName(source, subSource)
	sql = ` SELECT * FROM %s WHERE 1=1 `
	sql = fmt.Sprintf(sql, tableName)

	var condition string
	var pars []interface{}
	condition += " AND edb_info_id=? "
	pars = append(pars, findEdbDataListAllCond.EdbInfoId)

	if findEdbDataListAllCond.StartDataTime != "" && findEdbDataListAllCond.StartDataTimeCond != `` {
		condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.StartDataTimeCond)
		pars = append(pars, findEdbDataListAllCond.StartDataTime)
	}
	if findEdbDataListAllCond.EndDataTime != "" && findEdbDataListAllCond.EndDataTimeCond != `` {
		condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.EndDataTimeCond)
		pars = append(pars, findEdbDataListAllCond.EndDataTime)
	}

	if condition != "" {
		sql += condition
	}
	if order == 1 {
		sql += ` ORDER BY data_time ASC `
	} else {
		sql += ` ORDER BY data_time DESC `
	}
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&items).Error
	if err != nil {
		return
	}
	for _, v := range items {
		v.ConvertTimeStr()
	}

	return
}

func GetEdbDataListAllByTo(to *gorm.DB, source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (item []*EdbInfoSearchData, err error) {
	if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo {
		return GetEdbDataListAllByMongo(source, subSource, findEdbDataListAllCond, order)
	}

	return GetEdbDataListAllByMysqlTo(to, source, subSource, findEdbDataListAllCond, order)
}

func GetEdbDataListAllByMysqlTo(to *gorm.DB, source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (items []*EdbInfoSearchData, err error) {
	if findEdbDataListAllCond.EdbInfoId <= 0 {
		return
	}
	sql := ``
	tableName := GetEdbDataTableName(source, subSource)
	sql = ` SELECT * FROM %s WHERE 1=1 `
	sql = fmt.Sprintf(sql, tableName)

	var condition string
	var pars []interface{}
	condition += " AND edb_info_id=? "
	pars = append(pars, findEdbDataListAllCond.EdbInfoId)

	if findEdbDataListAllCond.StartDataTime != "" && findEdbDataListAllCond.StartDataTimeCond != `` {
		condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.StartDataTimeCond)
		pars = append(pars, findEdbDataListAllCond.StartDataTime)
	}
	if findEdbDataListAllCond.EndDataTime != "" && findEdbDataListAllCond.EndDataTimeCond != `` {
		condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.EndDataTimeCond)
		pars = append(pars, findEdbDataListAllCond.EndDataTime)
	}

	if condition != "" {
		sql += condition
	}
	if order == 1 {
		sql += ` ORDER BY data_time ASC `
	} else {
		sql += ` ORDER BY data_time DESC `
	}
	err = to.Raw(sql, pars...).Scan(&items).Error
	if err != nil {
		return
	}
	for _, v := range items {
		v.ConvertTimeStr()
	}

	return
}

func GetEdbDataListAllByMongo(source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (dataList []*EdbInfoSearchData, err error) {
	dataList = make([]*EdbInfoSearchData, 0)
	if findEdbDataListAllCond.EdbInfoId <= 0 {
		return
	}
	mogDataObj := mgo.EdbDataBusiness{}
	queryConditions := bson.M{
		"edb_info_id": findEdbDataListAllCond.EdbInfoId,
	}
	dateCondition, err := mgo.BuildDateCondition(findEdbDataListAllCond.StartDataTime, findEdbDataListAllCond.EndDataTime)
	if err != nil {
		return
	}
	if len(dateCondition) > 0 {
		queryConditions["data_time"] = dateCondition
	}

	sortList := make([]string, 0)
	if order == 1 {
		sortList = append(sortList, "data_time")
	} else {
		sortList = append(sortList, "-data_time")
	}
	tmpDataList, tmpErr := mogDataObj.GetAllDataList(queryConditions, sortList)
	if tmpErr != nil {
		err = tmpErr
		return
	}
	for k, v := range tmpDataList {
		dataList = append(dataList, &EdbInfoSearchData{
			EdbDataId:     k + 1,
			EdbInfoId:     v.EdbInfoId,
			EdbCode:       v.EdbCode,
			DataTime:      v.DataTime.Format(utils.FormatDate),
			Value:         v.Value,
			DataTimestamp: v.DataTimestamp,
		})
	}
	return
}

type EdbInfoMaxAndMinInfo struct {
	MinDate     string  `description:"最小日期" bson:"min_date"`
	MaxDate     string  `description:"最大日期" bson:"max_date"`
	MinValue    float64 `description:"最小值" bson:"min_value"`
	MaxValue    float64 `description:"最大值" bson:"max_value"`
	LatestValue float64 `gorm:"-" description:"最新值" bson:"latest_value"`
	LatestDate  string  `gorm:"-" description:"实际数据最新日期" bson:"latest_date"`
	EndValue    float64 `description:"最新值" bson:"end_value"`
}

func (m *EdbInfoMaxAndMinInfo) AfterFind(db *gorm.DB) (err error) {
	if m.MinDate != "" {
		minDateTmp, e := time.ParseInLocation(utils.FormatDateWallWithLoc, m.MinDate, time.Local)
		if e != nil {
			err = e
			return
		}
		m.MinDate = minDateTmp.Format(utils.FormatDate)
	}
	if m.MaxDate != "" {
		maxDateTmp, e := time.ParseInLocation(utils.FormatDateWallWithLoc, m.MaxDate, time.Local)
		if e != nil {
			err = e
			return
		}
		m.MaxDate = maxDateTmp.Format(utils.FormatDate)
	}
	return
}

type EdbInfoMaxAndMinInfoTmp struct {
	MinDate     time.Time `description:"最小日期" bson:"min_date"`
	MaxDate     time.Time `description:"最大日期" bson:"max_date"`
	MinValue    float64   `description:"最小值" bson:"min_value"`
	MaxValue    float64   `description:"最大值" bson:"max_value"`
	LatestValue float64   `gorm:"-" description:"最新值" bson:"latest_value"`
	LatestDate  time.Time `gorm:"-" description:"实际数据最新日期" bson:"latest_date"`
	EndValue    float64   `description:"最新值" bson:"end_value"`
}

func GetEdbInfoMaxAndMinInfo(source, subSource int, edbCode string) (item *EdbInfoMaxAndMinInfo, err error) {
	sql := ``
	tableName := GetEdbDataTableName(source, subSource)
	sql = ` SELECT MIN(data_time) AS min_date,MAX(data_time) AS max_date,MIN(value) AS min_value,MAX(value) AS max_value FROM %s WHERE edb_code=? `
	sql = fmt.Sprintf(sql, tableName)
	err = global.DEFAULT_DmSQL.Raw(sql, edbCode).First(&item).Error

	var latest_value float64
	sql = ` SELECT value AS latest_value FROM %s WHERE edb_code=? ORDER BY data_time DESC LIMIT 1 `
	sql = fmt.Sprintf(sql, tableName)
	err = global.DEFAULT_DmSQL.Raw(sql, edbCode).Scan(&latest_value).Error
	item.LatestValue = latest_value
	return
}

type EdbIndoDataUpdateTime struct {
	MinDate string `description:"本次更新，数据发生变化的最早日期"`
}

func (m *EdbIndoDataUpdateTime) AfterFind(db *gorm.DB) (err error) {
	if m.MinDate == "" {
		return
	}
	minDateTmp, err := time.ParseInLocation(utils.FormatDateWallWithLoc, m.MinDate, time.Local)
	if err != nil {
		return
	}
	m.MinDate = minDateTmp.Format(utils.FormatDate)
	return
}

func GetEdbDataUpdateTimeByModify(source, subSource int, edbCode string, dataUpdateTime string) (item *EdbIndoDataUpdateTime, err error) {
	sql := ``
	tableName := GetEdbDataTableName(source, subSource)
	sql = ` SELECT MIN(data_time) AS min_date FROM %s WHERE edb_code=? and modify_time>=?`
	sql = fmt.Sprintf(sql, tableName)
	err = global.DEFAULT_DmSQL.Raw(sql, edbCode, dataUpdateTime).First(&item).Error

	return
}

func ModifyEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
	sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?, end_value = ?,modify_time=NOW() WHERE edb_info_id=? `
	err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error

	return
}

func ModifyEdbTimeAndLastInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
	sql := `UPDATE edb_info SET start_date = ?, end_date = ?, is_update = 2,latest_date = ?, latest_value = ?, end_value = ?, modify_time = NOW() WHERE edb_info_id = ?`
	err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error

	return
}

func ModifyEdbDataUpdateTime(edbInfoId int, dataUpdateTime, erDataUpdateDate string) (err error) {
	sql := `UPDATE edb_info SET data_update_time = ?, er_data_update_date = ?, modify_time = NOW() WHERE edb_info_id = ?`
	err = global.DEFAULT_DmSQL.Exec(sql, dataUpdateTime, erDataUpdateDate, edbInfoId).Error

	return
}

func GetLteZeroEdbDataCount(source, subSource, edbInfoId int) (count int, err error) {
	if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo {
		return GetLteZeroEdbDataCountByMongo(source, subSource, edbInfoId)
	}

	return GetLteZeroEdbDataCountByMysql(source, subSource, edbInfoId)
}

func GetLteZeroEdbDataCountByMongo(source, subSource, edbInfoId int) (count int, err error) {
	mogDataObj := mgo.EdbDataBusiness{}
	whereQuery := bson.M{
		"edb_info_id": edbInfoId,
		"value":       bson.M{"$lte": 0},
	}
	tmpCount, err := mogDataObj.GetCountDataList(whereQuery)
	if err != nil {
		return
	}

	count = int(tmpCount)

	return
}

func GetLteZeroEdbDataCountByMysql(source, subSource, edbInfoId int) (count int, err error) {
	sql := ``
	tableName := GetEdbDataTableName(source, subSource)
	sql = ` SELECT COUNT(1) AS count FROM %s WHERE 1=1 AND edb_info_id =?  AND value <=0 `
	sql = fmt.Sprintf(sql, tableName)
	err = global.DEFAULT_DmSQL.Raw(sql, edbInfoId).Scan(&count).Error
	return
}

func GetLastEdbData(condition string, pars []interface{}, source, subSource int) (item *EdbInfoSearchData, err error) {
	sql := ``
	tableName := GetEdbDataTableName(source, subSource)
	sql = ` SELECT * FROM %s WHERE 1=1 `
	sql = fmt.Sprintf(sql, tableName)

	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY data_time DESC `
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).First(&item).Error
	return
}

func GetEdbInfoByEdbCode(source int, edbCode string) (item *EdbInfo, err error) {
	sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code=? `
	err = global.DEFAULT_DmSQL.Raw(sql, source, edbCode).First(&item).Error

	if errors.Is(err, orm.ErrNoRows) {
		err = nil
	} else {
		item.ConvertDate()
	}

	return
}

func GetEdbInfoOnlyByEdbCode(edbCode string) (item *EdbInfo, err error) {
	sql := ` SELECT * FROM edb_info WHERE edb_code=? `
	err = global.DEFAULT_DmSQL.Raw(sql, edbCode).First(&item).Error
	if err != nil {
		return
	}

	item.ConvertDate()

	return
}

func GetEdbInfoCalculateListByCondition(condition string, pars []interface{}) (items []*EdbInfoCalculateMapping, err error) {
	sql := ` SELECT * FROM edb_info_calculate_mapping WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&items).Error

	return
}

func GetEdbInfoCalculateCountByCondition(condition string, pars []interface{}) (count int, err error) {
	sql := ` SELECT COUNT(1) AS count FROM edb_info_calculate_mapping WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&count).Error

	return
}

type EdbInfoSearchDataV1 struct {
	EdbDataId int    `description:"数据ID"`
	DataTime  string `description:"数据日期"`
	Value     string `description:"数据"`
}

func GetEdbDataListAllV1ByTo(to *gorm.DB, condition string, pars []interface{}, source, subSource, order int) (items []*EdbInfoSearchDataV1, err error) {
	sql := ``
	tableName := GetEdbDataTableName(source, subSource)
	sql = ` SELECT * FROM %s WHERE 1=1 `
	sql = fmt.Sprintf(sql, tableName)

	if condition != "" {
		sql += condition
	}
	if order == 1 {
		sql += ` ORDER BY data_time ASC `
	} else {
		sql += ` ORDER BY data_time DESC `
	}
	err = to.Raw(sql, pars...).Scan(&items).Error
	if err != nil {
		return
	}

	for _, item := range items {
		item.ConvertDate()
	}

	return
}

func UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo *EdbInfo) (err error, errMsg string) {
	maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
	if err != nil {
		if err.Error() == utils.ErrNoRow() {
			err = nil
			return
		}

		errMsg = "刷新指标失败！"
		err = errors.New("获取指标最大最小值失败，err:" + err.Error())
		return
	}
	if maxAndMinItem != nil {
		if edbInfo.ManualSave == 1 {
			err = ModifyEdbTimeAndLastInfo(edbInfo.EdbInfoId, maxAndMinItem)
			if err != nil {
				errMsg = "刷新指标失败！"
				err = errors.New("修改指标最新日期和数据失败，err:" + err.Error())
				return
			}
		} else {
			err = ModifyEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
			if err != nil {
				errMsg = "刷新指标失败！"
				err = errors.New("修改指标最大最小值失败，err:" + err.Error())
				return
			}
		}
	}

	go RefreshPredictStandardBaseByGeneralEdbInfoId(edbInfo.EdbInfoId)

	return
}

func UnifiedModifyEdbInfoMaxAndMinInfoDataUpdate(edbInfo *EdbInfo, dataUpdateTime string) (erDataUpdateDate string, err error, errMsg string) {
	err, errMsg = UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
	if err != nil {
		return
	}
	erDataUpdateTime, tErr := GetEdbDataUpdateTimeByModify(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode, dataUpdateTime)
	if tErr != nil {
		if tErr.Error() != utils.ErrNoRow() { //本次刷新指标数据无变化
			errMsg = "刷新指标失败！"
			err = errors.New("查询指标刷新，数据更新的最早日期失败，err:" + tErr.Error())
			return
		}
	} else if erDataUpdateTime.MinDate != "" {
		err = ModifyEdbDataUpdateTime(edbInfo.EdbInfoId, dataUpdateTime, erDataUpdateTime.MinDate)
		if err != nil {
			errMsg = "刷新指标失败！"
			err = errors.New("修改指标数据更新的最早日期失败，err:" + tErr.Error())
			return
		}
		erDataUpdateDate = erDataUpdateTime.MinDate
	}

	return
}

func UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo *EdbInfo, latestDateStr string, latestValue float64) (err error, errMsg string) {
	maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
	if err != nil {
		if err.Error() == utils.ErrNoRow() {
			err = nil
			return
		}

		errMsg = "刷新指标失败！"
		err = errors.New("获取指标最大最小值失败，err:" + err.Error())
		return
	}

	maxAndMinItem.EndValue = maxAndMinItem.LatestValue

	if maxAndMinItem != nil {
		latestDateStr = utils.GormDateStrToDateStr(latestDateStr)
		maxAndMinItem.LatestDate = latestDateStr
		maxAndMinItem.LatestValue = latestValue
		err = ModifyPredictEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
		if err != nil {
			errMsg = "刷新指标失败！"
			err = errors.New("修改指标最大最小值失败，err:" + err.Error())
			return
		}
	}

	return
}

func GetChartPredictEdbInfoDataListByConfList(predictEdbConfList []*PredictEdbConfAndData, filtrateStartDateStr, latestDateStr, endDateStr, frequency, dataDateType string, realPredictEdbInfoData []*EdbInfoSearchData) (predictEdbInfoData []*EdbInfoSearchData, minValue, maxValue float64, err error) {
	endDate, err := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
	if err != nil {
		return
	}

	latestDate, err := time.ParseInLocation(utils.FormatDate, latestDateStr, time.Local)
	if err != nil {
		return
	}

	startDate := latestDate

	if filtrateStartDateStr != `` {
		filtrateStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, filtrateStartDateStr, time.Local)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		if filtrateStartDate.After(latestDate) {
			startDate = filtrateStartDate.AddDate(0, 0, -1)
		}
	}

	existMap := make(map[string]float64)
	for _, v := range realPredictEdbInfoData {
		existMap[v.DataTime] = v.Value
	}

	predictEdbInfoData = make([]*EdbInfoSearchData, 0)

	for _, predictEdbConf := range predictEdbConfList {
		endDate = predictEdbConf.EndDate
		var tmpMinValue, tmpMaxValue float64 // 当前预测结果中的最大/最小值

		dayList := getPredictEdbDayList(startDate, endDate, frequency, dataDateType)
		if len(dayList) <= 0 { // 如果未来没有日期的话，那么就退出当前循环，进入下一个循环
			continue
		}

		switch predictEdbConf.RuleType {
		case 1: //1：最新
			var lastDataValue float64 //最新值
			tmpAllData := make([]*EdbInfoSearchData, 0)
			tmpAllData = append(tmpAllData, realPredictEdbInfoData...)
			tmpAllData = append(tmpAllData, predictEdbInfoData...)
			lenTmpAllData := len(tmpAllData)
			if lenTmpAllData > 0 {
				lastDataValue = tmpAllData[lenTmpAllData-1].Value
			}
			predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, lastDataValue, dayList, predictEdbInfoData, existMap)
			tmpMaxValue = lastDataValue
			tmpMinValue = lastDataValue
		case 2: //2：固定值
			tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			dataValue, _ := tmpValDecimal.Float64()
			predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, dataValue, dayList, predictEdbInfoData, existMap)

			tmpMaxValue = dataValue
			tmpMinValue = dataValue
		case 3: //3：同比
			tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			tbValue, _ := tmpValDecimal.Float64()
			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTb(predictEdbConf.PredictEdbInfoId, tbValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
		case 4: //4：同差
			tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			tcValue, _ := tmpValDecimal.Float64()
			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTc(predictEdbConf.PredictEdbInfoId, tcValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
		case 5: //5：环比
			tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			hbValue, _ := tmpValDecimal.Float64()
			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHb(predictEdbConf.PredictEdbInfoId, hbValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
		case 6: //6：环差
			tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			hcValue, _ := tmpValDecimal.Float64()
			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHc(predictEdbConf.PredictEdbInfoId, hcValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
		case 7: //7：N期移动均值
			nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleNMoveMeanValue(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
		case 8: //8：N期段线性外推值
			nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNLinearRegression(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
			if err != nil {
				return
			}
		case 9: //9：动态环差”预测规则；
			hcDataMap := make(map[string]float64) //规则计算的环差值map

			if predictEdbConf.PredictEdbInfoId > 0 {
				tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
				if tmpErr != nil {
					err = tmpErr
					return
				}
				for _, v := range tmpPredictEdbRuleDataList {
					hcDataMap[v.DataTime] = v.Value
				}
			} else {
				if len(predictEdbConf.DataList) <= 0 {
					return
				}
				for _, v := range predictEdbConf.DataList {
					currentDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.DataTime, time.Local)
					if tmpErr != nil {
						continue
					}
					if currentDate.Before(startDate) || currentDate.After(endDate) {
						continue
					}
					hcDataMap[v.DataTime] = v.Value
				}
			}

			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTrendsHC(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, hcDataMap, existMap)
		case 10: //10：根据 给定终值后插值 规则获取预测数据
			tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			finalValue, _ := tmpValDecimal.Float64()
			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleFinalValueHc(predictEdbConf.PredictEdbInfoId, finalValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
		case 11: //11：根据 季节性 规则获取预测数据
			var seasonConf SeasonConf
			tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &seasonConf)
			if tmpErr != nil {
				err = errors.New("季节性配置信息异常:" + tmpErr.Error())
				return
			}
			calendar := "公历"
			if seasonConf.Calendar == "农历" {
				calendar = "农历"
			}
			yearList := make([]int, 0)
			if seasonConf.YearType == 1 {
				if seasonConf.NValue < 1 {
					err = errors.New("连续N年不允许小于1")
					return
				}

				currYear := time.Now().Year()
				for i := 0; i < seasonConf.NValue; i++ {
					yearList = append(yearList, currYear-i-1)
				}
			} else {
				yearList = seasonConf.YearList
			}
			predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleSeason(predictEdbConf.PredictEdbInfoId, yearList, calendar, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
			if err != nil {
				return
			}
		case 12: //12：根据 移动平均同比 规则获取预测数据
			var moveAverageConf MoveAverageConf
			tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &moveAverageConf)
			if tmpErr != nil {
				err = errors.New("季节性配置信息异常:" + tmpErr.Error())
				return
			}
			predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleMoveAverageTb(predictEdbConf.PredictEdbInfoId, moveAverageConf.NValue, moveAverageConf.Year, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
			if err != nil {
				return
			}
		case 13: //13：根据 同比增速差值 规则获取预测数据
			tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			tbEndValue, _ := tmpValDecimal.Float64()
			predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTbzscz(predictEdbConf.PredictEdbInfoId, tbEndValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
		case 14: //14：根据 一元线性拟合 规则获取预测数据
			var ruleConf RuleLineNhConf
			err = json.Unmarshal([]byte(predictEdbConf.Value), &ruleConf)
			if err != nil {
				err = errors.New("一元线性拟合规则配置信息异常:" + err.Error())
				return
			}

			newNhccDataMap := make(map[string]float64)
			if predictEdbConf.PredictEdbInfoId > 0 { //已经生成的动态数据
				tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
				if tmpErr != nil {
					err = tmpErr
					return
				}
				for _, v := range tmpPredictEdbRuleDataList {
					newNhccDataMap[v.DataTime] = v.Value
				}
			} else { //未生成的动态数据，需要使用外部传入的数据进行计算
				newNhccDataMap, err, _ = getCalculateNhccData(append(realPredictEdbInfoData, predictEdbInfoData...), ruleConf)
			}

			predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleLineNh(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, newNhccDataMap, existMap)
			if err != nil {
				return
			}

		case 15: //15：N年均值：过去N年同期均值。过去N年可以连续或者不连续，指标数据均用线性插值补全为日度数据后计算；
			predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNAnnualAverage(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
			if err != nil {
				return
			}

		case 16: //16：年度值倒推
			predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleAnnualValueInversion(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
			if err != nil {
				return
			}
		}

		{
			lenPredictEdbInfoData := len(predictEdbInfoData)
			if lenPredictEdbInfoData > 0 {
				tmpDataEndTime, _ := time.ParseInLocation(utils.FormatDate, predictEdbInfoData[lenPredictEdbInfoData-1].DataTime, time.Local)
				if startDate.Before(tmpDataEndTime) {
					startDate = tmpDataEndTime
				}
			}
		}

		if tmpMinValue < minValue {
			minValue = tmpMinValue
		}
		if tmpMaxValue < maxValue {
			maxValue = tmpMaxValue
		}
	}

	return
}

func getPredictEdbDayList(startDate, endDate time.Time, frequency, dataDateType string) (dayList []time.Time) {
	if dataDateType == `` {
		dataDateType = `交易日`
	}
	switch frequency {
	case "日度":
		for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 1) {
			if dataDateType == `交易日` && (currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday) {
				continue
			}
			dayList = append(dayList, currDate)
		}
	case "周度":
		for currDate := startDate.AddDate(0, 0, 7); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 7) {
			dayList = append(dayList, currDate)
		}
	case "旬度":
		for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); {
			nextDate := currDate.AddDate(0, 0, 1)
			if nextDate.Day() == 11 || nextDate.Day() == 21 || nextDate.Day() == 1 {
				dayList = append(dayList, currDate)
			}
			currDate = nextDate
		}
	case "月度":
		for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
			currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
			if !currDate.After(endDate) && !currDate.Equal(startDate) {
				dayList = append(dayList, currDate)
			}
			currDate = currDate.AddDate(0, 0, 1)
		}
	case "季度":
		for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
			currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
			if !currDate.After(endDate) && !currDate.Equal(startDate) {
				if currDate.Month() == 3 || currDate.Month() == 6 || currDate.Month() == 9 || currDate.Month() == 12 {
					dayList = append(dayList, currDate)
				}
			}
			currDate = currDate.AddDate(0, 0, 1)
		}
	case "半年度":
		for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
			currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
			if !currDate.After(endDate) && !currDate.Equal(startDate) {
				if currDate.Month() == 6 || currDate.Month() == 12 {
					dayList = append(dayList, currDate)
				}
			}
			currDate = currDate.AddDate(0, 0, 1)
		}
	case "年度":
		for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
			currDate = time.Date(currDate.Year()+1, 12, 31, 0, 0, 0, 0, time.Now().Location())
			if !currDate.After(endDate) && !currDate.Equal(startDate) {
				dayList = append(dayList, currDate)
			}
		}
	}
	return
}

func GetPredictDataListByPredictEdbInfo(edbInfo *EdbInfo, order int, startDate string) (dataList []*EdbInfoSearchData, sourceEdbInfoItem *EdbInfo, err error, errMsg string) {
	predictEdbConfList, err := GetPredictEdbConfAndDataListById(edbInfo.EdbInfoId)
	if err != nil && err.Error() != utils.ErrNoRow() {
		errMsg = "获取预测指标配置信息失败"
		return
	}
	if len(predictEdbConfList) == 0 {
		errMsg = "获取预测指标配置信息失败"
		err = errors.New(errMsg)
		return
	}
	predictEdbConf := predictEdbConfList[0]

	sourceEdbInfoItem, err = GetEdbInfoById(predictEdbConf.SourceEdbInfoId)
	if err != nil {
		if err.Error() == utils.ErrNoRow() {
			errMsg = "找不到来源指标信息"
			err = errors.New(errMsg)
		}
		return
	}

	dataList, err, errMsg = GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem, predictEdbConfList, order, startDate)

	return
}

func GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem *EdbInfo, predictEdbConfList []*PredictEdbConfAndData, order int, startDate string) (dataList []*EdbInfoSearchData, err error, errMsg string) {
	allDataList := make([]*EdbInfoSearchData, 0)
	tmpDataList, err := GetEdbDataListAll(sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, FindEdbDataListAllCond{
		EdbInfoId:         sourceEdbInfoItem.EdbInfoId,
		StartDataTime:     startDate,
		StartDataTimeCond: ">=",
	}, 1)
	if err != nil {
		return
	}

	allDataList = tmpDataList

	if startDate != `` {
		allDataList, err = GetEdbDataListAll(sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, FindEdbDataListAllCond{
			EdbInfoId: sourceEdbInfoItem.EdbInfoId,
		}, 1)
		if err != nil {
			return
		}
	}

	predictDataList := make([]*EdbInfoSearchData, 0)

	endDateStr := edbInfo.EndDate //预测指标的结束日期

	var predictMinValue, predictMaxValue float64
	if len(predictEdbConfList) > 0 {
		predictDataList, predictMinValue, predictMaxValue, err = GetChartPredictEdbInfoDataListByConfList(predictEdbConfList, startDate, sourceEdbInfoItem.LatestDate, endDateStr, edbInfo.Frequency, edbInfo.DataDateType, allDataList)
		if err != nil {
			return
		}
	}

	if order == 1 {
		dataList = append(tmpDataList, predictDataList...)
	} else {
		lenPredictDataList := len(predictDataList)
		if lenPredictDataList > 0 {
			for k := range predictDataList {
				dataList = append(dataList, predictDataList[lenPredictDataList-k-1])
			}
		}

		lenDataList := len(tmpDataList)
		for k := range tmpDataList {
			dataList = append(dataList, tmpDataList[lenDataList-k-1])
		}
	}
	if len(predictDataList) > 0 {
		if edbInfo.MinValue > predictMinValue {
			edbInfo.MinValue = predictMinValue
		}

		if edbInfo.MaxValue < predictMaxValue {
			edbInfo.MaxValue = predictMaxValue
		}
	}
	return
}

func GetPredictEdbDataListAll(edbInfo *EdbInfo, order int) (items []*EdbInfoSearchData, err error) {
	items, err = GetEdbDataListAll(edbInfo.Source, edbInfo.SubSource, FindEdbDataListAllCond{
		EdbInfoId: edbInfo.EdbInfoId,
	}, order)

	return
}

func GetPredictEdbDataListAllByStartDate(edbInfo *EdbInfo, order int, startDate string) (items []*EdbInfoSearchData, err error) {
	items, err = GetEdbDataListAll(edbInfo.Source, edbInfo.SubSource, FindEdbDataListAllCond{
		EdbInfoId:         edbInfo.EdbInfoId,
		StartDataTime:     startDate,
		StartDataTimeCond: ">=",
	}, order)

	return
}

func ModifyPredictEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {

	sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?,end_value=?,modify_time=NOW() WHERE edb_info_id=? `
	err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, item.EndValue, edbInfoId).Error
	return
}

func GetEdbInfoItemByCondition(condition string, pars []interface{}) (item *EdbInfoList, err error) {
	sql := ` SELECT * FROM edb_info WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).First(&item).Error
	if err != nil {
		return
	}

	item.ConvertDate()

	return
}

type EdbInfoDetailReq struct {
	UniqueCode string `description:"指标唯一编码"`
	EdbCode    string `description:"指标编码"`
}

type EdbInfoDetailResp struct {
	SourceName    string `description:"来源名称"`
	EdbCode       string `description:"指标编码"`
	EdbName       string `description:"指标名称"`
	EdbNameSource string `description:"指标名称来源"`
	Frequency     string `description:"频率"`
	Unit          string `description:"单位"`
	StartDate     string `description:"起始日期"`
	EndDate       string `description:"终止日期"`
	UniqueCode    string `description:"指标唯一编码"`
	CreateTime    time.Time
	ModifyTime    time.Time
	MinValue      float64 `description:"指标最小值"`
	MaxValue      float64 `description:"指标最大值"`
	EdbNameEn     string  `description:"英文指标名称"`
	UnitEn        string  `description:"英文单位"`
	LatestDate    string  `description:"数据最新日期"`
	LatestValue   float64 `description:"数据最新值"`
	ChartImage    string  `description:"图表图片"`
	DataList      []*EdbInfoSearchData
}

type AddEdbInfoParams struct {
	Source       int    `description:"来源id"`
	EdbCode      string `description:"指标编码"`
	EdbName      string `description:"指标名称"`
	Frequency    string `description:"频率"`
	Unit         string `description:"单位"`
	ClassifyId   int    `description:"分类id"`
	TerminalCode string `description:"终端编码"`
	AdminId      int    `description:"创建人ID"`
	AdminName    string `description:"创建人姓名"`
}

func ModifyEdbInfoBaseTimeById(edbInfoId int, cTime time.Time) (err error) {

	sql := ` UPDATE edb_info SET base_modify_time = ? WHERE edb_info_id = ? `
	err = global.DEFAULT_DmSQL.Exec(sql, cTime, edbInfoId).Error
	return
}

type EdbInfoRefreshCheckReq struct {
	Source         int    `description:"来源id"`
	LatestDate     string `description:"数据最新日期"`
	FrequencyBatch string `description:"更新频度"`
}

type EdbInfoRefreshCheckResp struct {
	UpdateNum   int `description:"已更新指标数"`
	UnUpdateNum int `description:"未更新指标数"`
}

type ResetEdbClassifyReq struct {
	IndexCode      string `description:"指标编码"`
	ClassifyFirst  string `description:"一级分类名称"`
	ClassifySecond string `description:"二级分类名称"`
	ClassifyThird  string `description:"三级分类名称"`
}

func GetEdbInfoMaxSortByClassifyId(classifyId int) (sort int, err error) {

	sql := `SELECT COALESCE(MAX(sort), 0) as sort FROM edb_info WHERE classify_id=? `
	err = global.DEFAULT_DmSQL.Raw(sql, classifyId).Scan(&sort).Error
	return
}

func GetEdbAndClassifyMaxSort(parentId int, classifyType uint8) (maxSort int, err error) {
	classifyMaxSort, err := GetEdbClassifyMaxSort(parentId, classifyType)
	if err != nil {
		return
	}
	maxSort = classifyMaxSort
	edbMaxSort, err := GetEdbInfoMaxSortByClassifyId(parentId)
	if err != nil {
		return
	}
	if maxSort < edbMaxSort {
		maxSort = edbMaxSort
	}
	return
}

type CalculateEdbExtra struct {
	DateTag string `description:"时间序列的生成方式，all 表示所选指标的时间序列并集"`
}

type CalculateLjzEdbExtra struct {
	LastValType int `description:"最新值处理：0默认、均值填充"`
}

type EdbInfoEditRecord struct {
	EdbInfoId           int    `description:"指标ID"`
	EdbName             string `description:"指标名称"`
	Frequency           string `description:"频率"`
	Unit                string `description:"单位"`
	ClassifyId          int    `description:"分类id"`
	CalculateFormula    string `description:"计算公式"`
	OperateUserId       int    `description:"操作人id"`
	OperateUserRealName string `description:"操作人姓名"`
}

func GetEdbInfoByEdbCodeList(source int, edbCodeList []string) (items []*EdbInfo, err error) {
	num := len(edbCodeList)
	if num <= 0 {
		return
	}

	sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code IN (` + utils.GetOrmInReplace(num) + `) `
	err = global.DEFAULT_DmSQL.Raw(sql, source, edbCodeList).Scan(&items).Error

	return
}

func GetAddEdbMaxSortByClassifyId(classifyId int, classifyType uint8) (sort int) {
	sort, _ = GetEdbAndClassifyMaxSort(classifyId, classifyType)
	sort = sort + 1

	return
}

type EdbInfoExtra struct {
	ApiExtraPars string `description:"API-额外参数(如同花顺日期序列)"`
}

func GetEdbInfoNoUpdateTotalByIdList(edbInfoIdList []int) (total int, err error) {
	num := len(edbInfoIdList)
	if num <= 0 {
		return
	}

	sql := ` SELECT count(*) FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `)  and no_update=1`
	err = global.DEFAULT_DmSQL.Raw(sql, edbInfoIdList).Scan(&total).Error
	return
}

func TransEdbInfoDataList2SearchData(items []*EdbDataList) (list []*EdbInfoSearchData) {
	if len(items) == 0 {
		return
	}
	list = make([]*EdbInfoSearchData, 0)
	for _, v := range items {
		list = append(list, &EdbInfoSearchData{
			EdbDataId:     v.EdbDataId,
			EdbInfoId:     v.EdbInfoId,
			DataTime:      v.DataTime,
			DataTimestamp: v.DataTimestamp,
			Value:         v.Value,
		})
	}
	return
}

type SortEdbDataList []*EdbInfoSearchData

func (m SortEdbDataList) Len() int {
	return len(m)
}

func (m SortEdbDataList) Less(i, j int) bool {
	return m[i].DataTime > m[j].DataTime
}

func (m SortEdbDataList) Swap(i, j int) {
	m[i], m[j] = m[j], m[i]
}

func (m *EdbInfoSearchDataV1) ConvertDate() {
	m.DataTime = utils.GormDateStrToDateStr(m.DataTime)

	return
}