eta_hub/controllers/business_chart.go

package controllers

import (
	"eta/eta_hub/models"
	"eta/eta_hub/models/data_manage"
	"eta/eta_hub/services"
	"eta/eta_hub/services/data"
	"eta/eta_hub/utils"

	"github.com/rdlucklib/rdluck_tools/paging"
)

type BusinessChartController struct {
	BaseAuthController
}

// List
// @Title 图表列表
// @Description 图表列表
// @Param   BusinessCode  query  string  true  "商家编码"
// @Success 200 {object} data_manage.ChartInfoItem
// @router /list [get]
func (this *BusinessChartController) List() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	businessCode := this.GetString("BusinessCode")
	pageSize, _ := this.GetInt("PageSize")
	currentIndex, _ := this.GetInt("CurrentIndex")


	if businessCode == "" {
		br.Msg = "商家编码不允许为空"
		return
	}
	if currentIndex <= 0 {
		currentIndex = 1
	}
	if pageSize <= 0 {
		pageSize = 100
	}
	startSize := utils.StartIndex(currentIndex, pageSize)

	chartOb := new(data_manage.EtaBusinessChartPermission)
		
	total, err := chartOb.GetChartListByBusinessCodeTotal(businessCode)
	if err != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "ChartList GetItemTotal err: " + err.Error()
		return
	}
	page := paging.GetPaging(currentIndex, pageSize, int(total))
	resp := new(data_manage.BusinessChartInfoResp)
	if total == 0 {
		resp.List = make([]*data_manage.BusinessChartInfoView, 0)
		resp.Paging = page

		br.Data = resp
		br.Ret = 200
		br.Msg = "获取成功"
		return
	}

	chartList, err := chartOb.GetChartListByBusinessCode(businessCode, startSize, pageSize)
	if err != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "ChartList GetChartListByBusinessCode err: " + err.Error()
		return
	}
	chartInfoIds := make([]int, 0)
	for _, v := range chartList {
		chartInfoIds = append(chartInfoIds, v.ChartInfoId)
	}
	descriptionList, err := chartOb.GetChartDescriptionListByChartInfoIds(chartInfoIds)
	if err != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "ChartList GetChartDescriptionListByChartInfoIds err: " + err.Error()
		return
	}
	descriptionMap := make(map[int][]string)
	for _, v := range descriptionList {
		descriptionMap[v.ChartInfoId] = append(descriptionMap[v.ChartInfoId], v.Description)
	}
	edbMappingList, err := data_manage.GetChartEdbMappingList(chartInfoIds)
	if err != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "ChartList GetChartEdbMappingList err: " + err.Error()
		return
	}
	edbInfoIds := make([]int, 0)
	chartMappingMap := make(map[int][]*data_manage.BusinessChartEdbListItem)
	for _, v := range edbMappingList {
		edbInfoIds = append(edbInfoIds, v.EdbInfoId)
	}
	edbInfoList, e := data_manage.GetEdbInfoByIdList(edbInfoIds)
	if e != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "ChartList GetItemsByIdList err: " + e.Error()
		return
	}
	edbInfoMap := make(map[int]*data_manage.BusinessChartEdbListItem)
	// 根据chartInfoId 将edbInfoList 转换为chartMappingMap
	for _, v := range edbInfoList {
		endValue := v.EndValue
		if v.EdbInfoType == 0 {
			endValue = v.LatestValue
		}
		item := &data_manage.BusinessChartEdbListItem{
			EdbInfoId: v.EdbInfoId,
			EdbName: v.EdbName,
			EdbCode: v.EdbCode,
			EdbInfoType: v.EdbInfoType,
			Frequency: v.Frequency,
			Unit: v.Unit,
			EndDate: v.EndDate,
			EndValue: endValue,
			LatestDate: v.LatestDate,
			LatestValue: v.LatestValue,
			UniqueCode: v.UniqueCode,
		}
		edbInfoMap[v.EdbInfoId] = item
	}
	for _, v := range edbMappingList {
		item := edbInfoMap[v.EdbInfoId]
		chartMappingMap[v.ChartInfoId] = append(chartMappingMap[v.ChartInfoId], item)
	}

	for _, v := range chartList {
		v.EdbInfoList = chartMappingMap[v.ChartInfoId]
		description, ok := descriptionMap[v.ChartInfoId]
		if !ok {
			v.Description = make([]string, 0)
		} else {
			v.Description = description
		}
	}

	resp.List = chartList
	resp.Paging = page

	br.Data = resp
	br.Ret = 200
	br.Msg = "获取成功"
}

// ChartEdbDataList
// @Title 图表指标数据列表
// @Description 图表指标数据列表
// @Param   EdbCode   query   string  true       "指标编码"
// @Success 200 {object} data_manage.EdbInfoDataResp
// @router /edb/data [get]
func (this *BusinessChartController) ChartEdbDataList() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	edbInfoCode := this.GetString("EdbCode")
	if edbInfoCode == "" {
		br.Msg = "参数有误"
		return
	}
	edbOb := new(data_manage.EdbInfo)
	edbInfo, e := edbOb.GetItemByEdbCode(edbInfoCode)
	if e != nil {
		if e.Error() == utils.ErrNoRow() {
			br.Msg = "指标不存在"
			return
		}
		br.Msg = "获取失败"
		br.ErrMsg = "ChartEdbDataList GetItemByEdbCode err: " + e.Error()
		return
	}

	edbDataList, e := services.GetEdbInfoDataList(edbInfo)
	if e != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "ChartEdbDataList GetEdbInfoDataList err: " + e.Error()
		return
	}

	br.Data = edbDataList
	br.Ret = 200
	br.Msg = "获取成功"

}

// TraceEdbInfo
// @Title 指标溯源接口
// @Description 指标溯源接口
// @Param   EdbInfoId   query   int  true       "指标id"
// @Success 200 {object} data_manage.ChartListResp
// @router /edb/trace [get]
func (this *BusinessChartController) TraceEdbInfo() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	edbInfoId, _ := this.GetInt("EdbInfoId")
	edbInfoCode := this.GetString("EdbCode")
	if edbInfoId <= 0 && edbInfoCode == "" {
		br.Msg = "参数错误"
		br.ErrMsg = "参数错误"
		return
	}

	resp, err := data.TraceEdbInfoByEdbInfoId(edbInfoId, edbInfoCode)
	if err != nil {
		br.Msg = "获取失败"
		br.Msg = "获取失败,Err:" + err.Error()
		return
	}
	br.Ret = 200
	br.Data = resp
	br.Msg = "刷新成功"
}