eta_chart_lib/models/data_manage/chart_info.go

package data_manage

import (
	"eta/eta_chart_lib/global"
	"eta/eta_chart_lib/utils"
	"gorm.io/gorm"
	"time"
)

type ChartInfo struct {
	ChartInfoId       int    `orm:"column(chart_info_id);pk" gorm:"primaryKey"`
	ChartName         string `description:"图表名称"`
	ChartNameEn       string `description:"英文图表名称"`
	ChartClassifyId   int    `description:"图表分类id"`
	SysUserId         int
	SysUserRealName   string
	UniqueCode        string `description:"图表唯一编码"`
	CreateTime        time.Time
	ModifyTime        time.Time
	DateType          int    `description:"日期类型:1:00年至今,2:10年至今,3:15年至今，4：年初至今，5:自定义时间"`
	StartDate         string `description:"自定义开始日期"`
	EndDate           string `description:"自定义结束日期"`
	IsSetName         int    `description:"设置名称"`
	EdbInfoIds        string `description:"指标id"`
	ChartType         int    `description:"生成样式:1:曲线图,2:季节性图,3:面积图，4：柱状图，5：散点图，6：组合图，7：柱方图，8：商品价格曲线图，9：相关性图，10:截面散点图, 11：雷达图"`
	Calendar          string `description:"公历/农历"`
	SeasonStartDate   string `description:"季节性图开始日期"`
	SeasonEndDate     string `description:"季节性图开始日期"`
	ChartImage        string `description:"图表图片"`
	Sort              int    `description:"排序字段，数字越小越排前面"`
	XMin              string `description:"图表X轴最小值"`
	XMax              string `description:"图表X轴最大值"`
	LeftMin           string `description:"图表左侧最小值"`
	LeftMax           string `description:"图表左侧最大值"`
	RightMin          string `description:"图表右侧最小值"`
	RightMax          string `description:"图表右侧最大值"`
	Right2Min         string `description:"图表右侧2最小值"`
	Right2Max         string `description:"图表右侧2最大值"`
	MinMaxSave        int    `description:"是否手动保存过上下限：0-否；1-是"`
	Disabled          int    `description:"是否禁用，0：启用，1：禁用，默认：0"`
	BarConfig         string `description:"柱方图的配置，json数据"`
	Source            int    `description:"1：ETA图库；2：商品价格曲线"`
	ExtraConfig       string `description:"图表额外配置,json数据"`
	SeasonExtraConfig string `description:"季节性图表中的配置,json数据"`
	StartYear         int    `description:"当选择的日期类型为最近N年类型时，即date_type=20, 用start_year表示N"`
	ChartThemeId      int    `description:"图表应用主题ID"`
	SourcesFrom       string `description:"图表来源"`
	Instructions      string `description:"图表说明"`
	MarkersLines      string `description:"标识线"`
	MarkersAreas      string `description:"标识区"`
	Unit              string `description:"中文单位名称"`
	UnitEn            string `description:"英文单位名称"`
	IsJoinPermission  int    `description:"是否加入权限管控，0：不加入；1：加入；默认：0"`
	ForumChartInfoId  int    `description:"社区的图表ID"`
	ChartAlias        string `description:"图表别名"`
	DateTypeNum       int    `description:"date_type=25(N月前)时的N值,其他N值可复用此字段"`
}

