eta_api/controllers/data_manage/cross_variety/chart_info.go

package cross_variety

import (
	"encoding/json"
	"eta/eta_api/controllers"
	"eta/eta_api/models"
	"eta/eta_api/models/data_manage"
	"eta/eta_api/models/data_manage/cross_variety/request"
	"eta/eta_api/services/data"
	"eta/eta_api/services/data/cross_variety"
	"eta/eta_api/utils"
	"github.com/rdlucklib/rdluck_tools/paging"
	"strconv"
	"strings"
	"time"
)

// ChartInfoController
// @Description: 跨品种分析图表
type ChartInfoController struct {
	controllers.BaseAuthController
}

// List
// @Title 跨品种分析图表列表接口
// @Description 跨品种分析图表列表接口
// @Param   PageSize   query   int  true       "每页数据条数"
// @Param   CurrentIndex   query   int  true       "当前页页码，从1开始"
// @Param   ChartClassifyId   query   int  true       "分类id"
// @Param   Keyword   query   string  true       "搜索关键词"
// @Param   IsShowMe   query   bool  true       "是否只看我的，true、false"
// @Param   Source   query   int  true       "图表类型，10：跨品种分析
// @Success 200 {object} data_manage.ChartListResp
// @router /chart_info/list [get]
func (c *ChartInfoController) List() {
	br := new(models.BaseResponse).Init()
	defer func() {
		c.Data["json"] = br
		c.ServeJSON()
	}()
	sysUser := c.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}

	chartClassifyId, _ := c.GetInt("ChartClassifyId")

	pageSize, _ := c.GetInt("PageSize")
	currentIndex, _ := c.GetInt("CurrentIndex")
	keyword := c.GetString("KeyWord")

	var total int
	page := paging.GetPaging(currentIndex, pageSize, total)

	var startSize int
	if pageSize <= 0 {
		pageSize = utils.PageSize20
	}
	if currentIndex <= 0 {
		currentIndex = 1
	}
	startSize = paging.StartIndex(currentIndex, pageSize)

	source, _ := c.GetInt("Source")
	if source <= 0 {
		source = utils.CHART_SOURCE_CROSS_HEDGING
	}

	var condition string
	var pars []interface{}

	// 普通图表
	condition += ` AND source = ? `
	pars = append(pars, source)

	if chartClassifyId > 0 {
		chartClassifyId, err := data_manage.GetChartClassify(chartClassifyId)
		if err != nil && err.Error() != utils.ErrNoRow() {
			br.Msg = "获取图表信息失败"
			br.ErrMsg = "获取信息失败,GetChartClassify,Err:" + err.Error()
			return
		}
		condition += " AND chart_classify_id IN(" + chartClassifyId + ") "
	}
	if keyword != "" {
		condition += ` AND  ( chart_name LIKE '%` + keyword + `%' )`
	}

	//只看我的
	isShowMe, _ := c.GetBool("IsShowMe")
	if isShowMe {
		condition += ` AND sys_user_id = ? `
		pars = append(pars, sysUser.AdminId)
	}

	// 获取当前账号的不可见指标
	noPermissionChartIdList := make([]int, 0)
	{
		obj := data_manage.EdbInfoNoPermissionAdmin{}
		confList, err := obj.GetAllChartListByAdminId(c.SysUser.AdminId)
		if err != nil && err.Error() != utils.ErrNoRow() {
			br.Msg = "获取失败"
			br.ErrMsg = "获取不可见指标配置数据失败，Err:" + err.Error()
			return
		}
		for _, v := range confList {
			noPermissionChartIdList = append(noPermissionChartIdList, v.ChartInfoId)
		}
	}

	lenNoPermissionChartIdList := len(noPermissionChartIdList)
	if lenNoPermissionChartIdList > 0 {
		condition += ` AND chart_info_id not in (` + utils.GetOrmInReplace(lenNoPermissionChartIdList) + `) `
		pars = append(pars, noPermissionChartIdList)
	}

	//获取图表信息
	list, err := data_manage.GetChartListByCondition(condition, pars, startSize, pageSize)
	if err != nil && err.Error() != utils.ErrNoRow() {
		br.Success = true
		br.Msg = "获取图表信息失败"
		br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
		return
	}

	myChartList, err := data_manage.GetMyChartListByAdminId(sysUser.AdminId)
	if err != nil && err.Error() != utils.ErrNoRow() {
		br.Msg = "获取图表信息失败"
		br.ErrMsg = "获取我的图表信息失败,Err:" + err.Error()
		return
	}
	myChartMap := make(map[int]*data_manage.MyChartView)
	for _, v := range myChartList {
		myChartMap[v.ChartInfoId] = v
	}
	listLen := len(list)
	chartEdbMap := make(map[int][]*data_manage.ChartEdbInfoMapping)
	if listLen > 0 {
		chartInfoIds := ""
		for _, v := range list {
			chartInfoIds += strconv.Itoa(v.ChartInfoId) + ","
		}
		if chartInfoIds != "" {
			chartInfoIds = strings.Trim(chartInfoIds, ",")
			//判断是否需要展示英文标识
			edbList, e := data_manage.GetChartEdbMappingListByChartInfoIds(chartInfoIds)
			if e != nil {
				br.Msg = "获取失败"
				br.ErrMsg = "获取图表，指标信息失败，Err:" + e.Error()
				return
			}
			for _, v := range edbList {
				chartEdbMap[v.ChartInfoId] = append(chartEdbMap[v.ChartInfoId], v)
			}
		}
	}
	for i := 0; i < listLen; i++ {
		//判断是否需要展示英文标识
		if edbTmpList, ok := chartEdbMap[list[i].ChartInfoId]; ok {
			list[i].IsEnChart = data.CheckIsEnChart(list[i].ChartNameEn, edbTmpList, list[i].Source, list[i].ChartType)
		}

		if existItem, ok := myChartMap[list[i].ChartInfoId]; ok {
			list[i].IsAdd = true
			list[i].MyChartId = existItem.MyChartId
			list[i].MyChartClassifyId = existItem.MyChartClassifyId
		}
	}

	resp := new(data_manage.ChartListResp)
	if list == nil || len(list) <= 0 || (err != nil && err.Error() == utils.ErrNoRow()) {
		items := make([]*data_manage.ChartInfoView, 0)
		resp.Paging = page
		resp.List = items
		br.Ret = 200
		br.Success = true
		br.Msg = "获取成功"
		return
	}

	dataCount, err := data_manage.GetChartListCountByCondition(condition, pars)
	if err != nil && err.Error() != utils.ErrNoRow() {
		br.Msg = "获取指标信息失败"
		br.ErrMsg = "获取指标数据总数失败,Err:" + err.Error()
		return
	}
	page = paging.GetPaging(currentIndex, pageSize, dataCount)
	resp.Paging = page
	resp.List = list
	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
	br.Data = resp
}

