package data_manage import ( "errors" "eta_gn/eta_api/global" "eta_gn/eta_api/models/mgo" "eta_gn/eta_api/utils" "fmt" "strconv" "strings" "time" "github.com/rdlucklib/rdluck_tools/paging" "go.mongodb.org/mongo-driver/bson" "gorm.io/gorm" ) 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:"图表别名"` PublicStatus int `description:"公开状态；0：未公开；1：审批中；2：已驳回；3：已公开"` ChartPublicClassifyId int `description:"图表公开分类id"` PublicTime time.Time `description:"设置公开的时间"` PublicSort int `description:"公开分类下的排序字段"` } type ChartInfoMore struct { ChartInfo IsEnChart bool `description:"是否展示英文标识"` HaveOperaAuth bool `description:"是否有数据权限，默认：false"` } func AddChartInfo(item *ChartInfo) (lastId int64, err error) { err = global.DmSQL["data"].Create(item).Error if err != nil { return } lastId = int64(item.ChartInfoId) return } type ChartInfoItem struct { ChartInfoId int `description:"图表id"` ChartName string `description:"图表名称"` ChartClassifyId int `description:"图表分类"` } // GetChartInfoAll 用于分类展示 // @param source int 1：ETA图库；2：商品价格曲线 func GetChartInfoAll(sourceList []int, sysUserId int) (items []*ChartClassifyItems, err error) { num := len(sourceList) if num <= 0 { return } o := global.DmSQL["data"] sql := ` SELECT chart_info_id,chart_classify_id,chart_name AS chart_classify_name,chart_name_en AS chart_classify_name_en, unique_code,sys_user_id,sys_user_real_name,date_type,start_date,end_date,chart_type,calendar,season_start_date,season_end_date,source FROM chart_info WHERE source in (` + utils.GetOrmInReplace(num) + `) ` if sysUserId > 0 { sql += fmt.Sprintf(` AND sys_user_id = %d `, sysUserId) } sql += ` ORDER BY sort asc,chart_info_id ASC ` err = o.Raw(sql, sourceList).Scan(&items).Error return } // GetChartInfoByAdminId 获取所有我创建的图表，用于分类展示 func GetChartInfoByAdminId(sourceList []int, adminId int) (items []*ChartClassifyItems, err error) { num := len(sourceList) if num <= 0 { return } o := global.DmSQL["data"] sql := ` SELECT chart_info_id,chart_classify_id,chart_name AS chart_classify_name, unique_code,sys_user_id,sys_user_real_name,date_type,start_date,end_date,chart_type,calendar,season_start_date,season_end_date FROM chart_info where source in (` + utils.GetOrmInReplace(num) + `) AND sys_user_id = ? ORDER BY sort asc,create_time ASC ` err = o.Raw(sql, sourceList, adminId).Scan(&items).Error return } // 用于分类展示 func GetChartInfoAllList() (items []*ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info ORDER BY create_time ASC ` err = o.Raw(sql).Scan(&items).Error return } // 图表检索数据 type ChartInfoSearch struct { EdbCode string `description:"图表编码"` StartDate string `description:"起始日期"` EndDate string `description:"终止日期"` DataList []*EdbInfoSearchData } type ChartInfoSearchData struct { DataTime string `description:"数据日期"` Value float64 `description:"数据"` } type ChartInfoSearchResp struct { SearchItem *EdbInfoSearch `description:"图表分类"` Status int `description:"1:数据已存在于国能数据库，2:新数据"` } // ChartEsInfo // @Description: Es中的数据 type ChartEsInfo 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:"图表别名"` SharedUserIdList []int `description:"共享用户id列表" gorm:"-"` PublicStatus int `description:"公开状态；0：未公开；1：审批中；2：已驳回；3：已公开"` ChartPublicClassifyId int `description:"图表公开分类id"` PublicTime time.Time `description:"设置公开的时间"` } func GetChartEsInfoById(chartInfoId int) (item *ChartEsInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE chart_info_id=? ` err = o.Raw(sql, chartInfoId).First(&item).Error return } func GetChartInfoById(chartInfoId int) (item *ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE chart_info_id=? ` err = o.Raw(sql, chartInfoId).First(&item).Error return } func GetChartInfoViewById(chartInfoId int) (item *ChartInfoView, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE chart_info_id=? ` err = o.Raw(sql, chartInfoId).First(&item).Error if err != nil { return } // 日期格式转换 item.ConvertToResp() return } type SaveChartInfoReq struct { ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` ChartInfoId int `description:"图表id,新增时传0"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今，4：年初至今，5:自定义时间,6:起始日期至今"` StartDate string `description:"自定义开始日期"` EndDate string `description:"自定义结束日期"` Calendar string `description:"公历/农历"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` Right2Min string `description:"图表右侧最小值"` Right2Max string `description:"图表右侧最大值"` MinMaxSave int `description:"是否手动保存过上下限：0-否；1-是"` ExtraConfig string `description:"图表额外配置,json数据"` StartYear int `description:"当选择的日期类型为最近N年类型时，即date_type=20, 用start_year表示N"` } type ChartSaveItem struct { EdbInfoId int `description:"指标id"` MaxData float64 `description:"上限"` MinData float64 `description:"下限"` IsOrder bool `description:"true:正序,false:逆序"` IsAxis int `description:"1:左轴,0:右轴"` EdbInfoType int `description:"1:标准指标,0:领先指标"` LeadValue int `description:"领先值"` LeadUnit string `description:"领先单位"` ChartStyle string `description:"图表类型"` ChartColor string `description:"颜色"` PredictChartColor string `description:"预测数据的颜色"` ChartWidth float64 `description:"线条大小"` Source int `description:"1：ETA图库；2：商品价格曲线"` EdbAliasName string `description:"中文别名"` IsConvert int `description:"是否数据转换 0不转 1转"` ConvertType int `description:"数据转换类型 1乘 2除 3对数"` ConvertValue float64 `description:"数据转换值"` ConvertUnit string `description:"数据转换单位"` ConvertEnUnit string `description:"数据转换单位"` } func DeleteChartInfoAndData(chartInfoId int) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() sql := ` DELETE FROM chart_info WHERE chart_info_id=? ` err = to.Exec(sql, chartInfoId).Error if err != nil { return } sql = ` DELETE FROM chart_edb_mapping WHERE chart_info_id=? ` err = to.Exec(sql, chartInfoId).Error return } func GetChartInfoCountByCondition(condition string, pars []interface{}) (count int, err error) { o := global.DmSQL["data"] sql := ` SELECT COUNT(1) AS count FROM chart_info WHERE 1=1 ` if condition != "" { sql += condition } err = o.Raw(sql, pars...).Scan(&count).Error return } type EditChartInfoReq struct { ChartInfoId int `description:"图表ID"` ChartName string `description:"图表名称"` ChartNameEn string `description:"英文图表名称"` ChartClassifyId int `description:"分类id"` ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` ChartType int `description:"生成样式:1:曲线图,2:季节性图,3:面积图，4：柱状图，5：散点图，6：组合图，7：柱方图，8：商品价格曲线图，9：相关性图,10:截面散点图"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今，4：年初至今，5:自定义时间,6:起始日期至今，20:N年至今"` StartDate string `description:"自定义开始日期"` EndDate string `description:"自定义结束日期"` Calendar string `description:"公历/农历"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` Right2Min string `description:"图表右侧最小值"` Right2Max string `description:"图表右侧最大值"` MinMaxSave int `description:"是否手动保存过上下限：0-否；1-是"` BarChartInfo BarChartInfoReq `description:"柱方图的配置"` CorrelationChartInfo CorrelationChartInfoReq `description:"相关性图表配置"` ExtraConfig string `description:"图表额外配置信息,json字符串"` SeasonExtraConfig SeasonExtraItem `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:"英文单位名称"` ChartAlias string `description:"图表别名"` } type EditFutureGoodChartInfoReq struct { EditChartInfoReq BarChartInfo FutureGoodBarChartInfoReq } type MarkersLine struct { Axis int `json:"axis" description:"1左轴 2右轴 3横轴"` AxisName string `json:"axisName" description:"轴的名称，例如'左轴'"` MarkerType string `json:"markerType" description:"标识线或标识区"` MarkLineType int `json:"markLineType" description:"1:固定 2:指标计算"` Value string `json:"value" description:"连线指向的数值，例如'4000'"` FromValue string `json:"fromValue" description:"连线的起始点，可以为空"` ToValue string `json:"toValue" description:"连线的结束点，可以为空"` LineWidth float64 `json:"lineWidth" description:"连线的宽度"` DashStyle string `json:"dashStyle" description:"连线的虚线样式，例如'ShortDashDot'"` Color string `json:"color" description:"连线的颜色"` Text string `json:"text" description:"连线旁边显示的文本"` TextPosition string `json:"textPosition" description:"文本的显示位置，例如'bottom'"` TextColor string `json:"textColor" description:"文本颜色"` TextFontSize int `json:"textFontSize" description:"文本的字号大小"` IsShow bool `json:"isShow" description:"是否显示连线及文本"` EdbType int `json:"edbType" description:"指标类型 0图中第一个指标 1其他指标 前端回显用"` EdbInfoId int `json:"edbInfoId" description:"指标id"` Calculation int `json:"calculation" description:"计算方式 1区间均值 2区间均值加N倍标准差 3区间个数分位 4区间数值分位"` CalculationValue float64 `json:"calculationValue" description:"计算方式对应的值 2就是几倍标准差 3就是分位值 4就是数值值·"` TimeIntervalType int `json:"timeInterval" description:"时间区间 0跟随图表 1自定义"` StartDate MarkersLineTime `json:"startTime" description:"开始时间"` EndDate MarkersLineTime `json:"endTime" description:"结束时间"` } type MarkersLineTime struct { TimeType int `json:"timeType" description:"时间类型 1固定 2动态 3至今(仅结束时间)"` Date string `json:"date" description:"日期"` Conf EdbDateChangeConf `json:"conf" description:"动态时间配置"` } // EdbDateExtraConf // @Description: 导入指标日期前移和日期变换 type EdbDateChangeConf struct { MoveForward int `json:"moveForward" description:"前移的期数"` BaseDate int `json:"baseDate" description:"基准日期 0系统日期 1指标最新日期"` DateChange []*EdbDateConfDateChange } type EdbDateConfDateChange struct { Year int `json:"year" description:"前移的期数"` Month int `json:"month" description:"前移的期数"` Day int `json:"day" description:"前移的期数"` Frequency string `json:"moveForward" description:"频度变换"` FrequencyDay string `json:"frequencyDay" description:"频度的固定日期"` ChangeType int `json:"changeType" description:"日期变换类型1日期位移，2指定频率"` } type EditChartEnInfoReq struct { ChartInfoId int `description:"图表ID"` ChartNameEn string `description:"英文图表名称"` ChartEdbInfoList []*EditChartEnInfoEdbItem `description:"指标及配置信息"` ExtraConfig string `description:"图表额外配置信息,json字符串"` ChartAlias string `description:"图表别名"` } type EditChartEnInfoEdbItem struct { EdbInfoId int `description:"指标ID"` EdbNameEn string `description:"英文指标名称"` UnitEn string `description:"英文单位"` } // EditChartInfoBaseReq // @Description: 图表基础信息修改 type EditChartInfoBaseReq struct { ChartInfoId int `description:"图表ID"` ChartName string `description:"图表名称（根据语言版本区分）"` ChartEdbInfoList []*EditChartInfoEdbBaseItem `description:"指标及配置信息"` ExtraConfig string `description:"图表额外配置信息,json字符串"` ChartAlias string `description:"图表别名"` } type EditChartInfoEdbBaseItem struct { EdbInfoId int `description:"指标ID"` EdbName string `description:"指标名称（根据语言版本区分）"` Unit string `description:"单位（根据语言版本区分）"` } func SetForumChartInfoId(chartInfoId, forumChartInfoId int) (err error) { o := global.DmSQL["data"] sql := ` UPDATE chart_info SET forum_chart_info_id=?, modify_time = NOW() WHERE chart_info_id = ?` err = o.Exec(sql, forumChartInfoId, chartInfoId).Error return } type ChartInfoList struct { ChartInfoId int `orm:"column(chart_info_id);pk" gorm:"primaryKey" ` ChartName string `description:"来源名称"` ChartChartClassifyId int `description:"图表分类id"` SysUserId int SysUserRealName string UniqueCode string `description:"图表唯一编码"` CreateTime time.Time ModifyTime time.Time EdbInfoList []*EdbInfoList `gorm:"-" description:"实际指标数据"` ChartAlias string `description:"图表别名"` } type ChartInfoListResp struct { Paging *paging.PagingItem Item *ChartInfoList } func GetChartInfoByCondition(condition string, pars []interface{}) (item *ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE 1=1 ` if condition != "" { sql += condition } err = o.Raw(sql, pars...).First(&item).Error return } func GetChartInfoByNewest(source int) (item *ChartInfoList, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE 1=1 AND source = ? ORDER BY modify_time DESC LIMIT 1 ` err = o.Raw(sql, source).Scan(&item).Error return } type MoveChartInfoReq struct { ChartInfoId int `description:"图表ID"` PrevChartInfoId int `description:"上一个图表ID"` NextChartInfoId int `description:"下一个图表ID"` ChartClassifyId int `description:"分类id"` } func MoveChartInfo(chartInfoId, classifyId int) (err error) { o := global.DmSQL["data"] sql := ` UPDATE chart_info SET chart_classify_id = ? WHERE chart_info_id = ?` err = o.Exec(sql, classifyId, chartInfoId).Error return } type DeleteChartInfoReq struct { ChartInfoId int `description:"图表ID"` } type EdbDataList struct { EdbDataId int `description:" 指标数据ID"` EdbInfoId int `description:"指标ID"` DataTime string //`json:"-" description:"数据日期"` DataTimestamp int64 `description:"数据日期"` Value float64 `description:"数据值"` } // GetEdbDataList // @Description: 获取指标的数据（日期正序返回） // @author: Roc // @datetime 2024-05-07 10:48:43 // @param source int // @param subSource int // @param edbInfoId int // @param startDate string // @param endDate string // @return list []*EdbDataList // @return err error func GetEdbDataList(source, subSource, edbInfoId int, startDate, endDate string) (list []*EdbDataList, err error) { // 自有数据需要额外处理（从mongo获取） if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo { return getEdbDataListByMongo(source, subSource, edbInfoId, startDate, endDate) } if source == utils.DATA_SOURCE_THS && subSource == utils.DATA_SUB_SOURCE_HIGH_FREQUENCY && utils.UseMongo { return getThsHfEdbDataListByMongo(source, subSource, edbInfoId, startDate, endDate) } return getEdbDataListByMysql(source, subSource, edbInfoId, startDate, endDate) } func (e *EdbDataList) AfterFind(db *gorm.DB) (err error) { tmp, err := time.Parse(utils.FormatDateWallWithLoc, e.DataTime) if err != nil { return } e.DataTime = tmp.Format(utils.FormatDate) return } // getEdbDataListByMysql // @Description: 通过mysql查询指标数据 // @author: Roc // @datetime 2024-05-07 10:40:43 // @param source int // @param subSource int // @param edbInfoId int // @param startDate string // @param endDate string // @return list []*EdbDataList // @return err error func getEdbDataListByMysql(source, subSource, edbInfoId int, startDate, endDate string) (list []*EdbDataList, err error) { tableName := GetEdbDataTableName(source, subSource) if tableName == "" { err = errors.New("无效的渠道：" + strconv.Itoa(source)) list = make([]*EdbDataList, 0) return list, err } var pars []interface{} //pars = append(pars, edbInfoId) sql := `SELECT edb_data_id,edb_info_id,data_time,"value",data_timestamp FROM %s WHERE edb_info_id=? ` if startDate != "" { sql += ` AND data_time>=? ` pars = append(pars, startDate) } if endDate != "" { sql += ` AND data_time<=? ` pars = append(pars, endDate) } sql += ` ORDER BY data_time ASC ` sql = fmt.Sprintf(sql, tableName) o := global.DmSQL["data"] err = o.Raw(sql, utils.ForwardPars(pars, edbInfoId)...).Find(&list).Error //err = o.Raw(sql, pars...).Scan(&list).Error return } // getEdbDataListByMongo // @Description: 通过mongo查询指标数据 // @author: Roc // @datetime 2024-05-07 10:40:43 // @param source int // @param subSource int // @param endInfoId int // @param startDate string // @param endDate string // @return list []*EdbDataList // @return err error func getEdbDataListByMongo(source, subSource, edbInfoId int, startDate, endDate string) (list []*EdbDataList, err error) { list = make([]*EdbDataList, 0) mogDataObj := mgo.EdbDataBusiness{} // 构建查询条件 queryConditions := bson.M{ "edb_info_id": edbInfoId, } // 数据日期 dateCondition, err := mgo.BuildDateCondition(startDate, endDate) if err != nil { return } if len(dateCondition) > 0 { queryConditions["data_time"] = dateCondition } // 获取列表数据 tmpDataList, tmpErr := mogDataObj.GetAllDataList(queryConditions, []string{"data_time"}) if tmpErr != nil { err = tmpErr return } for k, v := range tmpDataList { list = append(list, &EdbDataList{ EdbDataId: k + 1, EdbInfoId: v.EdbInfoId, DataTime: v.DataTime.Format(utils.FormatDate), DataTimestamp: v.DataTimestamp, Value: v.Value, }) } return } type EdbDataListByUniqueCode struct { EdbDataId int `description:" 指标数据ID" json:"-"` EdbInfoId int `description:"指标ID" json:"-"` DataTime string //`json:"-" description:"数据日期"` DataTimestamp int64 `description:"数据日期"` Value float64 `description:"数据值"` } func GetEdbDataLunarList(endInfoId int, startDate, endDate string) (list []*EdbDataList, err error) { tableName := "edb_data_quarter" sql := `SELECT edb_data_id,edb_info_id,data_time,TRUNCATE(value,2) AS value,data_timestamp FROM %s WHERE edb_info_id=? AND data_time>=? AND data_time<=? ORDER BY data_time ASC ` sql = fmt.Sprintf(sql, tableName) o := global.DmSQL["data"] err = o.Raw(sql, endInfoId, startDate, endDate).Scan(&list).Error return } // GetEdbDataListMinAndMax // @Description: 根据条件数据库中获取指标数据最小值和最大值 // @author: Roc // @datetime 2024-05-07 13:11:32 // @param source int // @param subSource int // @param edbInfoId int // @param startDate string // @param endDate string // @return min_data float64 // @return max_data float64 // @return err error func GetEdbDataListMinAndMax(source, subSource, edbInfoId int, startDate, endDate string) (min_data, max_data float64, err error) { // 自有数据需要额外处理（从mongo获取） if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo { return getEdbDataListMinAndMaxByMongo(source, subSource, edbInfoId, startDate, endDate) } if source == utils.DATA_SOURCE_THS && subSource == utils.DATA_SUB_SOURCE_HIGH_FREQUENCY && utils.UseMongo { return getThsHfEdbDataListMinAndMaxByMongo(source, subSource, edbInfoId, startDate, endDate) } return getEdbDataListMinAndMaxByMysql(source, subSource, edbInfoId, startDate, endDate) } type MaxOrMinData struct { MinData float64 `json:"min_data"` MaxData float64 `json:"max_data"` } // getEdbDataListMinAndMaxByMysql // @Description: 根据条件从mysql数据库中获取指标数据最小值和最大值 // @author: Roc // @datetime 2024-05-07 13:11:22 // @param source int // @param subSource int // @param edbInfoId int // @param startDate string // @param endDate string // @return minData float64 // @return maxData float64 // @return err error func getEdbDataListMinAndMaxByMysql(source, subSource, edbInfoId int, startDate, endDate string) (minData, maxData float64, err error) { tableName := GetEdbDataTableName(source, subSource) sql := `SELECT min(value) AS min_data,max(value) AS max_data FROM %s WHERE edb_info_id=? ` var pars []interface{} var condition string if startDate != "" { condition += ` AND data_time>=? ` pars = append(pars, startDate) } if endDate != "" { condition += ` AND data_time<=? ` pars = append(pars, endDate) } sql = fmt.Sprintf(sql, tableName) o := global.DmSQL["data"] if condition != "" { sql += condition } var tmpData MaxOrMinData err = o.Raw(sql, utils.ForwardPars(pars, edbInfoId)...).Scan(&tmpData).Error if err != nil { return } minData = tmpData.MinData maxData = tmpData.MaxData return } // getEdbDataListMinAndMaxByMongo // @Description: 根据条件从mongo数据库中获取指标数据最小值和最大值 // @author: Roc // @datetime 2024-05-07 13:10:26 // @param source int // @param subSource int // @param edbInfoId int // @param startDate string // @param endDate string // @return minData float64 // @return maxData float64 // @return err error func getEdbDataListMinAndMaxByMongo(source, subSource, edbInfoId int, startDate, endDate string) (minData, maxData float64, err error) { mogDataObj := mgo.EdbDataBusiness{} // 构建查询条件 queryConditions := bson.M{ "edb_info_id": edbInfoId, } // 日期 dateCondition, err := mgo.BuildDateCondition(startDate, endDate) if err != nil { return } if len(dateCondition) > 0 { queryConditions["data_time"] = dateCondition } pipeline := []bson.M{ {"$match": queryConditions}, {"$group": bson.M{ "_id": nil, "min_value": bson.M{"$min": "$value"}, "max_value": bson.M{"$max": "$value"}, }}, {"$project": bson.M{"_id": 0}}, // 可选，如果不需要_id字段 } result, err := mogDataObj.GetEdbInfoMaxAndMinInfo(pipeline) if err != nil { return } minData = result.MinValue maxData = result.MaxValue return } type ChartEdbInfoMapping struct { EdbInfoId int `description:"指标id" gorm:"column:edb_info_id"` SourceName string `description:"来源名称" gorm:"column:source_name"` Source int `description:"来源id" gorm:"column:source"` SubSource int `description:"来源id" gorm:"column:sub_source"` EdbCode string `description:"指标编码" gorm:"column:edb_code"` EdbName string `description:"指标名称" gorm:"column:edb_name"` EdbAliasName string `description:"指标名称（别名）" gorm:"column:edb_alias_name"` EdbNameEn string `description:"英文指标名称" gorm:"column:edb_name_en"` EdbAliasNameEn string `description:"英文指标名称（别名）" gorm:"column:edb_alias_name_en"` EdbType int `description:"指标类型:1:基础指标,2:计算指标" gorm:"column:edb_type"` Frequency string `description:"频率" gorm:"column:frequency"` FrequencyEn string `description:"英文频率" gorm:"column:frequency_en"` Unit string `description:"单位" gorm:"column:unit"` UnitEn string `description:"英文单位" gorm:"column:unit_en"` StartDate string `description:"起始日期" gorm:"column:start_date"` EndDate string `description:"终止日期" gorm:"column:end_date"` ModifyTime string `description:"指标最后更新时间" gorm:"column:modify_time"` ChartEdbMappingId int `description:"图表指标id" gorm:"column:chart_edb_mapping_id"` ChartInfoId int `description:"图表id" gorm:"column:chart_info_id"` MaxData float64 `description:"上限" gorm:"column:max_data"` MinData float64 `description:"下限" gorm:"column:min_data"` IsOrder bool `description:"true:正序,false:逆序" gorm:"column:is_order"` IsAxis int `description:"1:左轴,0:右轴" gorm:"column:is_axis"` EdbInfoType int `description:"1:标准指标,0:领先指标" gorm:"column:edb_info_type"` EdbInfoCategoryType int `description:"0:普通指标,1:预测指标" gorm:"column:edb_info_category_type"` LeadValue int `description:"领先值" gorm:"column:lead_value"` LeadUnit string `description:"领先单位" gorm:"column:lead_unit"` LeadUnitEn string `description:"领先英文单位" gorm:"column:lead_unit_en"` ChartStyle string `description:"图表类型" gorm:"column:chart_style"` ChartColor string `description:"颜色" gorm:"column:chart_color"` PredictChartColor string `description:"预测数据的颜色" gorm:"column:predict_chart_color"` ChartWidth float64 `description:"线条大小" gorm:"column:chart_width"` ChartType int `description:"生成样式:1:曲线图,2:季节性图,3:面积图，4：柱状图，5：散点图，6：组合图，7：柱方图，8：商品价格曲线图，9：相关性图" gorm:"column:chart_type"` LatestDate string `description:"数据最新日期" gorm:"column:latest_date"` LatestValue float64 `description:"数据最新值" gorm:"column:latest_value"` MoveLatestDate string `description:"移动后的数据最新日期" gorm:"column:move_latest_date"` UniqueCode string `description:"指标唯一编码" gorm:"column:unique_code"` MinValue float64 `json:"-" description:"最小值" gorm:"column:min_value"` MaxValue float64 `json:"-" description:"最大值" gorm:"column:max_value"` DataList interface{} `gorm:"-"` IsNullData bool `json:"-" description:"是否空数据" gorm:"column:is_null_data"` MappingSource int `description:"1：ETA图库；2：商品价格曲线" gorm:"column:mapping_source"` RegionType string `description:"交易所来源，海外还是国内" json:"-" gorm:"column:region_type"` ClassifyId int `description:"分类id" gorm:"column:classify_id"` SubSourceName string `description:"子数据来源名称" gorm:"column:sub_source_name"` IndicatorCode string `description:"指标代码" gorm:"column:indicator_code"` IsConvert int `description:"是否数据转换 0不转 1转" gorm:"column:is_convert"` ConvertType int `description:"数据转换类型 1乘 2除 3对数" gorm:"column:convert_type"` ConvertValue float64 `description:"数据转换值" gorm:"column:convert_value"` ConvertUnit string `description:"数据转换单位" gorm:"column:convert_unit"` ConvertEnUnit string `description:"数据转换单位" gorm:"column:convert_en_unit"` IsJoinPermission int `description:"是否加入权限管控，0：不加入；1：加入；默认：0" gorm:"column:is_join_permission"` HaveOperaAuth bool `description:"是否有数据权限，默认：false" gorm:"column:have_opera_auth"` } type QuarterData struct { Year int DataList []*EdbDataList CuttingDataTimestamp int64 `description:"切割的时间戳"` ChartLegend string Years string } type QuarterXDateItem struct { StartDate time.Time EndDate time.Time ShowName string ChartLegend string CuttingDataTimestamp int64 `description:"切割的时间戳"` } type QuarterDataByUniqueCode struct { Year int DataList []*EdbDataListByUniqueCode } type ChartEdbInfoDetailResp struct { EdbInfo *ChartEdbInfoMapping } type ChartInfoDetailResp struct { ChartInfo *ChartInfoView EdbInfoList []*ChartEdbInfoMapping XEdbIdValue []int `description:"柱方图的x轴数据，指标id"` YDataList []YData `description:"柱方图的y轴数据"` XDataList []XData `description:"商品价格曲线的X轴数据"` BarChartInfo BarChartInfoReq `description:"柱方图的配置"` CorrelationChartInfo *CorrelationInfo `description:"相关性图表信息"` DataResp interface{} `description:"图表数据,根据图的类型而定的,没有确定的数据格式"` ClassifyLevels []string `description:"图表分类的UniqueCode-从最顶级到当前分类(给前端回显定位用的=_=!)"` } type FutureGoodChartInfoDetailResp struct { ChartInfo *ChartInfoView EdbInfoList []*ChartEdbInfoMapping XEdbIdValue []int `description:"柱方图的x轴数据，指标id"` YDataList []YData `description:"柱方图的y轴数据"` XDataList []XData `description:"商品价格曲线的X轴数据"` BarChartInfo FutureGoodBarChartInfoReq `description:"柱方图的配置"` CorrelationChartInfo *CorrelationInfo `description:"相关性图表信息"` DataResp interface{} `description:"图表数据,根据图的类型而定的,没有确定的数据格式"` } type BalanceTableChartListResp struct { List []*BalanceChartInfoDetailResp } type BalanceChartInfoDetailResp struct { *ChartInfoDetailResp ExcelEdbList []*ExcelChartEdbView } type ExcelChartEdbView struct { ExcelChartEdbId int DateSequenceStr string `description:"日期序列选区"` DataSequenceStr string `description:"数据序列选区"` FromTag string `description:"标签"` } // XData 商品价格曲线的的x轴数据 type XData struct { Name string `description:"别名"` NameEn string `description:"英文别名"` IsHide int `description:"是否隐藏，0不隐藏，1隐藏"` } // YData 柱方图的y轴数据 type YData struct { Date string `description:"数据日期"` ConfigDate time.Time `description:"配置的日期" json:"-"` Color string `description:"数据颜色"` Name string `description:"别名"` NameEn string `description:"英文别名"` Value []float64 `description:"每个指标的值"` NoDataEdbList []int `description:"没有数据的指标列表"` XEdbInfoIdList []int `description:"对应X轴的指标id列表"` NameList []string `description:"每个值对应的名称"` EnNameList []string `description:"每个值对应的英文名称"` EdbValMap map[int]float64 `description:"指标与值的对应" json:"-"` M []int `description:"对应开始日期的间隔值" json:"-"` Unit string `description:"中文单位名称"` UnitEn string `description:"英文单位名称"` SeriesEdb struct { SeriesId int `description:"因子指标系列ID"` EdbInfoId int `description:"指标ID"` } `description:"对应的系列指标"` } func ModifyChartInfoAndMapping(edbInfoIdStr string, req *SaveChartInfoReq, chartType int) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() //非季节图 if chartType != 2 { updateStr := `edb_info_ids=?, modify_time = NOW(), date_type=?, start_date=?, end_date=?, left_min=?, left_max=?, right_min=?, right_max=?, right2_min=?, right2_max=?, min_max_save=?, start_year=?` pars := []interface{}{edbInfoIdStr, req.DateType, req.StartDate, req.EndDate, req.LeftMin, req.LeftMax, req.RightMin, req.RightMax, req.Right2Min, req.Right2Max, req.MinMaxSave, req.StartYear} if req.ExtraConfig != `` { updateStr += `,extra_config=? ` pars = append(pars, req.ExtraConfig) } pars = append(pars, req.ChartInfoId) sql := ` UPDATE chart_info SET ` + updateStr + ` WHERE chart_info_id = ?` err = to.Exec(sql, pars...).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } } else { sql := ` UPDATE chart_info SET edb_info_ids=?, modify_time = NOW(), calendar=?, date_type=?, start_date=?, end_date=?, season_start_date=?, season_end_date=?, left_min=?, left_max=?, right_min=?, right_max=?, right2_min=?, right2_max=?, min_max_save=?, start_year=? WHERE chart_info_id = ?` err = to.Exec(sql, edbInfoIdStr, req.Calendar, req.DateType, req.StartDate, req.EndDate, req.StartDate, req.EndDate, req.LeftMin, req.LeftMax, req.RightMin, req.RightMax, req.Right2Min, req.Right2Max, req.MinMaxSave, req.StartYear, req.ChartInfoId).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } } var edbInfoIdArr []string mapList := make([]*ChartEdbMapping, 0) for _, v := range req.ChartEdbInfoList { edbInfoIdArr = append(edbInfoIdArr, strconv.Itoa(v.EdbInfoId)) var count int csql := `SELECT COUNT(1) AS count FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id=? ` err = to.Raw(csql, req.ChartInfoId, v.EdbInfoId).Scan(&count).Error if err != nil { fmt.Println("QueryRow Err:", err.Error()) return err } if count > 0 { msql := `UPDATE chart_edb_mapping SET modify_time = NOW(), max_data = ?, min_data = ?, is_order = ?, is_axis = ?, edb_info_type = ?, lead_value = ?, lead_unit = ?, chart_style = ?, chart_color = ?, predict_chart_color = ?, chart_width = ? WHERE chart_info_id =? AND edb_info_id=? ` err = to.Exec(msql, v.MaxData, v.MinData, v.IsOrder, v.IsAxis, v.EdbInfoType, v.LeadValue, v.LeadUnit, v.ChartStyle, v.ChartColor, v.PredictChartColor, v.ChartWidth, req.ChartInfoId, v.EdbInfoId).Error if err != nil { fmt.Println("chart_edb_mapping Err:" + err.Error()) return err } } else { mapItem := new(ChartEdbMapping) mapItem.ChartInfoId = req.ChartInfoId mapItem.EdbInfoId = v.EdbInfoId mapItem.CreateTime = time.Now() mapItem.ModifyTime = time.Now() timestamp := strconv.FormatInt(time.Now().UnixNano(), 10) mapItem.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp) mapItem.MaxData = v.MaxData mapItem.MinData = v.MinData mapItem.IsOrder = v.IsOrder mapItem.IsAxis = v.IsAxis mapItem.EdbInfoType = v.EdbInfoType mapItem.LeadValue = v.LeadValue mapItem.LeadUnit = v.LeadUnit mapItem.ChartStyle = v.ChartStyle mapItem.ChartColor = v.ChartColor mapItem.PredictChartColor = v.PredictChartColor mapItem.ChartWidth = v.ChartWidth mapList = append(mapList, mapItem) } } if len(mapList) > 0 { err = to.CreateInBatches(mapList, utils.MultiAddNum).Error if err != nil { fmt.Println("AddChartEdbMapping Err:" + err.Error()) return err } } if len(edbInfoIdArr) > 0 { edbInfoIdStr := strings.Join(edbInfoIdArr, ",") if edbInfoIdStr != "" { dsql := `DELETE FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id NOT IN(` + edbInfoIdStr + `)` err = to.Exec(dsql, req.ChartInfoId).Error if err != nil { fmt.Println("delete err:" + err.Error()) return err } } } return } // EditChartInfoAndMapping 修改 ETA图库 的 图表与指标 的关系 func EditChartInfoAndMapping(req *EditChartInfoReq, edbInfoIdStr string, calendar string, dateType, disabled int, barChartConf string, chartEdbInfoList []*ChartSaveItem, seasonExtra string) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() var pars []interface{} pars = append(pars, req.ChartName) pars = append(pars, req.ChartNameEn) pars = append(pars, edbInfoIdStr) pars = append(pars, req.ChartType) pars = append(pars, req.ChartClassifyId) pars = append(pars, disabled) pars = append(pars, barChartConf) pars = append(pars, req.ExtraConfig) pars = append(pars, seasonExtra) pars = append(pars, req.StartYear) pars = append(pars, req.ChartThemeId) pars = append(pars, req.SourcesFrom) pars = append(pars, req.Instructions) pars = append(pars, req.MarkersLines) pars = append(pars, req.MarkersAreas) pars = append(pars, req.Unit) pars = append(pars, req.UnitEn) sql := ` UPDATE chart_info SET chart_name =?, chart_name_en =?, edb_info_ids=?, chart_type=?, chart_classify_id = ?, modify_time = NOW(), disabled = ?, bar_config = ?, extra_config = ?, season_extra_config = ?, start_year = ?, chart_theme_id = ?, sources_from = ?, instructions = ?, markers_lines = ?, markers_areas = ?, unit = ?, unit_en = ? ` if calendar != "" { sql += `,calendar = ? ` pars = append(pars, calendar) } if dateType > 0 { sql += `,date_type = ? ` pars = append(pars, dateType) } sql += `,start_date = ? ` pars = append(pars, req.StartDate) sql += `,end_date = ? ` pars = append(pars, req.EndDate) sql += `,season_start_date = ? ` pars = append(pars, req.StartDate) sql += `,season_end_date = ? ` pars = append(pars, req.EndDate) sql += `,left_min = ? ` pars = append(pars, req.LeftMin) sql += `,left_max = ? ` pars = append(pars, req.LeftMax) sql += `,right_min = ? ` pars = append(pars, req.RightMin) sql += `,right_max = ? ` pars = append(pars, req.RightMax) sql += `,right2_min = ? ` pars = append(pars, req.Right2Min) sql += `,right2_max = ? ` pars = append(pars, req.Right2Max) sql += `,min_max_save = ? ` pars = append(pars, req.MinMaxSave) sql += `,chart_alias = ? ` pars = append(pars, req.ChartAlias) sql += `WHERE chart_info_id = ?` pars = append(pars, req.ChartInfoId) err = to.Exec(sql, pars...).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } chartEdbMappingIdList := make([]string, 0) for _, v := range chartEdbInfoList { // 查询该指标是否存在,如果存在的话,那么就去修改,否则新增 var tmpChartEdbMapping *ChartEdbMapping csql := `SELECT * FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id=? AND source = ? ` err = to.Raw(csql, req.ChartInfoId, v.EdbInfoId, utils.CHART_SOURCE_DEFAULT).Scan(&tmpChartEdbMapping).Error if err != nil && !utils.IsErrNoRow(err) { fmt.Println("QueryRow Err:", err.Error()) return err } if tmpChartEdbMapping != nil { chartEdbMappingIdList = append(chartEdbMappingIdList, strconv.Itoa(tmpChartEdbMapping.ChartEdbMappingId)) tmpChartEdbMapping.ModifyTime = time.Now() tmpChartEdbMapping.MaxData = v.MaxData tmpChartEdbMapping.MinData = v.MinData tmpChartEdbMapping.IsOrder = v.IsOrder tmpChartEdbMapping.IsAxis = v.IsAxis tmpChartEdbMapping.EdbInfoType = v.EdbInfoType tmpChartEdbMapping.LeadValue = v.LeadValue tmpChartEdbMapping.LeadUnit = v.LeadUnit tmpChartEdbMapping.ChartStyle = v.ChartStyle tmpChartEdbMapping.ChartColor = v.ChartColor tmpChartEdbMapping.PredictChartColor = v.PredictChartColor tmpChartEdbMapping.ChartWidth = v.ChartWidth tmpChartEdbMapping.EdbAliasName = v.EdbAliasName tmpChartEdbMapping.IsConvert = v.IsConvert tmpChartEdbMapping.ConvertType = v.ConvertType tmpChartEdbMapping.ConvertValue = v.ConvertValue tmpChartEdbMapping.ConvertUnit = v.ConvertUnit tmpChartEdbMapping.ConvertEnUnit = v.ConvertEnUnit err = to.Select("ModifyTime", "MaxData", "MinData", "IsOrder", "IsAxis", "EdbInfoType", "LeadValue", "LeadUnit", "ChartStyle", "ChartColor", "PredictChartColor", "ChartWidth", "EdbAliasName", "IsConvert", "ConvertType", "ConvertValue", "ConvertUnit", "ConvertEnUnit").Updates(tmpChartEdbMapping).Error if err != nil { fmt.Println("chart_edb_mapping Err:" + err.Error()) return err } } else { mapItem := new(ChartEdbMapping) mapItem.ChartInfoId = req.ChartInfoId mapItem.EdbInfoId = v.EdbInfoId mapItem.CreateTime = time.Now() mapItem.ModifyTime = time.Now() timestamp := strconv.FormatInt(time.Now().UnixNano(), 10) mapItem.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp + "_" + strconv.Itoa(v.EdbInfoId)) mapItem.MaxData = v.MaxData mapItem.MinData = v.MinData mapItem.IsOrder = v.IsOrder mapItem.IsAxis = v.IsAxis mapItem.EdbInfoType = v.EdbInfoType mapItem.LeadValue = v.LeadValue mapItem.LeadUnit = v.LeadUnit mapItem.ChartStyle = v.ChartStyle mapItem.ChartColor = v.ChartColor mapItem.PredictChartColor = v.PredictChartColor mapItem.ChartWidth = v.ChartWidth mapItem.Source = utils.CHART_SOURCE_DEFAULT mapItem.EdbAliasName = v.EdbAliasName mapItem.IsConvert = v.IsConvert mapItem.ConvertType = v.ConvertType mapItem.ConvertValue = v.ConvertValue mapItem.ConvertUnit = v.ConvertUnit mapItem.ConvertEnUnit = v.ConvertEnUnit err = to.Create(mapItem).Error if err != nil { fmt.Println("AddChartEdbMapping Err:" + err.Error()) return err } chartEdbMappingIdList = append(chartEdbMappingIdList, strconv.Itoa(mapItem.ChartEdbMappingId)) } } if len(chartEdbMappingIdList) > 0 { chartEdbMappingIdStr := strings.Join(chartEdbMappingIdList, ",") dsql := `DELETE FROM chart_edb_mapping WHERE chart_info_id=? AND chart_edb_mapping_id NOT IN(` + chartEdbMappingIdStr + `)` err = to.Exec(dsql, req.ChartInfoId).Error if err != nil { fmt.Println("delete err:" + err.Error()) return err } } return } // EditFutureGoodChartInfoAndMapping 修改商品价格曲线的 图表与指标 的关系 func EditFutureGoodChartInfoAndMapping(req *EditFutureGoodChartInfoReq, edbInfoIdStr string, calendar string, dateType, disabled int, barChartConf string) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() var pars []interface{} pars = append(pars, req.ChartName) pars = append(pars, edbInfoIdStr) pars = append(pars, req.ChartType) pars = append(pars, req.ChartClassifyId) pars = append(pars, disabled) pars = append(pars, barChartConf) sql := ` UPDATE chart_info SET chart_name =?, edb_info_ids=?, chart_type=?, chart_classify_id = ?, modify_time = NOW(), disabled = ?, bar_config = ? ` if calendar != "" { sql += `,calendar = ? ` pars = append(pars, calendar) } if dateType > 0 { sql += `,date_type = ? ` pars = append(pars, dateType) } sql += `,start_date = ? ` pars = append(pars, req.StartDate) sql += `,end_date = ? ` pars = append(pars, req.EndDate) sql += `,season_start_date = ? ` pars = append(pars, req.StartDate) sql += `,season_end_date = ? ` pars = append(pars, req.EndDate) sql += `,left_min = ? ` pars = append(pars, req.LeftMin) sql += `,left_max = ? ` pars = append(pars, req.LeftMax) sql += `,right_min = ? ` pars = append(pars, req.RightMin) sql += `,right_max = ? ` pars = append(pars, req.RightMax) sql += `,right2_min = ? ` pars = append(pars, req.Right2Min) sql += `,right2_max = ? ` pars = append(pars, req.Right2Max) sql += `WHERE chart_info_id = ?` pars = append(pars, req.ChartInfoId) err = to.Exec(sql, pars...).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } chartEdbMappingIdList := make([]string, 0) hasAddMap := make(map[int]struct{}) for _, v := range req.BarChartInfo.EdbInfoIdList { if _, ok := hasAddMap[v.EdbInfoId]; ok { continue } hasAddMap[v.EdbInfoId] = struct{}{} // 查询该指标是否存在,如果存在的话,那么就去修改,否则新增 var tmpChartEdbMapping *ChartEdbMapping csql := `SELECT * FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id=? AND source = ? ` err = to.Raw(csql, req.ChartInfoId, v.EdbInfoId, v.Source).Scan(&tmpChartEdbMapping).Error if err != nil && !utils.IsErrNoRow(err) { fmt.Println("QueryRow Err:", err.Error()) return err } if tmpChartEdbMapping != nil { chartEdbMappingIdList = append(chartEdbMappingIdList, strconv.Itoa(tmpChartEdbMapping.ChartEdbMappingId)) //tmpChartEdbMapping.ModifyTime = time.Now() //tmpChartEdbMapping.MaxData = v.MaxData //tmpChartEdbMapping.MinData = v.MinData //tmpChartEdbMapping.IsOrder = v.IsOrder //tmpChartEdbMapping.IsAxis = v.IsAxis //tmpChartEdbMapping.EdbInfoType = v.EdbInfoType //tmpChartEdbMapping.LeadValue = v.LeadValue //tmpChartEdbMapping.LeadUnit = v.LeadUnit //tmpChartEdbMapping.ChartStyle = v.ChartStyle //tmpChartEdbMapping.ChartColor = v.ChartColor //tmpChartEdbMapping.PredictChartColor = v.PredictChartColor //tmpChartEdbMapping.ChartWidth = v.ChartWidth //_, err = to.Update(tmpChartEdbMapping, "ModifyTime", "MaxData", "MinData", "IsOrder", "IsAxis", "EdbInfoType", "LeadValue", "LeadUnit", "ChartStyle", "ChartColor", "PredictChartColor", "ChartWidth") //if err != nil { // fmt.Println("chart_edb_mapping Err:" + err.Error()) // return err //} } else { mapItem := new(ChartEdbMapping) mapItem.ChartInfoId = req.ChartInfoId mapItem.EdbInfoId = v.EdbInfoId mapItem.CreateTime = time.Now() mapItem.ModifyTime = time.Now() timestamp := strconv.FormatInt(time.Now().UnixNano(), 10) mapItem.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp + "_" + strconv.Itoa(v.EdbInfoId)) //mapItem.MaxData = v.MaxData //mapItem.MinData = v.MinData mapItem.IsOrder = true mapItem.IsAxis = 1 mapItem.EdbInfoType = 1 //mapItem.LeadValue = v.LeadValue //mapItem.LeadUnit = v.LeadUnit //mapItem.ChartStyle = v.ChartStyle //mapItem.ChartColor = v.ChartColor //mapItem.PredictChartColor = v.PredictChartColor //mapItem.ChartWidth = v.ChartWidth mapItem.Source = v.Source err = to.Create(mapItem).Error if err != nil { fmt.Println("AddChartEdbMapping Err:" + err.Error()) return err } chartEdbMappingIdList = append(chartEdbMappingIdList, strconv.Itoa(mapItem.ChartEdbMappingId)) } } if len(chartEdbMappingIdList) > 0 { chartEdbMappingIdStr := strings.Join(chartEdbMappingIdList, ",") dsql := `DELETE FROM chart_edb_mapping WHERE chart_info_id=? AND chart_edb_mapping_id NOT IN(` + chartEdbMappingIdStr + `)` err = to.Exec(dsql, req.ChartInfoId).Error if err != nil { fmt.Println("delete err:" + err.Error()) return err } } return } // EditChartEnInfoAndEdbEnInfo // @Description: 修改图表和指标的英文信息 // @author: Roc // @datetime 2024-04-18 13:58:13 // @param req *EditChartEnInfoReq // @return err error func EditChartEnInfoAndEdbEnInfo(req *EditChartEnInfoReq) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() var pars []interface{} updateField := ` chart_name_en =?, ` pars = append(pars, req.ChartNameEn) if req.ExtraConfig != `` { updateField += ` extra_config =?, ` pars = append(pars, req.ExtraConfig) } updateField += ` chart_alias =?, ` pars = append(pars, req.ChartAlias) sql := ` UPDATE chart_info SET ` + updateField + ` modify_time = NOW() WHERE chart_info_id = ?` pars = append(pars, req.ChartInfoId) err = to.Exec(sql, pars).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } var edbInfoIdArr []string for _, v := range req.ChartEdbInfoList { edbInfoIdArr = append(edbInfoIdArr, strconv.Itoa(v.EdbInfoId)) var count int csql := `SELECT COUNT(1) AS count FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id=? ` err = to.Raw(csql, req.ChartInfoId, v.EdbInfoId).Scan(&count).Error if err != nil { fmt.Println("QueryRow Err:", err.Error()) return err } if count > 0 { msql := ` UPDATE edb_info SET edb_name_en =?, unit_en = ?, modify_time = NOW() WHERE edb_info_id = ? ` err = to.Exec(msql, v.EdbNameEn, v.UnitEn, v.EdbInfoId).Error if err != nil { fmt.Println("edb_info Err:" + err.Error()) return err } } } return } // EditChartBaseInfoAndEdbEnInfo // @Description: 修改图表和指标的基础信息 // @author: Roc // @datetime 2024-04-18 13:28:16 // @param req *EditChartInfoBaseReq // @param chartItem *ChartInfo // @param lang string // @return err error func EditChartBaseInfoAndEdbEnInfo(req *EditChartInfoBaseReq, chartItem *ChartInfo, lang string) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() updateChartCols := make([]string, 0) switch lang { case utils.EnLangVersion: chartItem.ChartNameEn = req.ChartName updateChartCols = append(updateChartCols, "ChartNameEn") default: chartItem.ChartName = req.ChartName updateChartCols = append(updateChartCols, "ChartName") } if req.ExtraConfig != `` { chartItem.ExtraConfig = req.ExtraConfig updateChartCols = append(updateChartCols, "ExtraConfig") } chartItem.ModifyTime = time.Now() updateChartCols = append(updateChartCols, "ModifyTime") err = to.Select(updateChartCols).Updates(chartItem).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } var edbInfoIdArr []string for _, v := range req.ChartEdbInfoList { edbInfoIdArr = append(edbInfoIdArr, strconv.Itoa(v.EdbInfoId)) var count int csql := `SELECT COUNT(1) AS count FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id=? ` err = to.Raw(csql, req.ChartInfoId, v.EdbInfoId).Scan(&count).Error if err != nil { fmt.Println("QueryRow Err:", err.Error()) return err } if count > 0 { msql := ` UPDATE edb_info SET modify_time = NOW() ` pars := make([]interface{}, 0) switch lang { case utils.EnLangVersion: msql += ` ,edb_name_en = ? ` pars = append(pars, v.EdbName) // 如果有传单位 if v.Unit != `` { msql += ` ,unit_en = ? ` pars = append(pars, v.Unit) } default: msql += ` ,edb_name = ? ` pars = append(pars, v.EdbName) // 如果有传单位 if v.Unit != `` { msql += ` ,unit = ? ` pars = append(pars, v.Unit) } } msql += ` WHERE edb_info_id = ? ` pars = append(pars, v.EdbInfoId) err = to.Exec(msql, pars...).Error if err != nil { fmt.Println("edb_info Err:" + err.Error()) return err } } } return } type AddChartInfoReq struct { ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` ChartClassifyId int `description:"分类id"` ChartName string `description:"图表名称"` ChartType int `description:"生成样式:1:曲线图,2:季节性图,3:面积图，4：柱状图，5：散点图，6：组合图，7：柱方图"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今，4：年初至今，5:自定义时间,6:起始日期至今"` StartDate string `description:"自定义开始日期"` EndDate string `description:"自定义结束日期"` Calendar string `description:"公历/农历"` 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-是"` BarChartInfo BarChartInfoReq `description:"柱方图的配置"` CorrelationChartInfo CorrelationChartInfoReq `description:"相关性图表配置"` ExtraConfig string `description:"图表额外配置信息,json字符串"` ChartImage string `description:"封面图" json:"-"` SeasonExtraConfig SeasonExtraItem `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:"英文单位名称"` ChartAlias string `description:"图表别名"` } type AddFutureGoodChartInfoReq struct { AddChartInfoReq BarChartInfo FutureGoodBarChartInfoReq } type PreviewChartInfoReq struct { ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今，4：年初至今，5:自定义时间,6:起始日期至今 20:最近N年"` ChartType int `description:"生成样式:1:曲线图,2:季节性图,3:面积图，4：柱状图，5：散点图，6：组合图，7：柱方图"` StartDate string `description:"自定义开始日期"` EndDate string `description:"自定义结束日期"` Calendar string `description:"公历/农历"` SeasonExtraConfig SeasonExtraItem `description:"季节性图表中的配置,json数据"` StartYear int `description:"当选择的日期类型为最近N年类型时，即date_type=20, 用start_year表示N"` ChartSource int ExtraConfig string `description:"图表额外配置信息,json字符串"` MarkersLines string `description:"标识线"` } type SeasonExtraItem struct { ChartLegend []SeasonChartLegend `description:"自定义的图例名称"` XStartDate string `description:"横坐标显示的起始日"` XEndDate string `description:"横坐标显示的截止日"` JumpYear int `description:"横坐标日期是否跨年，1跨年，0不跨年"` RightAxis SeasonRightAxis `description:"自定义右轴指标"` MaxMinLimits MaxMinLimits `description:"自定义同期上下限"` SamePeriodAverage SamePeriodAverage `description:"自定义同期均线"` SamePeriodStandardDeviation SamePeriodStandardDeviation `description:"自定义同期标准差"` } type SeasonChartLegend struct { Name string Value string } // 自定义右轴指标 type SeasonRightAxis struct { IndicatorType int `description:"右轴指标类型 1:左轴指标同比，2:指标库，3:预测指标 "` Style string `description:"生成样式"` Shape string `description:"形状"` ChartColor string `description:"图表颜色"` Size float64 `description:"大小"` Legend string `description:"图例名称"` NumFormat int `description:"数值格式 1:百分比 2:小数"` IsConnected int `description:"是否连接 0不连接 1连接"` LineColor string `description:"线条颜色"` LineWidth float64 `description:"线条宽度"` LineStyle string `description:"线条样式"` IsShow bool `description:"是否显示"` IsAdd bool `description:"是否添加"` } // 自定义同期上下限 type MaxMinLimits struct { Color string `description:"颜色"` Year int `description:"上下限取值范围"` Legend string `description:"图例名称"` IsShow bool `description:"是否显示"` IsAdd bool `description:"是否添加"` } // 自定义同期均线 type SamePeriodAverage struct { Color string `description:"颜色"` Year int `description:"均线取值范围"` Legend string `description:"图例名称"` LineType string `description:"线型"` LineWidth float64 `description:"线宽"` IsShow bool `description:"是否显示"` IsAdd bool `description:"是否添加"` } // 自定义同期均线 type SamePeriodAverageResp struct { Color string `description:"颜色"` Year int `description:"均线取值范围"` Legend string `description:"图例名称"` LineType string `description:"线型"` LineWidth float64 `description:"线宽"` IsShow bool `description:"是否显示"` List []*SamePeriodAverageData `description:"自定义均线列表"` IsAdd bool `description:"是否添加"` } type SamePeriodAverageData struct { DataTime string `description:"数据日期"` DataTimestamp int64 `description:"数据日期时间戳"` Value float64 `description:"均值"` } // 自定义同期标准差 type SamePeriodStandardDeviation struct { Color string `description:"颜色"` Year int `description:"标准差取值范围"` Legend string `description:"图例名称"` Multiple float64 `description:"标准差倍数"` IsShow bool `description:"是否显示"` IsAdd bool `description:"是否添加"` } type SamePeriodStandardDeviationResp struct { Color string `description:"颜色"` Year int `description:"标准差取值范围"` Legend string `description:"图例名称"` Multiple float64 `description:"标准差倍数"` IsShow bool `description:"是否显示"` List []*MaxMinLimitsData `description:"自定义标准差列表"` IsAdd bool `description:"是否添加"` } type SeasonChartResp struct { MaxMinLimits MaxMinLimitsResp `description:"自定义上下限"` SamePeriodAverage SamePeriodAverageResp `description:"自定义同期均线"` SamePeriodStandardDeviation SamePeriodStandardDeviationResp `description:"自定义同期标准差线"` RightAxis SeasonRightAxisResp `description:"自定义右轴指标"` } // 自定义右轴指标 type SeasonRightAxisResp struct { SeasonRightAxis EdbInfoList []*ChartEdbInfoMapping } type MaxMinLimitsResp struct { List []*MaxMinLimitsData `description:"自定义上下限列表"` Color string `description:"颜色"` Year int `description:"上下限取值范围"` Legend string `description:"图例名称"` IsShow bool `description:"是否显示"` IsAdd bool `description:"是否添加"` } type MaxMinLimitsData struct { DataTime string `description:"数据日期"` DataTimestamp int64 `description:"数据日期时间戳"` MaxValue float64 `description:"最大值"` MinValue float64 `description:"最小值"` } type AddChartInfoResp struct { ChartInfoId int `description:"图表id"` UniqueCode string `description:"图表唯一编码"` ChartType int `description:"生成样式:1:曲线图,2:季节性图"` ClassifyId int `description:"分类ID"` } type QuarterDataList []*QuarterData func (m QuarterDataList) Len() int { return len(m) } func (m QuarterDataList) Less(i, j int) bool { return m[i].Years < m[j].Years } func (m QuarterDataList) Swap(i, j int) { m[i], m[j] = m[j], m[i] } // BalanceSeasonChartLegendPreviewResp 表格中季节性图表预览接口 type BalanceSeasonChartLegendPreviewResp struct { List QuarterDataList } func ChartInfoSearchByKeyWord(keyword string, showSysId int) (searchList []*ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE 1=1 ` var pars []interface{} if keyword != "" { sql += `AND chart_name LIKE ? ` pars = append(pars, utils.GetLikeKeyword(keyword)) } if showSysId > 0 { sql += ` AND sys_user_id = ? ` pars = append(pars, showSysId) } sql += ` ORDER BY create_time DESC ` if keyword == "" { sql += ` LIMIT 100 ` } err = o.Raw(sql, pars...).Scan(&searchList).Error return } // ChartInfoSearchByEmptyKeyWord 没有关键字的时候获取默认100条数据 func ChartInfoSearchByEmptyKeyWord(showSysId int, sourceList []int, noPermissionChartIdList []int, startSize, pageSize int) (total int64, searchList []*ChartInfo, err error) { num := len(sourceList) o := global.DmSQL["data"] baseSql := ` FROM chart_info WHERE 1=1 AND source in (` + utils.GetOrmInReplace(num) + `)` var basePars []interface{} basePars = append(basePars, sourceList) if showSysId > 0 { baseSql += ` AND sys_user_id = ? ` basePars = append(basePars, showSysId) } lenNoPermissionChartIdList := len(noPermissionChartIdList) if lenNoPermissionChartIdList > 0 { baseSql += ` AND chart_info_id not in (` + utils.GetOrmInReplace(lenNoPermissionChartIdList) + `) ` basePars = append(basePars, noPermissionChartIdList) } // 查找数量 totalSql := " SELECT count(1) as total " + baseSql err = o.Raw(totalSql, basePars...).Scan(&total).Error if err != nil { return } // 查找列表数据 sql := " SELECT * " + baseSql + ` ORDER BY create_time DESC LIMIT ?,? ` basePars = append(basePars, startSize) basePars = append(basePars, pageSize) err = o.Raw(sql, basePars...).Scan(&searchList).Error return } func GetNextChartInfo(classifyId int) (item *ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT b.* FROM chart_classify AS a INNER JOIN chart_info AS b ON a.chart_classify_id=b.chart_classify_id WHERE a.chart_classify_id>? ORDER BY a.chart_classify_id ASC LIMIT 1 ` err = o.Raw(sql, classifyId).Scan(&item).Error return } // GetChartInfoRefreshData 获取图表关联的指标信息 func GetChartInfoRefreshData(chartInfoId int) (baseEdbInfoArr, calculateInfoArr []*EdbInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT b.* FROM chart_edb_mapping AS a INNER JOIN edb_info AS b ON a.edb_info_id=b.edb_info_id WHERE a.chart_info_id=? ` edbInfoList := make([]*EdbInfo, 0) err = o.Raw(sql, chartInfoId).Scan(&edbInfoList).Error if err != nil { return } for _, v := range edbInfoList { fmt.Println(v.EdbInfoId, v.EdbType) if v.EdbType == 1 { baseEdbInfoArr = append(baseEdbInfoArr, v) } else { calculateInfoArr = append(calculateInfoArr, v) newBaseEdbInfoArr, newCalculateInfoArr, err := GetChartRefreshEdbInfo(v.EdbInfoId, v.Source, 0) if err != nil { return baseEdbInfoArr, calculateInfoArr, err } baseEdbInfoArr = append(baseEdbInfoArr, newBaseEdbInfoArr...) calculateInfoArr = append(calculateInfoArr, newCalculateInfoArr...) } } return } func GetChartRefreshEdbInfo(edbInfoId, source, n int) (baseEdbInfoArr, calculateInfoArr []*EdbInfo, err error) { calculateList, err := GetEdbInfoCalculateMap(edbInfoId, source) if err != nil && !utils.IsErrNoRow(err) { return } n++ for _, item := range calculateList { fmt.Println(item.EdbInfoId) if item.EdbType == 1 { baseEdbInfoArr = append(baseEdbInfoArr, item) } else { calculateInfoArr = append(calculateInfoArr, item) if n > 10 { return } newBaseEdbInfoArr, newCalculateInfoArr, _ := GetChartRefreshEdbInfo(item.EdbInfoId, item.Source, n) baseEdbInfoArr = append(baseEdbInfoArr, newBaseEdbInfoArr...) calculateInfoArr = append(calculateInfoArr, newCalculateInfoArr...) } } return } type SetChartInfoImageReq struct { ChartInfoId int `description:"图表ID"` ImageUrl string `description:"图表图片地址"` } func EditChartInfoImage(req *SetChartInfoImageReq) (err error) { o := global.DmSQL["data"] sql := ` UPDATE chart_info SET chart_image=?, modify_time = NOW() WHERE chart_info_id = ? ` err = o.Exec(sql, req.ImageUrl, req.ChartInfoId).Error if err != nil { fmt.Println("EditChartInfoImage Err:", err.Error()) return err } return } type ChartInfoDetailFromUniqueCodeResp struct { ChartInfo *ChartInfoView Status bool `description:"true:图表存在，false:图表不存在"` EdbInfoList []*ChartEdbInfoMapping XEdbIdValue []int `description:"柱方图的x轴数据，指标id"` YDataList []YData `description:"柱方图的y轴数据"` XDataList []XData `description:"商品价格曲线的X轴数据"` BarChartInfo BarChartInfoReq `description:"柱方图的配置"` CorrelationChartInfo *CorrelationInfo `description:"相关性图表信息"` DataResp interface{} `description:"图表数据,根据图的类型而定的,没有确定的数据格式"` } type FutureChartInfoDetailFromUniqueCodeResp struct { ChartInfo *ChartInfoView Status bool `description:"true:图表存在，false:图表不存在"` EdbInfoList []*ChartEdbInfoMapping XEdbIdValue []int `description:"柱方图的x轴数据，指标id"` YDataList []YData `description:"柱方图的y轴数据"` XDataList []XData `description:"商品价格曲线的X轴数据"` BarChartInfo FutureGoodBarChartInfoReq `description:"柱方图的配置"` CorrelationChartInfo *CorrelationInfo `description:"相关性图表信息"` DataResp interface{} `description:"图表数据,根据图的类型而定的,没有确定的数据格式"` } func GetChartInfoByUniqueCode(uniqueCode string) (item *ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE unique_code=? ` err = o.Raw(sql, uniqueCode).First(&item).Error return } func GetChartInfoViewByUniqueCode(uniqueCode string) (item *ChartInfoView, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE unique_code=? ` err = o.Raw(sql, uniqueCode).First(&item).Error if err != nil { return } // 日期格式转换 item.ConvertToResp() return } // GetFirstChartInfoByClassifyId 获取当前分类下，且排序数相同 的排序第一条的数据 func GetFirstChartInfoByClassifyId(classifyId int) (item *ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE chart_classify_id=? order by sort asc,chart_info_id asc limit 1` err = o.Raw(sql, classifyId).First(&item).Error return } // UpdateChartInfoSortByClassifyId 根据图表id更新排序 func UpdateChartInfoSortByClassifyId(classifyId, nowSort, prevChartInfoId int, sourceList []int, updateSort string) (err error) { o := global.DmSQL["data"] num := len(sourceList) if num <= 0 { return } sql := ` update chart_info set sort = ` + updateSort + ` WHERE chart_classify_id=? AND source in (` + utils.GetOrmInReplace(num) + `) AND ` if prevChartInfoId > 0 { sql += ` (sort > ? or (chart_info_id > ` + fmt.Sprint(prevChartInfoId) + ` and sort = ` + fmt.Sprint(nowSort) + `))` } err = o.Exec(sql, classifyId, sourceList, nowSort).Error return } // Update 更新图表基础信息 func (chartInfo *ChartInfo) Update(cols []string) (err error) { o := global.DmSQL["data"] err = o.Select(cols).Updates(chartInfo).Error return } type ChartInfoView struct { ChartInfoId int `orm:"column(chart_info_id);pk" gorm:"primaryKey" ` ChartName string `description:"来源名称"` ChartNameEn string `description:"英文图表名称"` Unit string `description:"中文单位名称"` UnitEn string `description:"英文单位名称"` ChartClassifyId int `description:"图表分类id"` ChartClassifyName string `description:"图表名称"` SysUserId int SysUserRealName string UniqueCode string `description:"图表唯一编码"` CreateTime string ModifyTime string 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:季节性图"` Calendar string `description:"公历/农历"` SeasonStartDate string `description:"季节性图开始日期"` SeasonEndDate string `description:"季节性图开始日期"` ChartImage string `description:"图表图片"` Sort int `description:"排序字段，数字越小越排前面"` IsAdd bool `description:"true:已加入我的图库,false:未加入我的图库"` MyChartId int MyChartClassifyId string `description:"我的图表分类，多个用逗号隔开"` ChartClassify []*ChartClassifyView `gorm:"-"` EdbEndDate string `description:"指标最新更新日期"` XMin string `description:"图表X轴最小值"` XMax string `description:"图表X轴最大值"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` Right2Min string `description:"图表右侧最小值"` Right2Max string `description:"图表右侧最大值"` MinMaxSave int `description:"是否手动保存过上下限：0-否；1-是"` IsEdit bool `description:"是否有编辑权限"` IsEnChart bool `description:"是否展示英文标识"` WarnMsg string `description:"错误信息"` Disabled int `description:"是否禁用，0：启用，1：禁用，默认：0"` BarConfig string `description:"柱方图的配置，json数据" json:"-"` Source int `description:"1：ETA图库；2：商品价格曲线；3：相关性图表"` //CorrelationLeadUnit string `description:"相关性图表-领先单位"` ExtraConfig string `description:"图表额外配置,json数据"` ChartSource string `description:"图表来源str"` ChartSourceEn string `description:"图表来源（英文）"` Button ChartViewButton `gorm:"-" description:"操作按钮"` SeasonExtraConfig string `description:"季节性图表中的配置,json数据"` StartYear int `description:"当选择的日期类型为最近N年类型时，即date_type=20, 用start_year表示N"` ChartThemeId int `description:"图表应用主题ID"` ChartThemeStyle string `description:"图表应用主题样式"` SourcesFrom string `description:"图表来源"` Instructions string `description:"图表说明"` MarkersLines string `description:"标识线"` MarkersAreas string `description:"标识区"` IsJoinPermission int `description:"是否加入权限管控，0：不加入；1：加入；默认：0"` HaveOperaAuth bool `description:"是否有数据权限，默认：false"` ForumChartInfoId int `description:"社区的图表ID"` ChartAlias string `description:"图表别名"` SharedUserIdList []int `description:"共享用户id列表" gorm:"-"` SharedUserName string `description:"共享用户姓名，多个用英文,隔开" gorm:"-"` PublicStatus int `description:"公开状态；0：未公开；1：审批中；2：已驳回；3：已公开"` ChartPublicClassifyId int `description:"图表公开分类id"` PublicTime time.Time `description:"设置公开的时间"` ClassifyNamePath string `description:"分类完整路径"` PublicClassifyNamePath string `description:"公共分类完整路径"` CollectClassifyIdList []int `description:"所属收藏分类id列表" gorm:"-"` CollectClassifyNameList []string `description:"所属收藏分类名称列表" gorm:"-"` } type ChartViewButton struct { IsEdit bool `description:"是否有编辑权限"` IsEnChart bool `description:"是否展示英文标识"` IsAdd bool `description:"true:已加入我的图库,false:未加入我的图库"` IsCopy bool `description:"是否有另存为按钮"` IsSetName int `description:"设置名称"` } type ImageSvgToPngResp struct { Data struct { ResourceURL string `json:"ResourceUrl"` } `json:"Data"` ErrCode string `json:"ErrCode"` ErrMsg string `json:"ErrMsg"` IsSendEmail bool `json:"IsSendEmail"` Msg string `json:"Msg"` Ret int64 `json:"Ret"` Success bool `json:"Success"` } // GetChartInfoByClassifyIdAndName 根据分类id和图表名获取图表信息 func GetChartInfoByClassifyIdAndName(classifyId int, chartName string) (item *ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE chart_classify_id = ? and chart_name=? ` err = o.Raw(sql, classifyId, chartName).First(&item).Error return } // BatchChartRefreshReq 批量刷新图表请求 type BatchChartRefreshReq struct { ChartInfoCode []string `description:"图表编码数组"` PrimaryId int `description:"研报/智能研报/PPT主键"` SubId int `description:"报告章节ID"` Source string `description:"来源，枚举值：report、english_report、smart_report、ppt"` } // GetChartInfoListByUniqueCodeSlice 根据图表编码获取图表列表数据 func GetChartInfoListByUniqueCodeSlice(uniqueCodeSlice []string) (total int64, items []*ChartInfo, err error) { num := len(uniqueCodeSlice) if num <= 0 { return } sql := ` SELECT * FROM chart_info WHERE unique_code in (` + utils.GetOrmInReplace(num) + `) ` err = global.DmSQL["data"].Raw(sql, uniqueCodeSlice).Scan(&items).Error if err != nil { return } total = int64(len(items)) return } // GetChartInfoRefreshDataByChartInfoIdSlice 根据图表id切片获取对应的指标数据 func GetChartInfoRefreshDataByChartInfoIdSlice(chartInfoIdSlice []string) (baseEdbInfoArr, calculateInfoArr []*EdbInfo, err error) { o := global.DmSQL["data"] if len(chartInfoIdSlice) <= 0 { return } sql := ` SELECT b.* FROM chart_edb_mapping AS a INNER JOIN edb_info AS b ON a.edb_info_id=b.edb_info_id WHERE a.chart_info_id in (` + strings.Join(chartInfoIdSlice, ",") + `) ` edbInfoList := make([]*EdbInfo, 0) err = o.Raw(sql).Scan(&edbInfoList).Error if err != nil { return } for _, v := range edbInfoList { if v.EdbType == 1 { baseEdbInfoArr = append(baseEdbInfoArr, v) } else { calculateInfoArr = append(calculateInfoArr, v) newBaseEdbInfoArr, newCalculateInfoArr, err := GetChartRefreshEdbInfo(v.EdbInfoId, v.Source, 0) if err != nil { return baseEdbInfoArr, calculateInfoArr, err } baseEdbInfoArr = append(baseEdbInfoArr, newBaseEdbInfoArr...) calculateInfoArr = append(calculateInfoArr, newCalculateInfoArr...) } } return } // CopyAddChartInfoReq 复制并新增图表 type CopyAddChartInfoReq struct { ChartInfoId int `description:"待复制的图表id"` ChartClassifyId int `description:"分类id"` ChartName string `description:"图表名称"` } // ChartInfoListByEsResp 图表数据Es搜索返回 type ChartInfoListByEsResp struct { Paging *paging.PagingItem List []*ChartInfoMore } // ChartInfoListByEsRespV2 图表数据Es搜索返回 type ChartInfoListByEsRespV2 struct { Paging *paging.PagingItem List []*ChartInfoView } // GetChartInfoListByCondition 根据条件获取图表数据了列表 func GetChartInfoListByCondition(condition string, pars []interface{}, startSize, pageSize int) (items []*ChartInfo, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE 1=1 ` if condition != "" { sql += condition } sql += ` ORDER BY chart_info_id DESC LIMIT ?,? ` pars = append(pars, startSize, pageSize) err = o.Raw(sql, pars...).Scan(&items).Error return } // GetChartInfoListGroupByUserId 根据指标id列表、用户分组获取指标信息 func GetChartInfoListGroupByUserId(edbIdList []string) (items []*ChartInfo, err error) { num := len(edbIdList) if num <= 0 { return } o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE chart_info_id in (` + utils.GetOrmInReplace(num) + `) ORDER BY chart_info_id desc ` err = o.Raw(sql, edbIdList).Scan(&items).Error return } // GetChartInfoListByChartIdList // @Description: 根据图表id列表获取列表信息 // @param edbIdList // @return items // @return err func GetChartInfoListByChartIdList(charIdList []string) (items []*ChartInfo, err error) { num := len(charIdList) if num <= 0 { return } o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE chart_info_id in (` + utils.GetOrmInReplace(num) + `) ORDER BY chart_info_id desc ` err = o.Raw(sql, charIdList).Scan(&items).Error return } // GetChartInfoListByUserId // @Description: 根据图表id列表获取列表信息 // @param userIdList []int // @param source int // @return items // @return err func GetChartInfoListByUserId(userIdList []int, source int) (items []*ChartInfo, err error) { num := len(userIdList) if num <= 0 { return } o := global.DmSQL["data"] sql := ` SELECT * FROM chart_info WHERE source = ? AND sys_user_id in (` + utils.GetOrmInReplace(num) + `) ORDER BY chart_info_id desc ` err = o.Raw(sql, source, userIdList).Scan(&items).Error return } // ModifyChartInfoUserIdByCodeList 根据指标code列表修改创建人 func ModifyChartInfoUserIdByCodeList(edbIdList []string, userId int, userName string) (err error) { num := len(edbIdList) if num <= 0 { return } o := global.DmSQL["data"] sql := `UPDATE chart_info SET sys_user_id=?,sys_user_real_name=? WHERE chart_info_id in (` + utils.GetOrmInReplace(num) + `) ` err = o.Exec(sql, userId, userName, edbIdList).Error return } // ModifyChartInfoUserIdByOldUserId // @Description: 根据旧的用户id修改创建人 // @author: Roc // @datetime 2024-03-25 19:14:59 // @param oldUserId int // @param userId int // @param userName string // @return err error func ModifyChartInfoUserIdByOldUserId(oldUserIdList []int, userId int, userName string) (err error) { num := len(oldUserIdList) if num <= 0 { return } o := global.DmSQL["data"] sql := `UPDATE chart_info SET sys_user_id=?,sys_user_real_name=? WHERE sys_user_id in (` + utils.GetOrmInReplace(num) + `) ` err = o.Exec(sql, userId, userName, oldUserIdList).Error return } // BarChartInfoReq 柱方图预览请求数据 type BarChartInfoReq struct { EdbInfoIdList []BarChartInfoEdbItemReq `description:"指标信息"` DateList []BarChartInfoDateReq `description:"日期配置"` Sort BarChartInfoSortReq `description:"排序"` XEdbList []BarChartInfoEdbItemReq `description:"X轴选择的指标列表"` YEdbList []BarChartInfoEdbItemReq `description:"Y轴选择的指标列表"` Unit string `description:"中文单位"` UnitEn string `description:"英文单位"` MarkersLines string `description:"标识线"` } type FutureGoodBarChartInfoReq struct { EdbInfoIdList []BarChartInfoEdbItemReq `description:"指标信息"` DateList []BarChartInfoDateReq `description:"日期配置"` XDataList []XData `description:"横轴配置"` BaseEdbInfoId int `description:"日期基准指标id"` } // BarChartInfoEdbItemReq 柱方图预览请求数据（指标相关） type BarChartInfoEdbItemReq struct { EdbInfoId int `description:"指标ID"` Name string `description:"别名"` NameEn string `description:"英文别名"` Source int `description:"1：ETA图库；2：商品价格"` IsConvert int `description:"是否数据转换 0不转 1转"` ConvertType int `description:"数据转换类型 1乘 2除 3对数"` ConvertValue float64 `description:"数据转换值"` ConvertUnit string `description:"数据转换单位"` ConvertEnUnit string `description:"数据转换单位"` 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开始"` } // CorrelationChartInfoReq 相关性图表请求体 type CorrelationChartInfoReq struct { LeadValue int `description:"领先期数"` LeadUnit string `description:"频度"` CalculateValue int `description:"计算窗口"` CalculateUnit string `description:"计算频度"` BaseCalculateValue int `description:"基础计算窗口（滚动相关性的时候用到）"` BaseCalculateUnit string `description:"基础计算频度（滚动相关性的时候用到）"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今，4：年初至今，5:自定义时间,6:起始日期至今"` StartDate string `description:"开始日期"` EndDate string `description:"结束日期"` EdbInfoIdList []CorrelationChartInfoEdbItemReq `description:"指标信息"` } // CorrelationChartInfoEdbItemReq 相关性图表请求指标 type CorrelationChartInfoEdbItemReq struct { EdbInfoId int `description:"指标ID"` Name string `description:"别名"` NameEn string `description:"英文别名"` //Source int `description:"1：ETA图库；2：商品价格"` } // EditCorrelationChartInfoAndMapping 修改相关性图表的 图表与指标 的关系 func EditCorrelationChartInfoAndMapping(req *EditChartInfoReq, edbInfoIdStr string, calendar string, dateType, disabled int, barChartConf string, correlationChart *ChartInfoCorrelation, correlationUpdateCols []string) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() var pars []interface{} pars = append(pars, req.ChartName) pars = append(pars, edbInfoIdStr) pars = append(pars, req.ChartType) pars = append(pars, req.ChartClassifyId) pars = append(pars, disabled) pars = append(pars, barChartConf) pars = append(pars, req.ExtraConfig) pars = append(pars, req.SourcesFrom) sql := ` UPDATE chart_info SET chart_name =?, edb_info_ids=?, chart_type=?, chart_classify_id = ?, modify_time = NOW(), disabled = ?, bar_config = ?, extra_config = ?, sources_from = ? ` if calendar != "" { sql += `,calendar = ? ` pars = append(pars, calendar) } if dateType > 0 { sql += `,date_type = ? ` pars = append(pars, dateType) } sql += `,start_date = ? ` pars = append(pars, req.StartDate) sql += `,end_date = ? ` pars = append(pars, req.EndDate) sql += `,season_start_date = ? ` pars = append(pars, req.StartDate) sql += `,season_end_date = ? ` pars = append(pars, req.EndDate) sql += `,left_min = ? ` pars = append(pars, req.LeftMin) sql += `,left_max = ? ` pars = append(pars, req.LeftMax) sql += `,right_min = ? ` pars = append(pars, req.RightMin) sql += `,right_max = ? ` pars = append(pars, req.RightMax) sql += `WHERE chart_info_id = ?` pars = append(pars, req.ChartInfoId) err = to.Exec(sql, pars...).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } chartEdbMappingIdList := make([]string, 0) for _, v := range req.CorrelationChartInfo.EdbInfoIdList { // 查询该指标是否存在,如果存在的话,那么就去修改,否则新增 var tmpChartEdbMapping *ChartEdbMapping csql := `SELECT * FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id=? AND source = ? ` err = to.Raw(csql, req.ChartInfoId, v.EdbInfoId, utils.CHART_SOURCE_CORRELATION).Scan(&tmpChartEdbMapping).Error if err != nil && !utils.IsErrNoRow(err) { fmt.Println("QueryRow Err:", err.Error()) return err } if tmpChartEdbMapping != nil { chartEdbMappingIdList = append(chartEdbMappingIdList, strconv.Itoa(tmpChartEdbMapping.ChartEdbMappingId)) //tmpChartEdbMapping.ModifyTime = time.Now() //tmpChartEdbMapping.MaxData = v.MaxData //tmpChartEdbMapping.MinData = v.MinData //tmpChartEdbMapping.IsOrder = v.IsOrder //tmpChartEdbMapping.IsAxis = v.IsAxis //tmpChartEdbMapping.EdbInfoType = v.EdbInfoType //tmpChartEdbMapping.LeadValue = v.LeadValue //tmpChartEdbMapping.LeadUnit = v.LeadUnit //tmpChartEdbMapping.ChartStyle = v.ChartStyle //tmpChartEdbMapping.ChartColor = v.ChartColor //tmpChartEdbMapping.PredictChartColor = v.PredictChartColor //tmpChartEdbMapping.ChartWidth = v.ChartWidth //_, err = to.Update(tmpChartEdbMapping, "ModifyTime", "MaxData", "MinData", "IsOrder", "IsAxis", "EdbInfoType", "LeadValue", "LeadUnit", "ChartStyle", "ChartColor", "PredictChartColor", "ChartWidth") //if err != nil { // fmt.Println("chart_edb_mapping Err:" + err.Error()) // return err //} } else { mapItem := new(ChartEdbMapping) mapItem.ChartInfoId = req.ChartInfoId mapItem.EdbInfoId = v.EdbInfoId mapItem.CreateTime = time.Now() mapItem.ModifyTime = time.Now() timestamp := strconv.FormatInt(time.Now().UnixNano(), 10) mapItem.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp + "_" + strconv.Itoa(v.EdbInfoId)) //mapItem.MaxData = v.MaxData //mapItem.MinData = v.MinData mapItem.IsOrder = true mapItem.IsAxis = 1 mapItem.EdbInfoType = 1 //mapItem.LeadValue = v.LeadValue //mapItem.LeadUnit = v.LeadUnit //mapItem.ChartStyle = v.ChartStyle //mapItem.ChartColor = v.ChartColor //mapItem.PredictChartColor = v.PredictChartColor //mapItem.ChartWidth = v.ChartWidth mapItem.Source = utils.CHART_SOURCE_CORRELATION err = to.Create(mapItem).Error if err != nil { fmt.Println("AddChartEdbMapping Err:" + err.Error()) return err } chartEdbMappingIdList = append(chartEdbMappingIdList, strconv.Itoa(mapItem.ChartEdbMappingId)) } } if len(chartEdbMappingIdList) > 0 { chartEdbMappingIdStr := strings.Join(chartEdbMappingIdList, ",") dsql := `DELETE FROM chart_edb_mapping WHERE chart_info_id=? AND chart_edb_mapping_id NOT IN(` + chartEdbMappingIdStr + `)` err = to.Exec(dsql, req.ChartInfoId).Error if err != nil { fmt.Println("delete err:" + err.Error()) return err } } // 相关性图表 err = to.Select(correlationUpdateCols).Updates(correlationChart).Error return } // PreviewSectionScatterChartReq 预览截面散点图的请求 type PreviewSectionScatterChartReq struct { ChartName string `description:"图表名称"` ChartClassifyId int `description:"分类id"` ExtraConfig string `description:"图表额外配置信息,json字符串"` } // 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:"系列名(英文名)"` IsNameDefault bool `description:"是否使用默认的系列名 false修改过 true默认 影响自动更新"` 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:"系列名(英文)"` IsNameDefault bool `description:"是否使用默认的系列名 false修改过 true默认 影响自动更新"` 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 } // RollingCorrelationChartDataResp 滚动相关性图 type RollingCorrelationChartDataResp struct { XDateTimeValue []string `description:"滚动相关性图的x轴数据，日期数据"` YDataList []YData `description:"滚动相关性图的y轴数据"` } // GetChartInfoAll 用于分类展示 // @param source int 1：ETA图库；2：商品价格曲线 func GetChartInfoAllByClassifyId(source, classifyId int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT chart_info_id,chart_classify_id,chart_name AS chart_classify_name,chart_name_en AS chart_classify_name_en, unique_code,sys_user_id,sys_user_real_name,date_type,start_date,end_date,chart_type,calendar,season_start_date,season_end_date,is_join_permission FROM chart_info WHERE chart_classify_id = ? AND source = ? ORDER BY sort asc,create_time DESC ` err = o.Raw(sql, classifyId, source).Scan(&items).Error return } // ProfitFutureGoodChartResp 商品利润图 type ProfitFutureGoodChartResp struct { XDataList []XData YDataList []YData ProfitName string `description:"利润的名称"` ProfitNameEn string `description:"利润的英文名称"` } func FIXChartClassifyId(newId, oldId int) (err error) { o := global.DmSQL["data"] sql := ` UPDATE chart_info SET chart_classify_id=? WHERE chart_classify_id=? ` err = o.Exec(sql, newId, oldId).Error return } // GetChartInfoByAdminIdAndClassify 获取所有我创建的图表，用于分类展示 func GetChartInfoByAdminIdAndClassify(sourceList []int, adminId, classifyId int) (items []*ChartClassifyItems, err error) { num := len(sourceList) if num <= 0 { return } o := global.DmSQL["data"] sql := ` SELECT chart_info_id,chart_classify_id,chart_name AS chart_classify_name, unique_code,sys_user_id,sys_user_real_name,date_type,start_date,end_date,chart_type,calendar,season_start_date,season_end_date,is_join_permission FROM chart_info where source in (` + utils.GetOrmInReplace(num) + `) AND sys_user_id = ? AND chart_classify_id = ? ORDER BY sort asc,create_time ASC ` err = o.Raw(sql, sourceList, adminId, classifyId).Scan(&items).Error return } // GetChartInfoAdminList 根据条件获取图表数据了列表 func GetChartInfoAdminList() (items []int, err error) { o := global.DmSQL["data"] sql := ` SELECT sys_user_id FROM chart_info GROUP BY sys_user_id ` err = o.Raw(sql).Scan(&items).Error return } // EditChartInfoExtraConfig 修改 ETA图库 的 图表额外配置 func EditChartInfoExtraConfig(chartId int, extraConfig string) (err error) { o := global.DmSQL["data"] var pars []interface{} pars = append(pars, extraConfig) sql := ` UPDATE chart_info SET modify_time = NOW(), extra_config = ? ` sql += `WHERE chart_info_id = ?` pars = append(pars, chartId) err = o.Exec(sql, pars).Error if err != nil { fmt.Println("UPDATE chart_info Err:", err.Error()) return err } return } // PreviewRadarChartReq 预览雷达图的请求入参 type PreviewRadarChartReq struct { ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` ExtraConfig string `description:"图表额外配置信息,json字符串"` } // 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:"每个指标的值"` } // 截面组合图额外配置 type ChartSectionExtraConf struct { DateConfList []*ChartSectionDateConfItem IsHeap int `description:"是否堆积（1.堆积，0不堆积）"` XDataList []XData `description:"横轴名称设置"` UnitList *ChartSectionCombineUnit `description:"纵轴单位设置"` BaseChartSeriesName string `description:"基准系列名称"` SortType int `description:"排序类型，0默认，1升序，2降序"` } // 截面组合图额外配置 type ChartSectionAllExtraConf struct { ChartSectionExtraConf SeriesList []*ChartSectionSeriesItem } type ChartSectionDateConfItem struct { MoveForward int `description:"前移的期数"` EdbInfoId int `description:"指标ID"` EdbName string `description:"指标名称"` EdbNameEn string `description:"指标名称英文"` EdbInfoType int `description:"指标类型"` Frequency string `description:"频度"` EndDate string `description:"最新日期"` StaticDate string `description:"固定日期"` DateType int `description:"日期类型:0 指标日期，1系统日期, 2固定日期"` DateConfName string `description:"引用日期名称"` // 引用日期名称不能重复 DateConfNameEn string `description:"引用日期英文名称"` DateChange []*ChartSectionDateChange } // 截面组合图引用日期配置 type ChartSectionDateChange struct { Year int Month int Day int Frequency string `description:"频度变换"` FrequencyDay string `description:"频度的固定日期"` ChangeType int `description:"日期变换类型1日期位移，2指定频率"` } // 截面组合图系列配置 type ChartSectionSeriesItem struct { ChartSeriesId int `description:"系列ID"` SeriesName string `description:"系列名称"` //系列名称不可同名 SeriesNameEn string `description:"系列英文名称"` ChartStyle string `description:"图表类型"` ChartColor string `description:"颜色"` ChartWidth int `description:"线条大小"` IsPoint int `description:"是否用数据点展示（0 否，1是）"` IsNumber int `description:"是否用数值展示（0 否，1是）"` IsAxis int `description:"1:左轴,0:右轴"` MaxData float64 `description:"上限"` MinData float64 `description:"下限"` //IsOrder bool `description:"true:正序,false:逆序"` EdbInfoList []*ChartSectionSeriesEdbConf DataList []float64 NoDataEdbIndex []int } type ChartSectionSeriesEdbConf struct { ChartSeriesEdbMappingId int `description:"映射ID"` ChartSeriesId int `description:"系列ID"` //ChartInfoId int `description:"图表ID"` EdbInfoId int `description:"指标id"` DateConf *ChartSectionSeriesDateConfItem EdbName string `description:"中文别名"` EdbNameEn string `description:"英文别名"` EdbInfoType int `description:"指标类型"` Unit string `description:"单位"` UnitEn string `description:"英文单位"` DateConfName string `description:"引用日期名称"` DateConfType int `description:"日期类型，0指标最新日期， 1引用日期"` } type ChartSectionCombineDataResp struct { DateConfList []*ChartSectionDateConfItem IsHeap int `description:"是否堆积（1.堆积，0不堆积）"` XDataList []XData `description:"横轴名称设置"` UnitList *ChartSectionCombineUnit `description:"纵轴单位设置"` BaseChartSeriesName string `description:"基准系列名称"` SortType int `description:"排序类型，0默认，1升序，2降序"` SeriesList []*ChartSectionSeriesItem LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` Right2Min string `description:"图表右侧最小值"` Right2Max string `description:"图表右侧最大值"` } // 系列里的指标日期配置 type ChartSectionSeriesDateConfItem struct { MoveForward int `description:"前移的期数"` DateChange []*ChartSectionDateChange } // PreviewSectionCombineChartReq 预览截面组合图的请求 type PreviewSectionCombineChartReq struct { ChartName string `description:"图表名称"` ChartClassifyId int `description:"分类id"` ExtraConfig string `description:"图表额外配置信息,json字符串"` } type ChartSectionCombineUnit struct { LeftName string `description:"左轴单位"` LeftNameEn string `description:"左轴英文单位"` RightName string `description:"右轴单位"` RightNameEn string `description:"右轴英文单位"` RightTwoName string `description:"右2轴单位"` RightTwoNameEn string `description:"右2轴英文单位"` } // 时序组合图额外配置 type ChartTimeCombineExtraConf struct { IsHeap int `description:"是否堆积（1.堆积，0不堆积）"` } type ChartTimeCombineDataResp struct { IsHeap int `description:"是否堆积（1.堆积，0不堆积）"` } type PreviewSectionCombineDateCalculateResp struct { Date string } type PreviewSectionCombineDateCalculateReq struct { MoveForward int `description:"前移的期数"` EdbInfoId int `description:"指标ID"` Frequency string `description:"频度"` DateConfName string `description:"引用日期名称"` // 引用日期名称不能重复 DateChange []*ChartSectionDateChange DateConfList []*ChartSectionDateConfItem DateType int `description:"日期类型，0指标最新日期， 1引用日期"` } // ChartInfoSourcesFrom 图表来源 type ChartInfoSourcesFrom struct { IsShow bool `description:"是否展示" json:"isShow"` Text string `description:"来源文本" json:"text"` Color string `description:"来源颜色" json:"color"` FontSize int `description:"来源字号" json:"fontSize"` } // UpdateChartInfoSortByClassifyIdV2 根据分类id更新排序 func UpdateChartInfoSortByClassifyIdV2(classifyId, nowSort int, prevChartInfoId int, updateSort string) (err error) { o := global.DmSQL["data"] sql := ` update chart_info set sort = ` + updateSort + ` WHERE chart_classify_id = ?` if prevChartInfoId > 0 { sql += ` AND ( sort > ? or ( chart_info_id > ` + fmt.Sprint(prevChartInfoId) + ` and sort=` + fmt.Sprint(nowSort) + ` )) ` } else { sql += ` AND ( sort > ? )` } err = o.Exec(sql, classifyId, nowSort).Error return } // GetChartInfoMaxSortByClassifyId 获取分类下指标的最大的排序数 func GetChartInfoMaxSortByClassifyId(classifyId int) (sort int, err error) { o := global.DmSQL["data"] sql := `SELECT COALESCE(MAX(sort),0) AS sort FROM chart_info WHERE chart_classify_id = ? ` err = o.Raw(sql, classifyId).Scan(&sort).Error return } // GetChartInfoBySourceAndParentId 根据图表来源及父级ID获取图表 func GetChartInfoBySourceAndParentId(source, parentId, adminId int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT chart_info_id,chart_classify_id,chart_name AS chart_classify_name,chart_name_en AS chart_classify_name_en, unique_code,sys_user_id,sys_user_real_name,date_type,start_date,end_date,chart_type,calendar,season_start_date,season_end_date,source FROM chart_info WHERE source = ? AND chart_classify_id = ?` pars := make([]interface{}, 0) pars = append(pars, source, parentId) if adminId > 0 { sql += ` AND sys_user_id = ?` pars = append(pars, adminId) } sql += ` ORDER BY sort asc,chart_info_id ASC ` err = o.Raw(sql, pars...).Scan(&items).Error return } // PreviewSeasonChartReq 预览季节性图的请求入参 type PreviewSeasonChartReq struct { ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` SeasonExtraConfig SeasonExtraItem `description:"季节图额外配置信息,json字符串"` } // EditChartInfoImageV2 // @Description: 修改图表的缩略图 // @author: Roc // @datetime 2024-07-09 10:52:34 // @param chartInfoId int // @param imageUrl string // @return err error func EditChartInfoImageV2(chartInfoId int, imageUrl string) (err error) { o := global.DmSQL["data"] sql := ` UPDATE chart_info SET chart_image=?, modify_time = NOW() WHERE chart_info_id = ? ` err = o.Exec(sql, imageUrl, chartInfoId).Error if err != nil { fmt.Println("EditChartInfoImageV2 Err:", err.Error()) return err } return } func getThsHfEdbDataListByMongo(source, subSource, edbInfoId int, startDate, endDate string) (list []*EdbDataList, err error) { list = make([]*EdbDataList, 0) mogDataObj := mgo.EdbDataThsHf{} // 构建查询条件 queryConditions := bson.M{ "edb_info_id": edbInfoId, } // 数据日期 dateCondition, err := mgo.BuildDateCondition(startDate, endDate) if err != nil { return } if len(dateCondition) > 0 { queryConditions["data_time"] = dateCondition } // 获取列表数据 tmpDataList, tmpErr := mogDataObj.GetAllDataList(queryConditions, []string{"data_time"}) if tmpErr != nil { err = tmpErr return } for k, v := range tmpDataList { list = append(list, &EdbDataList{ EdbDataId: k + 1, EdbInfoId: v.EdbInfoId, DataTime: v.DataTime.Format(utils.FormatDate), DataTimestamp: v.DataTimestamp, Value: v.Value, }) } return } func getThsHfEdbDataListMinAndMaxByMongo(source, subSource, edbInfoId int, startDate, endDate string) (minData, maxData float64, err error) { mogDataObj := mgo.EdbDataThsHf{} // 构建查询条件 queryConditions := bson.M{ "edb_info_id": edbInfoId, } // 日期 dateCondition, err := mgo.BuildDateCondition(startDate, endDate) if err != nil { return } if len(dateCondition) > 0 { queryConditions["data_time"] = dateCondition } pipeline := []bson.M{ {"$match": queryConditions}, {"$group": bson.M{ "_id": nil, "min_value": bson.M{"$min": "$value"}, "max_value": bson.M{"$max": "$value"}, }}, {"$project": bson.M{"_id": 0}}, // 可选，如果不需要_id字段 } result, err := mogDataObj.GetEdbInfoMaxAndMinInfo(pipeline) if err != nil { return } minData = result.MinValue maxData = result.MaxValue return } // UpdateClassifyIdByChartInfoIdList // @Description: 根据图表id列表更新所属分类id // @author: Roc // @datetime 2024-12-10 13:18:52 // @param chartInfoIdList int // @param chartClassifyId int // @return err error func UpdateClassifyIdByChartInfoIdList(chartInfoIdList []int, chartClassifyId int) (err error) { if len(chartInfoIdList) <= 0 { return } sql := ` UPDATE chart_info SET chart_classify_id = ? WHERE chart_info_id in (?) ` err = global.DmSQL["data"].Exec(sql, chartClassifyId, chartInfoIdList).Error return } // UpdateChartInfoPublicSortByClassifyId // @Description: 根据公共分类id更新图表公共排序 // @param classifyId // @param nowSort // @param prevChartInfoId // @param updateSort // @param source // @return err func UpdateChartInfoPublicSortByClassifyId(classifyId, nowSort int, prevChartInfoId int, updateSort string, source int) (err error) { sql := ` update chart_info set sort = ` + updateSort + ` WHERE chart_public_classify_id=? AND public_status = 3 AND source = ? ` if prevChartInfoId > 0 { sql += ` AND ( public_sort > ? or ( chart_info_id > ` + fmt.Sprint(prevChartInfoId) + ` and public_sort=` + fmt.Sprint(nowSort) + ` )) ` } else { sql += ` AND ( public_sort > ? )` } err = global.DmSQL["data"].Exec(sql, classifyId, nowSort, source).Error return } // GetFirstChartInfoByPublicClassifyId // @Description: 获取当前公共分类下，且排序数相同 的排序第一条的图表数据 // @param classifyId // @return item // @return err func GetFirstChartInfoByPublicClassifyId(classifyId int) (item *ChartInfo, err error) { sql := ` SELECT * FROM chart_info WHERE chart_public_classify_id=? AND public_status = 3 order by public_sort asc,chart_info_id asc limit 1` err = global.DmSQL["data"].Raw(sql, classifyId).First(&item).Error return }