type ChartEdbInfoMapping struct {
	EdbInfoId           int         `description:"指标id"`
	SourceName          string      `description:"来源名称"`
	Source              int         `description:"来源id"`
	EdbCode             string      `description:"指标编码"`
	EdbName             string      `description:"指标名称"`
	EdbAliasName        string      `description:"指标名称（别名）"`
	EdbNameEn           string      `description:"英文指标名称"`
	EdbAliasNameEn      string      `description:"英文指标名称（别名）"`
	EdbType             int         `description:"指标类型:1:基础指标,2:计算指标"`
	Frequency           string      `description:"频率"`
	FrequencyEn         string      `description:"英文频率"`
	Unit                string      `description:"单位"`
	UnitEn              string      `description:"英文单位"`
	StartDate           string      `description:"起始日期"`
	EndDate             string      `description:"终止日期"`
	ModifyTime          string      `description:"指标最后更新时间"`
	ChartEdbMappingId   int         `description:"图表指标id"`
	ChartInfoId         int         `description:"图表id"`
	MaxData             float64     `description:"上限"`
	MinData             float64     `description:"下限"`
	IsOrder             bool        `description:"true:正序,false:逆序"`
	IsAxis              int         `description:"1:左轴,0:右轴"`
	EdbInfoType         int         `description:"1:标准指标,0:领先指标"`
	EdbInfoCategoryType int         `description:"0:普通指标,1:预测指标"`
	LeadValue           int         `description:"领先值"`
	LeadUnit            string      `description:"领先单位"`
	LeadUnitEn          string      `description:"领先英文单位"`
	ChartStyle          string      `description:"图表类型"`
	ChartColor          string      `description:"颜色"`
	PredictChartColor   string      `description:"预测数据的颜色"`
	ChartWidth          float64     `description:"线条大小"`
	ChartType           int         `description:"生成样式:1:曲线图,2:季节性图,3:面积图，4：柱状图，5：散点图，6：组合图"`
	LatestDate          string      `description:"数据最新日期"`
	LatestValue         float64     `description:"数据最新值"`
	UniqueCode          string      `description:"指标唯一编码"`
	MinValue            float64     `json:"-" description:"最小值"`
	MaxValue            float64     `json:"-" description:"最大值"`
	DataList            interface{} `gorm:"-"`
	IsNullData          bool        `gorm:"-" json:"-" description:"是否空数据"`
}

// AfterFind 在该模型上设置钩子函数，把日期转成正确的string，所以查询函数只能用Find函数，First或者Scan是不会触发该函数的来获取数据
func (m *ChartEdbInfoMapping) AfterFind(db *gorm.DB) (err error) {
	m.StartDate = utils.GormDateStrToDateStr(m.StartDate)
	m.EndDate = utils.GormDateStrToDateStr(m.EndDate)
	m.LatestDate = utils.GormDateStrToDateStr(m.LatestDate)
	m.ModifyTime = utils.GormDateStrToDateTimeStr(m.ModifyTime)

	return
}

// AreaExtraConf 面积图配置
type AreaExtraConf struct {
	IsHeap            int `description:"是否堆积 1-堆积 2-不堆积"`
	HeapWay           int `description:"堆积方式 1-普通 2-百分比"`
	StandardEdbInfoId int `description:"基准指标id"`
	NullDealWay       int `description:"空值处理方式，1-插值填充 2-前值填充 3-后值填充 4-等于0 5-删除日期"`
}

func GetChartEdbMappingList(chartInfoId int) (list []*ChartEdbInfoMapping, err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := ` SELECT a.*,b.source_name,b.source,b.edb_code,b.edb_name,b.edb_name_en,b.frequency,b.unit,b.unit_en,b.start_date,b.end_date,b.modify_time,b.latest_date,b.latest_value,b.unique_code,b.edb_info_type AS edb_info_category_type
             FROM chart_edb_mapping AS a
			 INNER JOIN edb_info AS b ON a.edb_info_id=b.edb_info_id
			 WHERE chart_info_id=? 
             ORDER BY chart_edb_mapping_id ASC `
	//_, err = o.Raw(sql, chartInfoId).QueryRows(&list)
	err = global.DbMap[utils.DbNameIndex].Raw(sql, chartInfoId).Find(&list).Error
	return
}

// BarChartInfoReq 柱方图预览请求数据
type BarChartInfoReq struct {
	EdbInfoIdList []BarChartInfoEdbItemReq `description:"指标信息"`
	DateList      []BarChartInfoDateReq    `description:"日期配置"`
	Sort          BarChartInfoSortReq      `description:"排序"`
}