// Preview
// @Title 跨品种分析图表-预览数据
// @Description 跨品种分析图表-获取预览数据
// // @Param	request	body request.ChartConfigReq true "type json string"
// @Success 200 {object} data_manage.ChartEdbInfoDetailResp
// @router /chart_info/preview [post]
func (c *ChartInfoController) Preview() {
	br := new(models.BaseResponse).Init()
	defer func() {
		c.Data["json"] = br
		c.ServeJSON()
	}()
	sysUser := c.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}
	var req request.ChartConfigReq
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &req); err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}

	if req.TagX <= 0 {
		br.Msg = "请选择X轴坐标的标签"
		br.IsSendEmail = false
		return
	}

	if req.TagY <= 0 {
		br.Msg = "请选择Y轴坐标的标签"
		br.IsSendEmail = false
		return
	}

	if req.CalculateValue <= 0 {
		br.Msg = "请设置时间长度"
		br.IsSendEmail = false
		return
	}
	if req.CalculateUnit == `` {
		br.Msg = "请设置时间频度"
		br.IsSendEmail = false
		return
	}

	// 品种配置
	varietyListList := len(req.VarietyList)
	if varietyListList < 0 {
		br.Msg = "请选择品种"
		br.IsSendEmail = false
		return
	}

	// 日期配置
	dateConfigList := len(req.DateConfigList)
	if dateConfigList < 0 {
		br.Msg = "请选择日期"
		br.IsSendEmail = false
		return
	}
	if dateConfigList > 5 {
		br.Msg = "日期数量已达上限！"
		br.IsSendEmail = false
		return
	}

	chartInfo := new(data_manage.ChartInfoView)
	chartInfo.ChartType = utils.CHART_SOURCE_CROSS_HEDGING

	// 获取图表x轴y轴
	_, dataResp, err, errMsg, isSendEmail := cross_variety.GetChartData(0, req)
	if err != nil {
		br.IsSendEmail = isSendEmail
		br.Msg = "获取失败"
		if errMsg != `` {
			br.Msg = errMsg
		}
		br.ErrMsg = "获取图表，指标信息失败，Err:" + err.Error()
		return
	}
	br.Data = dataResp
	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
}

