package data_manage import ( "errors" "fmt" "hongze/hongze_chart_lib/utils" "github.com/rdlucklib/rdluck_tools/orm" "github.com/rdlucklib/rdluck_tools/paging" "strconv" "strings" "time" ) type ChartInfo struct { ChartInfoId int `orm:"column(chart_info_id);pk"` ChartName 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:季节性图"` Calendar string `description:"公历/农历"` SeasonStartDate string `description:"季节性图开始日期"` SeasonEndDate string `description:"季节性图开始日期"` ChartImage string `description:"图表图片"` Sort int `description:"排序字段,数字越小越排前面"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` } func AddChartInfo(item *ChartInfo) (lastId int64, err error) { o := orm.NewOrm() o.Using("data") lastId, err = o.Insert(item) return } type ChartInfoItem struct { ChartInfoId int `description:"图表id"` ChartName string `description:"图表名称"` ChartClassifyId int `description:"图表分类"` } //用于分类展示 func GetChartInfoAll() (items []*ChartClassifyItems, err error) { o := orm.NewOrm() o.Using("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 ORDER BY sort asc,create_time ASC ` _, err = o.Raw(sql).QueryRows(&items) return } //用于分类展示 func GetChartInfoAllList() (items []*ChartInfo, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info ORDER BY create_time ASC ` _, err = o.Raw(sql).QueryRows(&items) 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:新数据"` } func GetChartInfoById(chartInfoId int) (item *ChartInfo, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info WHERE chart_info_id=? ` err = o.Raw(sql, chartInfoId).QueryRow(&item) return } func GetChartInfoViewById(chartInfoId int) (item *ChartInfoView, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info WHERE chart_info_id=? ` err = o.Raw(sql, chartInfoId).QueryRow(&item) 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:"公历/农历"` SeasonStartDate string `description:"季节性图开始日期"` SeasonEndDate string `description:"季节性图开始日期"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` } 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:"颜色"` ChartWidth float64 `description:"线条大小"` } func DeleteChartInfoAndData(chartInfoId int) (err error) { o := orm.NewOrm() o.Using("data") defer func() { if err != nil { o.Rollback() } else { o.Commit() } }() sql := ` DELETE FROM chart_info WHERE chart_info_id=? ` _, err = o.Raw(sql, chartInfoId).Exec() if err != nil { return } sql = ` DELETE FROM chart_edb_mapping WHERE chart_info_id=? ` _, err = o.Raw(sql, chartInfoId).Exec() return } func GetChartInfoCountByCondition(condition string, pars []interface{}) (count int, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT COUNT(1) AS count FROM chart_info WHERE 1=1 ` if condition != "" { sql += condition } err = o.Raw(sql, pars).QueryRow(&count) return } type EditChartInfoReq struct { ChartInfoId int `description:"图表ID"` ChartName string `description:"图表名称"` ChartClassifyId int `description:"分类id"` ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` ChartType int `description:"生成样式:1:曲线图,2:季节性图"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今,4:年初至今,5:自定义时间,6:起始日期至今"` StartDate string `description:"自定义开始日期"` EndDate string `description:"自定义结束日期"` Calendar string `description:"公历/农历"` SeasonStartDate string `description:"季节性图开始日期"` SeasonEndDate string `description:"季节性图开始日期"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` } func ModifyChartInfo(item *EditChartInfoReq) (err error) { o := orm.NewOrm() o.Using("data") sql := ` UPDATE chart_info SET chart_name =?, chart_classify_id = ?, is_set_name=1, modify_time = NOW() WHERE chart_info_id = ?` _, err = o.Raw(sql, item.ChartName, item.ChartClassifyId, item.ChartInfoId).Exec() return } type ChartInfoList struct { ChartInfoId int `orm:"column(chart_info_id);pk"` ChartName string `description:"来源名称"` ChartChartClassifyId int `description:"图表分类id"` SysUserId int SysUserRealName string UniqueCode string `description:"图表唯一编码"` CreateTime time.Time ModifyTime time.Time EdbInfoList []*EdbInfoList } type ChartInfoListResp struct { Paging *paging.PagingItem Item *ChartInfoList } func GetChartInfoByCondition(condition string, pars []interface{}) (item *ChartInfo, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info WHERE 1=1 ` if condition != "" { sql += condition } err = o.Raw(sql, pars).QueryRow(&item) return } func GetChartInfoByNewest() (item *EdbInfoList, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info WHERE 1=1 ORDER BY modify_time DESC LIMIT 1 ` err = o.Raw(sql).QueryRow(&item) return } func ModifyChartInfoModifyTime(chartInfoId int64) (err error) { o := orm.NewOrm() o.Using("data") sql := ` UPDATE chart_info SET modify_time = NOW() WHERE chart_info_id = ? ` _, err = o.Raw(sql, chartInfoId).Exec() 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 := orm.NewOrm() o.Using("data") sql := ` UPDATE chart_info SET chart_classify_id = ? WHERE chart_info_id = ?` _, err = o.Raw(sql, classifyId, chartInfoId).Exec() 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:"数据值"` } func GetEdbDataList(source, endInfoId int, startDate, endDate string) (list []*EdbDataList, err error) { tableName := GetEdbDataTableName(source) if tableName == "" { err = errors.New("无效的渠道:" + strconv.Itoa(source)) list = make([]*EdbDataList, 0) return list, err } var pars []interface{} 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 := orm.NewOrm() o.Using("data") _, err = o.Raw(sql, endInfoId, pars).QueryRows(&list) return } 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 := orm.NewOrm() o.Using("data") _, err = o.Raw(sql, endInfoId, startDate, endDate).QueryRows(&list) return } func GetEdbDataListMinAndMax(source, endInfoId int, startDate, endDate string) (min_data, max_data float64, err error) { tableName := GetEdbDataTableName(source) 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 := orm.NewOrm() o.Using("data") if condition != "" { sql += condition } err = o.Raw(sql, endInfoId, pars).QueryRow(&min_data, &max_data) return } type ChartEdbInfoMapping struct { EdbInfoId int `description:"指标id"` SourceName string `description:"来源名称"` Source int `description:"来源id"` EdbCode string `description:"指标编码"` EdbName string `description:"指标名称"` EdbType int `description:"指标类型:1:基础指标,2:计算指标"` Frequency string `description:"频率"` Unit string `description:"单位"` StartDate string `description:"起始日期"` EndDate string `description:"终止日期"` ModifyTime string `description:"指标最后更新时间"` ChartEdbMappingId int `description:"图表指标id"` ChartInfoId int `description:"图表id"` MaxData float64 `description:"上限"` MinData float64 `description:"下限"` IsOrder bool `description:"true:正序,false:逆序"` IsAxis int `description:"1:左轴,0:右轴"` EdbInfoType int `description:"1:标准指标,0:领先指标"` LeadValue int `description:"领先值"` LeadUnit string `description:"领先单位"` ChartStyle string `description:"图表类型"` ChartColor string `description:"颜色"` ChartWidth float64 `description:"线条大小"` LatestDate string `description:"数据最新日期"` LatestValue float64 `description:"数据最新值"` UniqueCode string `description:"指标唯一编码"` MinValue float64 `json:"-" description:"最小值"` MaxValue float64 `json:"-" description:"最大值"` DataList interface{} } type QuarterData struct { Year int DataList []*EdbDataList } type ChartEdbInfoDetailResp struct { EdbInfo *ChartEdbInfoMapping } type ChartInfoDetailResp struct { ChartInfo *ChartInfoView EdbInfoList []*ChartEdbInfoMapping } func ModifyChartInfoAndMapping(edbInfoIdStr string, req *SaveChartInfoReq, chartType int) (err error) { o := orm.NewOrm() o.Using("data") o.Begin() defer func() { if err != nil { o.Rollback() } else { o.Commit() } }() if chartType == 1 { sql := ` UPDATE chart_info SET edb_info_ids=?, modify_time = NOW(), date_type=?, start_date=?, end_date=?, left_min=?, left_max=?, right_min=?, right_max=? WHERE chart_info_id = ?` _, err = o.Raw(sql, edbInfoIdStr, req.DateType, req.StartDate, req.EndDate, req.LeftMin, req.LeftMax, req.RightMin, req.RightMax, req.ChartInfoId).Exec() 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=?, season_start_date=?, season_end_date=?, left_min=?, left_max=?, right_min=?, right_max=? WHERE chart_info_id = ?` _, err = o.Raw(sql, edbInfoIdStr, req.Calendar, req.SeasonStartDate, req.SeasonEndDate, req.LeftMin, req.LeftMax, req.RightMin, req.RightMax, req.ChartInfoId).Exec() 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 = o.Raw(csql, req.ChartInfoId, v.EdbInfoId).QueryRow(&count) 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 = ?, chart_width = ? WHERE chart_info_id =? AND edb_info_id=? ` _, err = o.Raw(msql, v.MaxData, v.MinData, v.IsOrder, v.IsAxis, v.EdbInfoType, v.LeadValue, v.LeadUnit, v.ChartStyle, v.ChartColor, v.ChartWidth, req.ChartInfoId, v.EdbInfoId).Exec() 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.ChartWidth = v.ChartWidth mapList = append(mapList, mapItem) } } if len(mapList) > 0 { _, err = o.InsertMulti(1, mapList) 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 = o.Raw(dsql, req.ChartInfoId).Exec() if err != nil { fmt.Println("delete err:" + err.Error()) return err } } } return } func EditChartInfoAndMapping(req *EditChartInfoReq, edbInfoIdStr string, calendar string, dateType int) (err error) { o := orm.NewOrm() o.Using("data") o.Begin() defer func() { if err != nil { o.Rollback() } else { o.Commit() } }() var pars []interface{} pars = append(pars, req.ChartName) pars = append(pars, edbInfoIdStr) pars = append(pars, req.ChartType) pars = append(pars, req.ChartClassifyId) sql := ` UPDATE chart_info SET chart_name =?, edb_info_ids=?, chart_type=?, chart_classify_id = ?, modify_time = NOW() ` 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.SeasonStartDate) sql += `,season_end_date = ? ` pars = append(pars, req.SeasonEndDate) 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 = o.Raw(sql, pars).Exec() 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 = o.Raw(csql, req.ChartInfoId, v.EdbInfoId).QueryRow(&count) 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 = ?, chart_width = ? WHERE chart_info_id =? AND edb_info_id=? ` _, err = o.Raw(msql, v.MaxData, v.MinData, v.IsOrder, v.IsAxis, v.EdbInfoType, v.LeadValue, v.LeadUnit, v.ChartStyle, v.ChartColor, v.ChartWidth, req.ChartInfoId, v.EdbInfoId).Exec() 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.ChartWidth = v.ChartWidth mapList = append(mapList, mapItem) } } if len(mapList) > 0 { _, err = o.InsertMulti(1, mapList) if err != nil { fmt.Println("AddChartEdbMapping Err:" + err.Error()) return err } } if len(edbInfoIdArr) > 0 { edbInfoIdStr := strings.Join(edbInfoIdArr, ",") dsql := `DELETE FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id NOT IN(` + edbInfoIdStr + `)` _, err = o.Raw(dsql, req.ChartInfoId).Exec() if err != nil { fmt.Println("delete err:" + err.Error()) return err } } return } func ModifyEdbDatadTimestamp(source, edbDataId int, dataTimestamp int64) (err error) { tableName := GetEdbDataTableName(source) sql := `UPDATE %s SET data_timestamp=? WHERE edb_data_id=? ` sql = fmt.Sprintf(sql, tableName) o := orm.NewOrm() o.Using("data") _, err = o.Raw(sql, dataTimestamp, edbDataId).Exec() return } type AddChartInfoReq struct { ChartEdbInfoList []*ChartSaveItem `description:"指标及配置信息"` ChartClassifyId int `description:"分类id"` ChartName string `description:"图表名称"` ChartType int `description:"生成样式:1:曲线图,2:季节性图"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今,4:年初至今,5:自定义时间,6:起始日期至今"` StartDate string `description:"自定义开始日期"` EndDate string `description:"自定义结束日期"` Calendar string `description:"公历/农历"` SeasonStartDate string `description:"季节性图开始日期"` SeasonEndDate string `description:"季节性图开始日期"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `description:"图表右侧最大值"` } type AddChartInfoResp struct { ChartInfoId int `description:"图表id"` UniqueCode string `description:"图表唯一编码"` ChartType int `description:"生成样式:1:曲线图,2:季节性图"` } //判断图表指标是否已经存在 func ChartInfoExist(condition, edbInfoIdStr string) (count int, err error) { sql := `SELECT COUNT(1) AS count FROM ( SELECT GROUP_CONCAT(edb_info_id ORDER BY edb_info_id ASC SEPARATOR ',') AS edb_info_id_str FROM chart_edb_mapping AS a WHERE 1=1` if condition != "" { sql += condition } sql += ` GROUP BY chart_info_id )AS t WHERE t.edb_info_id_str=? GROUP BY t.edb_info_id_str ` o := orm.NewOrm() o.Using("data") err = o.Raw(sql, edbInfoIdStr).QueryRow(&count) return } func ChartInfoSearchByKeyWord(KeyWord string) (searchList []*ChartInfo, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info ` if KeyWord != "" { sql += `WHERE chart_name LIKE '%` + KeyWord + `%' ` } sql += ` ORDER BY create_time DESC ` if KeyWord == "" { sql += ` LIMIT 100 ` } _, err = o.Raw(sql).QueryRows(&searchList) return } func GetNextChartInfo(classifyId int) (item *ChartInfo, err error) { o := orm.NewOrm() o.Using("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).QueryRow(&item) return } func GetChartInfoRefreshData(chartInfoId int) (baseEdbInfoArr, calculateInfoArr []*EdbInfo, err error) { o := orm.NewOrm() o.Using("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).QueryRows(&edbInfoList) 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 && err.Error() != utils.ErrNoRow() { 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 := orm.NewOrm() o.Using("data") sql := ` UPDATE chart_info SET chart_image=?, modify_time = NOW() WHERE chart_info_id = ? ` _, err = o.Raw(sql, req.ImageUrl, req.ChartInfoId).Exec() if err != nil { fmt.Println("EditChartInfoImage Err:", err.Error()) return err } return } type ChartInfoDetailFromUniqueCodeResp struct { ChartInfo *ChartInfoView Status bool `description:"true:图表存在,false:图表不存在"` EdbInfoList []*ChartEdbInfoMapping } func GetChartInfoByUniqueCode(uniqueCode string) (item *ChartInfoView, err error) { o := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info WHERE unique_code=? ` err = o.Raw(sql, uniqueCode).QueryRow(&item) return } // GetFirstChartInfoByClassifyId 获取当前分类下,且排序数相同 的排序第一条的数据 func GetFirstChartInfoByClassifyId(classifyId int) (item *ChartInfo, err error) { o := orm.NewOrm() o.Using("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).QueryRow(&item) return } // UpdateChartInfoSortByClassifyId 根据图表id更新排序 func UpdateChartInfoSortByClassifyId(classifyId, nowSort, prevChartInfoId int, updateSort string) (err error) { o := orm.NewOrm() o.Using("data") sql := ` update chart_info set sort = ` + updateSort + ` WHERE chart_classify_id=? and sort > ? ` if prevChartInfoId > 0 { sql += ` or (chart_info_id > ` + fmt.Sprint(prevChartInfoId) + ` and sort = ` + fmt.Sprint(nowSort) + `)` } _, err = o.Raw(sql, classifyId, nowSort).Exec() return } // Update 更新图表基础信息 func (chartInfo *ChartInfo) Update(cols []string) (err error) { o := orm.NewOrm() o.Using("data") _, err = o.Update(chartInfo, cols...) return } type ChartInfoView struct { ChartInfoId int `orm:"column(chart_info_id);pk"` ChartName string `description:"来源名称"` ChartClassifyId int `description:"图表分类id"` ChartClassifyName string `description:"图表名称"` 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:季节性图"` 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 EdbEndDate string `description:"指标最新更新日期"` LeftMin string `description:"图表左侧最小值"` LeftMax string `description:"图表左侧最大值"` RightMin string `description:"图表右侧最小值"` RightMax string `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 := orm.NewOrm() o.Using("data") sql := ` SELECT * FROM chart_info WHERE chart_classify_id = ? and chart_name=? ` err = o.Raw(sql, classifyId, chartName).QueryRow(&item) return }