type FutureGoodBarChartInfoReq struct {
	EdbInfoIdList []BarChartInfoEdbItemReq `description:"指标信息"`
	DateList      []BarChartInfoDateReq    `description:"日期配置"`
	XDataList     []XData                  `description:"横轴配置"`
	BaseEdbInfoId int                      `description:"日期基准指标id"`
}

// XData 商品价格曲线的的x轴数据
type XData struct {
	Name   string `description:"别名"`
	NameEn string `description:"英文别名"`
	IsHide int    `description:"是否隐藏，0不隐藏，1隐藏"`
}

// BarChartInfoEdbItemReq 柱方图预览请求数据（指标相关）
type BarChartInfoEdbItemReq struct {
	EdbInfoId int    `description:"指标ID"`
	Name      string `description:"别名"`
	NameEn    string `description:"英文别名"`
	Source    int    `description:"1：ETA图库；2：商品价格"`
	IsHide    int    `description:"是否隐藏该项，0不隐藏，1隐藏"`
}

// BarChartInfoDateReq 柱方图预览请求数据（日期相关）
type BarChartInfoDateReq struct {
	Type  int    `description:"配置类型"`
	Date  string `description:"固定日期"`
	Value int    `description:"N天的值"`
	Color string `description:"颜色"`
	Name  string `description:"别名"`
}

// BarChartInfoSortReq 柱方图预览请求数据（排序相关）
type BarChartInfoSortReq struct {
	Sort      int `description:"排序类型，0：默认，1：升序，2：降序"`
	DateIndex int `description:"日期数据的下标，从0开始"`
}

// SectionScatterReq 截面散点请求
type SectionScatterReq struct {
	XName       string `description:"x轴名称"`
	XNameEn     string `description:"x轴名称(英文)"`
	XUnitName   string `description:"x轴单位名称"`
	XUnitNameEn string `description:"x轴单位名称(英文)"`
	YName       string `description:"y轴名称"`
	YNameEn     string `description:"y轴名称(英文)"`
	YUnitName   string `description:"y轴单位名称"`
	YUnitNameEn string `description:"y轴单位名称(英文)"`
	XMinValue   string `description:"X轴的最小值"`
	XMaxValue   string `description:"X轴的最大值"`
	YMinValue   string `description:"Y轴的最小值"`
	YMaxValue   string `description:"Y轴的最大值"`
	//EdbList     []SectionScatterEdbItemReq    `description:"指标数据"`
	SeriesList []SectionScatterSeriesItemReq `description:"系列数据"`
}

// SectionScatterSeriesItemReq 系列的请求
type SectionScatterSeriesItemReq struct {
	Name            string `description:"系列名"`
	NameEn          string `description:"系列名(英文)"`
	Color           string `description:"颜色"`
	EdbInfoList     []SectionScatterEdbItemReq
	ShowTrendLine   bool `description:"是否展示趋势线"`
	ShowFitEquation bool `description:"是否展示方程式"`
	ShowRSquare     bool `description:"是否展示R平方"`
}

// SectionScatterEdbItemReq 截面散点请求的指标
type SectionScatterEdbItemReq struct {
	XEdbInfoId int    `description:"X轴的指标ID"`
	YEdbInfoId int    `description:"Y轴的指标ID"`
	Name       string `description:"别名"`
	NameEn     string `description:"英文别名"`
	XDateType  int    `description:"X轴的日期配置类型"`
	XDate      string `description:"X轴的日期固定日期"`
	XDateValue int    `description:"X轴的日期N天的值"`
	YDateType  int    `description:"Y轴的日期配置类型"`
	YDate      string `description:"Y轴的日期固定日期"`
	YDateValue int    `description:"Y轴的日期N天的值"`
	IsShow     bool   `description:"是否展示"`
}