// Add
// @Title 新增图表接口
// @Description 新增图表接口
// @Param	request	body request.AddChartInfoReq true "type json string"
// @Success 200 {object} data_manage.AddChartInfoResp
// @router /chart_info/add [post]
func (c *ChartInfoController) Add() {
	br := new(models.BaseResponse).Init()
	defer func() {
		c.Data["json"] = br
		c.ServeJSON()
	}()

	sysUser := c.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}

	cacheKey := "CACHE_CHART_INFO_ADD_" + strconv.Itoa(sysUser.AdminId)
	if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
		br.Msg = "系统处理中，请稍后重试！"
		br.ErrMsg = "系统处理中，请稍后重试！" + sysUser.RealName + ";data:" + string(c.Ctx.Input.RequestBody)
		return
	}
	defer func() {
		_ = utils.Rc.Delete(cacheKey)
	}()

	var req request.AddChartReq
	err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}

	// 添加图表
	chartInfo, err, errMsg, isSendEmail := cross_variety.AddChartInfo(req, utils.CHART_SOURCE_CROSS_HEDGING, sysUser)
	if err != nil {
		br.Msg = "保存失败"
		if errMsg != `` {
			br.Msg = errMsg
		}
		br.ErrMsg = err.Error()
		br.IsSendEmail = isSendEmail
		return
	}

	//新增操作日志
	{
		chartLog := new(data_manage.ChartInfoLog)
		chartLog.ChartInfoId = chartInfo.ChartInfoId
		chartLog.ChartName = req.ChartName
		chartLog.ChartClassifyId = chartInfo.ChartClassifyId
		chartLog.SysUserId = sysUser.AdminId
		chartLog.SysUserRealName = sysUser.RealName
		chartLog.UniqueCode = chartInfo.UniqueCode
		chartLog.CreateTime = time.Now()
		chartLog.Content = string(c.Ctx.Input.RequestBody)
		chartLog.Status = "新增跨品种分析图表"
		chartLog.Method = c.Ctx.Input.URI()
		go data_manage.AddChartInfoLog(chartLog)
	}

	resp := new(data_manage.AddChartInfoResp)
	resp.ChartInfoId = chartInfo.ChartInfoId
	resp.UniqueCode = chartInfo.UniqueCode
	resp.ChartType = chartInfo.ChartType
	br.Ret = 200
	br.Success = true
	br.Msg = "保存成功"
	br.Data = resp
	br.IsAddLog = true
}

// Edit
// @Title 编辑图表接口
// @Description 编辑图表接口
// @Param	request	body request.EditChartInfoReq true "type json string"
// @Success Ret=200 保存成功
// @router /chart_info/edit [post]
func (c *ChartInfoController) Edit() {
	br := new(models.BaseResponse).Init()
	defer func() {
		c.Data["json"] = br
		c.ServeJSON()
	}()

	sysUser := c.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}

	var req request.EditChartReq
	err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}

	chartItem, err, errMsg, isSendEmail := cross_variety.EditChartInfo(req, sysUser)
	if err != nil {
		br.Msg = "保存失败"
		if errMsg != `` {
			br.Msg = errMsg
		}
		br.ErrMsg = err.Error()
		br.IsSendEmail = isSendEmail
		return
	}

	resp := new(data_manage.AddChartInfoResp)
	resp.ChartInfoId = chartItem.ChartInfoId
	resp.UniqueCode = chartItem.UniqueCode
	//resp.ChartType = req.ChartType

	//新增操作日志
	{
		chartLog := new(data_manage.ChartInfoLog)
		chartLog.ChartName = chartItem.ChartName
		chartLog.ChartInfoId = req.ChartInfoId
		chartLog.ChartClassifyId = chartItem.ChartClassifyId
		chartLog.SysUserId = sysUser.AdminId
		chartLog.SysUserRealName = sysUser.RealName
		chartLog.UniqueCode = chartItem.UniqueCode
		chartLog.CreateTime = time.Now()
		chartLog.Content = string(c.Ctx.Input.RequestBody)
		chartLog.Status = "编辑跨品种分析图表"
		chartLog.Method = c.Ctx.Input.URL()
		go data_manage.AddChartInfoLog(chartLog)
	}

	br.Ret = 200
	br.Success = true
	br.Msg = "保存成功"
	br.Data = resp
	br.IsAddLog = true
}