// SectionScatterInfoResp 截面散点图数据
type SectionScatterInfoResp struct {
	XName       string                         `description:"x轴名称"`
	XNameEn     string                         `description:"x轴名称(英文)"`
	XUnitName   string                         `description:"x轴单位名称"`
	XUnitNameEn string                         `description:"x轴单位名称(英文)"`
	YName       string                         `description:"y轴名称"`
	YNameEn     string                         `description:"y轴名称(英文)"`
	YUnitName   string                         `description:"y轴单位名称"`
	YUnitNameEn string                         `description:"y轴单位名称(英文)"`
	XMinValue   string                         `description:"X轴的最小值"`
	XMaxValue   string                         `description:"X轴的最大值"`
	YMinValue   string                         `description:"Y轴的最小值"`
	YMaxValue   string                         `description:"Y轴的最大值"`
	DataList    []SectionScatterSeriesItemResp `description:"数据列"`
}

// SectionScatterSeriesItemResp 系列的返回
type SectionScatterSeriesItemResp struct {
	Name            string `description:"系列名"`
	NameEn          string `description:"系列名(英文)"`
	Color           string `description:"颜色"`
	EdbInfoList     []SectionScatterEdbItemResp
	ShowTrendLine   bool              `description:"是否展示趋势线"`
	ShowFitEquation bool              `description:"是否展示方程式"`
	ShowRSquare     bool              `description:"是否展示R平方"`
	TrendLine       string            `description:"方程式"`
	RSquare         string            `description:"R平方的值(决定系数R2)"`
	TrendLimitData  []CoordinatePoint `description:"趋势线的前后坐标点"`
}

// SectionScatterEdbItemResp 截面散点的返回参数
type SectionScatterEdbItemResp struct {
	XEdbInfoId int     `description:"X轴指标id"`
	XDate      string  `description:"X轴指标实际日期"`
	XName      string  `description:"X轴指标名称"`
	XNameEn    string  `description:"X轴指标英文名称"`
	XValue     float64 `description:"X轴实际值"`
	YEdbInfoId int     `description:"Y轴指标id"`
	YDate      string  `description:"Y轴指标实际日期"`
	YName      string  `description:"Y轴指标名称"`
	YNameEn    string  `description:"Y轴指标英文名称"`
	YValue     float64 `description:"Y轴实际值"`
	IsShow     bool    `description:"是否展示"`
	Name       string  `description:"标签名称"`
	NameEn     string  `description:"英文标签名称"`
}

// CoordinatePoint 坐标点
type CoordinatePoint struct {
	X float64
	Y float64
}

// RadarChartInfoReq 雷达图预览请求数据
type RadarChartInfoReq struct {
	DateList []RadarChartInfoDateReq `description:"日期配置"`
}

// RadarChartInfoEdbItemReq 雷达图预览请求数据（指标相关）
type RadarChartInfoEdbItemReq struct {
	EdbInfoId int    `description:"指标ID"`
	Name      string `description:"别名"`
	//NameEn    string `description:"英文别名"`
	//Source    int    `description:"1：ETA图库；2：商品价格"`
}

// RadarChartInfoDateReq 雷达图预览请求数据（日期相关）
type RadarChartInfoDateReq struct {
	Type  int    `description:"配置类型"`
	Date  string `description:"固定日期"`
	Value int    `description:"N天的值"`
	Color string `description:"颜色"`
	Name  string `description:"别名"`
}

// RadarChartInfoResp 雷达图数据
type RadarChartInfoResp struct {
	YDataList   []RadarYData `description:"数据列"`
	XEdbIdValue []int
}

// RadarYData 雷达图的y轴数据
type RadarYData struct {
	Date  string    `description:"数据日期"`
	Color string    `description:"数据颜色"`
	Name  string    `description:"别名"`
	Value []float64 `description:"每个指标的值"`
}

// GetAiPredictChartInfoByIndexId 获取AI预测模型图表
func GetAiPredictChartInfoByIndexId(source, indexId int) (item *ChartInfo, err error) {
	sql := `SELECT * FROM chart_info WHERE chart_info_id = (
		  SELECT chart_info_id FROM chart_edb_mapping WHERE source = ? AND edb_info_id = ?
		) LIMIT 1`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, source, indexId).First(&item).Error
	return
}