//
//// Detail
//// @Title 获取图表详情
//// @Description 获取图表详情接口
//// @Param   ChartInfoId   query   int  true       "图表id"
//// @Param   DateType   query   int  true       "日期类型:1:00年至今,2:10年至今,3:15年至今，4：年初至今，5:自定义时间"
//// @Param   StartDate   query   string  true       "自定义开始日期"
//// @Param   EndDate   query   string  true       "自定义结束日期"
//// @Param   Calendar   query   string  true       "公历/农历"
//// @Param   SeasonStartDate   query   string  true       "季节性图开始日期"
//// @Param   SeasonEndDate   query   string  true       "季节性图结束日期"
//// @Param   EdbInfoId   query   string  true       "指标ID，多个用英文逗号隔开"
//// @Param   ChartType   query   int  true       "生成样式:1:曲线图,2:季节性图"
//// @Success 200 {object} data_manage.ChartInfoDetailResp
//// @router /chart_info/detail [get]
//func (c *ChartInfoController) Detail() {
//	br := new(models.BaseResponse).Init()
//	defer func() {
//		c.Data["json"] = br
//		c.ServeJSON()
//	}()
//	sysUser := c.SysUser
//	if sysUser == nil {
//		br.Msg = "请登录"
//		br.ErrMsg = "请登录，SysUser Is Empty"
//		br.Ret = 408
//		return
//	}
//	chartInfoId, _ := c.GetInt("ChartInfoId")
//	if chartInfoId <= 0 {
//		br.Msg = "参数有误"
//		return
//	}
//
//	var err error
//	chartInfo := new(data_manage.ChartInfoView)
//	chartInfo, err = data_manage.GetChartInfoViewById(chartInfoId)
//	if err != nil {
//		if err.Error() == utils.ErrNoRow() {
//			br.Msg = "图被删除，请刷新页面"
//			br.ErrMsg = "图被删除，请刷新页面，Err:" + err.Error()
//			return
//		}
//		br.Msg = "获取失败"
//		br.ErrMsg = "获取图表信息失败，Err:" + err.Error()
//		return
//	}
//
//	// 相关性图表信息
//	correlationChart := new(data_manage.ChartInfoCorrelation)
//	if e := correlationChart.GetItemById(chartInfoId); e != nil {
//		br.Msg = "获取失败"
//		br.ErrMsg = "获取图表相关性信息失败, Err: " + e.Error()
//		return
//	}
//
//	// 获取指标信息
//	edbInfoMappingA, e := data_manage.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdFirst)
//	if e != nil {
//		br.Msg = "获取失败"
//		br.ErrMsg = "获取相关性图表, A指标mapping信息失败, Err:" + e.Error()
//		return
//	}
//	edbInfoMappingB, e := data_manage.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdSecond)
//	if e != nil {
//		br.Msg = "获取失败"
//		br.ErrMsg = "获取相关性图表, B指标mapping信息失败, Err:" + e.Error()
//		return
//	}
//
//
//
//	// 获取图表x轴y轴
//	_, dataResp, err, errMsg, isSendEmail := cross_variety.GetChartData(0, req)
//
//	var dataResp interface{} // 绘图数据返回（目前是滚动相关性的图）
//	var xEdbIdValue []int
//	var yDataList []data_manage.YData
//	switch chartInfo.Source {
//	case utils.CHART_SOURCE_CORRELATION: // 相关性图
//		moveUnitDays, ok := utils.FrequencyDaysMap[correlationChart.CalculateUnit]
//		if !ok {
//			br.Msg = "错误的分析周期"
//			br.IsSendEmail = false
//			return
//		}
//		startDate := time.Now().AddDate(0, 0, -correlationChart.CalculateValue*moveUnitDays).Format(utils.FormatDate)
//		endDate := time.Now().Format(utils.FormatDate)
//
//		xEdbIdValue, yDataList, e = correlationServ.GetChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, startDate, endDate)
//		if e != nil {
//			br.Msg = "获取失败"
//			br.ErrMsg = "获取相关性图表, 图表计算值失败, Err:" + e.Error()
//			return
//		}
//	case utils.CHART_SOURCE_ROLLING_CORRELATION: // 滚动相关性图
//		startDate, endDate := utils.GetDateByDateType(correlationChart.DateType, correlationChart.StartDate.Format(utils.FormatDate), correlationChart.EndDate.Format(utils.FormatDate))
//		dataResp, e = correlationServ.GetRollingCorrelationChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, correlationChart.CalculateValue, correlationChart.CalculateUnit, startDate, endDate, chartInfo.ChartName, chartInfo.ChartNameEn)
//		if e != nil {
//			br.Msg = "获取失败"
//			br.ErrMsg = "获取滚动相关性图表, 图表计算值失败, Err:" + e.Error()
//			return
//		}
//	}
//
//	// 完善指标信息
//	edbList, e := correlationServ.GetChartEdbInfoFormat(chartInfo.ChartInfoId, edbInfoMappingA, edbInfoMappingB)
//	if e != nil {
//		br.Msg = "获取失败"
//		br.ErrMsg = "获取相关性图表, 完善指标信息失败, Err:" + e.Error()
//		return
//	}
//	correlationInfo := new(data_manage.CorrelationInfo)
//	correlationInfo.LeadValue = correlationChart.LeadValue
//	correlationInfo.LeadUnit = correlationChart.LeadUnit
//	correlationInfo.CalculateValue = correlationChart.CalculateValue
//	correlationInfo.CalculateUnit = correlationChart.CalculateUnit
//	correlationInfo.StartDate = correlationChart.StartDate.Format(utils.FormatDate)
//	correlationInfo.EndDate = correlationChart.EndDate.Format(utils.FormatDate)
//	correlationInfo.LeadValue = correlationChart.LeadValue
//	correlationInfo.EdbInfoIdFirst = correlationChart.EdbInfoIdFirst
//	correlationInfo.EdbInfoIdSecond = correlationChart.EdbInfoIdSecond
//
//	// 判断是否加入我的图库
//	if chartInfoId > 0 && chartInfo != nil {
//		{
//			var myChartCondition string
//			var myChartPars []interface{}
//			myChartCondition += ` AND a.admin_id=? `
//			myChartPars = append(myChartPars, sysUser.AdminId)
//			myChartCondition += ` AND a.chart_info_id=? `
//			myChartPars = append(myChartPars, chartInfo.ChartInfoId)
//
//			myChartList, err := data_manage.GetMyChartByCondition(myChartCondition, myChartPars)
//			if err != nil && err.Error() != utils.ErrNoRow() {
//				br.Msg = "获取失败"
//				br.ErrMsg = "获取我的图表信息失败，GetMyChartByCondition，Err:" + err.Error()
//				return
//			}
//			if myChartList != nil && len(myChartList) > 0 {
//				chartInfo.IsAdd = true
//				chartInfo.MyChartId = myChartList[0].MyChartId
//				chartInfo.MyChartClassifyId = myChartList[0].MyChartClassifyId
//			}
//		}
//	}
//
//	//图表操作权限
//	chartInfo.IsEdit = data.CheckOpChartPermission(sysUser, chartInfo.SysUserId)
//	//判断是否需要展示英文标识
//	chartInfo.IsEnChart = data.CheckIsEnChart(chartInfo.ChartNameEn, edbList[0:1], chartInfo.Source, chartInfo.ChartType)
//	chartInfo.UnitEn = edbInfoMappingA.UnitEn
//
//	isSaveAs := true
//	if chartInfo.Source == utils.CHART_SOURCE_ROLLING_CORRELATION {
//		isSaveAs = false
//	}
//	// 另存为
//	chartInfo.Button = data_manage.ChartViewButton{
//		IsEdit:    chartInfo.IsEdit,
//		IsEnChart: chartInfo.IsEnChart,
//		IsAdd:     chartInfo.IsAdd,
//		IsCopy:    isSaveAs,
//		IsSetName: chartInfo.IsSetName,
//	}
//
//	resp := new(data_manage.ChartInfoDetailResp)
//	resp.ChartInfo = chartInfo
//	resp.EdbInfoList = edbList
//	resp.XEdbIdValue = xEdbIdValue
//	resp.YDataList = yDataList
//	resp.CorrelationChartInfo = correlationInfo
//	resp.DataResp = dataResp
//	br.Ret = 200
//	br.Success = true
//	br.Msg = "获取成功"
//	br.Data = resp
//}