eta_api/services/data/cross_variety/chart.go

package cross_variety

import (
	"encoding/json"
	"errors"
	"eta/eta_api/models/data_manage"
	cross_varietyModel "eta/eta_api/models/data_manage/cross_variety"
	"eta/eta_api/models/data_manage/cross_variety/request"
	"eta/eta_api/models/system"
	"eta/eta_api/services/data"
	"eta/eta_api/utils"
	"fmt"
	"github.com/shopspring/decimal"
	"strconv"
	"strings"
	"time"
)

type LineEquationDataResp struct {
	MaxData             float64
	MinData             float64
	LatestDate          string `description:"真实数据的最后日期"`
	EdbInfoCategoryType int
	ChartColor          string
	ChartStyle          string
	PredictChartColor   string
	ChartType           int
	ChartWidth          int
	EdbName             string
	EdbNameEn           string
	Unit                string
	UnitEn              string
	IsAxis              int
	DataList            []data_manage.EdbDataList
}

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

// SectionScatterSeriesItemResp 系列的返回
type SectionScatterSeriesItemResp struct {
	Name                string            `description:"系列名"`
	NameEn              string            `description:"系列名(英文)"`
	Color               string            `description:"颜色"`
	CoordinatePointData []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
	XDate string
	YDate string
}

// GetChartData
// @Description: 获取跨品种分析图表数据
// @author: Roc
// @datetime 2023-11-24 09:42:59
// @param chartInfoId int
// @param config request.ChartConfigReq
// @return edbList []*data_manage.ChartEdbInfoMapping
// @return dataResp ChartInfoResp
// @return err error
// @return errMsg string
// @return isSendEmail bool
func GetChartData(chartInfoId int, config request.ChartConfigReq) (edbList []*data_manage.ChartEdbInfoMapping, dataResp ChartInfoResp, err error, errMsg string, isSendEmail bool) {
	moveUnitDays, ok := utils.FrequencyDaysMap[config.CalculateUnit]
	if !ok {
		errMsg = "错误的分析周期"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	isSendEmail = true

	// 品种map
	varietyMap := make(map[int]*cross_varietyModel.ChartVariety)
	{
		varietyList, tmpErr := cross_varietyModel.GetVarietyListByIdList(config.VarietyList)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		for _, v := range varietyList {
			varietyMap[v.ChartVarietyId] = v
		}
	}

	// 标签m
	var xTagInfo, yTagInfo *cross_varietyModel.ChartTag
	{
		tagList, tmpErr := cross_varietyModel.GetTagListByIdList([]int{config.TagX, config.TagY})
		if tmpErr != nil {
			err = tmpErr
			return
		}
		for _, v := range tagList {
			if v.ChartTagId == config.TagX {
				xTagInfo = v
			} else if v.ChartTagId == config.TagY {
				yTagInfo = v
			}
		}
	}
	if xTagInfo == nil {
		errMsg = "找不到对应的X轴标签"
		err = errors.New(errMsg)
		return
	}
	if yTagInfo == nil {
		errMsg = "找不到对应的Y轴标签"
		err = errors.New(errMsg)
		return
	}

	xVarietyEdbMap := make(map[int]int)
	yVarietyEdbMap := make(map[int]int)
	xList, err := cross_varietyModel.GetChartTagVarietyListByTagAndVariety(config.TagX, config.VarietyList)
	if err != nil {
		errMsg = "获取失败"
		err = errors.New("获取X轴的品种指标配置信息失败，Err:" + err.Error())
		return
	}
	yList, err := cross_varietyModel.GetChartTagVarietyListByTagAndVariety(config.TagY, config.VarietyList)
	if err != nil {
		errMsg = "获取失败"
		err = errors.New("获取Y轴的品种指标配置信息失败，Err:" + err.Error())
		return
	}
	edbInfoIdList := make([]int, 0)
	for _, v := range xList {
		edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
		xVarietyEdbMap[v.ChartVarietyId] = v.EdbInfoId
	}
	for _, v := range yList {
		edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
		yVarietyEdbMap[v.ChartVarietyId] = v.EdbInfoId
	}

	// todo 需要取交集

	if len(edbInfoIdList) <= 0 {
		errMsg = "品种未配置指标"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	mappingList, err := data_manage.GetChartEdbMappingListByEdbInfoIdList(edbInfoIdList)
	if err != nil {
		errMsg = "获取指标信息失败"
		err = errors.New("获取指标信息失败，ERR：" + err.Error())
		return
	}

	// 指标对应的所有数据
	chartType := 1 //1：普通图，2：季节性图
	calendar := "公历"
	edbDataListMap, edbList, err := data.GetEdbDataMapList(chartInfoId, chartType, calendar, "", "", mappingList, "")
	if err != nil {
		return
	}

	currDay := time.Now()
	currDay = time.Date(currDay.Year(), currDay.Month(), currDay.Day(), 0, 0, 0, 0, time.Local)

	dataMap := make(map[string]float64)
	dateMap := make(map[string]string)
	for dateIndex, dateConfig := range config.DateConfigList {
		for _, edbInfoMapping := range mappingList {
			// 数据会是正序的
			dataList, ok := edbDataListMap[edbInfoMapping.EdbInfoId]
			if !ok {
				continue
			}

			lenData := len(dataList)
			if lenData <= 0 {
				continue
			}

			// 数据开始日期
			endDateStr := ``
			var endDate time.Time
			// 数据的开始索引
			k := lenData - 1
			var currVal float64
			switch dateConfig.DateType {
			case 1: // 1：最新日期；
				endDateStr = dataList[k].DataTime
				tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
				if tmpErr != nil {
					err = tmpErr
					return
				}
				endDate = tmpDate
				currVal = dataList[k].Value
			case 2: // 2：N天前
				tmpEndDate := currDay.AddDate(0, 0, -dateConfig.Num)
				tmpEndDateStr := tmpEndDate.Format(utils.FormatDate)

				for i := k; i >= 0; i-- {
					tmpDateStr := dataList[i].DataTime
					// 如果正好是这一天，那么就直接break了
					if tmpEndDateStr == tmpDateStr {
						k = i
						endDateStr = tmpDateStr
						currVal = dataList[i].Value
						break
					}
					tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDateStr, time.Local)
					if tmpErr != nil {
						err = tmpErr
						return
					}
					// 如果这期的日期晚于选择的日期，那么继续遍历
					if tmpDate.After(tmpEndDate) {
						continue
					}
					k = i
					endDateStr = tmpDateStr
					endDate = tmpDate
					currVal = dataList[i].Value
					break
				}
			}

			// 没有找到日期，那么就不处理
			if endDateStr == `` || endDate.IsZero() {
				continue
			}

			// 最早的日期
			earliestDate := endDate.AddDate(0, 0, -config.CalculateValue*moveUnitDays)
			earliestDateStr := earliestDate.Format(utils.FormatDate)

			var minVal, maxVal float64
			var isNotFirst bool // 是否是第一条数据
			for i := k; i >= 0; i-- {
				tmpData := dataList[i]
				if !isNotFirst {
					maxVal = tmpData.Value
					minVal = tmpData.Value
					isNotFirst = true
					continue
				}

				tmpDateStr := dataList[i].DataTime
				// 如果正好是这一天，那么就直接break了
				if earliestDateStr == tmpDateStr {
					break
				}
				tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDateStr, time.Local)
				if tmpErr != nil {
					err = tmpErr
					return
				}
				// 如果这期的日期早于选择的日期，那么继续停止遍历
				if tmpDate.Before(earliestDate) {
					continue
				}

				if tmpData.Value > maxVal {
					maxVal = tmpData.Value
				}
				if tmpData.Value < minVal {
					minVal = tmpData.Value
				}
			}

			// 最大值等于最小值，说明计算结果无效
			if maxVal == minVal {
				continue
			}
			//百分位=(现值-Min)/(Max-Min)
			tmpV := (currVal - minVal) / (maxVal - minVal) * 100
			tmpV, _ = decimal.NewFromFloat(tmpV).Round(4).Float64()

			// key的生成（日期配置下标+指标id)
			key := fmt.Sprint(dateIndex, "_", edbInfoMapping.EdbInfoId)
			dataMap[key] = tmpV
			dateMap[key] = endDateStr
		}
	}

	// 返回数据处理
	dataList := make([]SectionScatterSeriesItemResp, 0)
	var xMinVal, xMaxVal, yMinVal, yMaxVal float64
	var isNotFirst bool

	for _, varietyId := range config.VarietyList {
		xEdbInfoId, ok1 := xVarietyEdbMap[varietyId]
		if !ok1 {
			continue
		}
		yEdbInfoId, ok2 := yVarietyEdbMap[varietyId]
		if !ok2 {
			continue
		}
		variety, ok := varietyMap[varietyId]
		if !ok {
			continue
		}

		coordinatePointList := make([]CoordinatePoint, 0)

		for dateIndex, _ := range config.DateConfigList {
			key1 := fmt.Sprint(dateIndex, "_", xEdbInfoId)
			xVal, ok1 := dataMap[key1]
			if !ok1 {
				continue
			}
			key2 := fmt.Sprint(dateIndex, "_", yEdbInfoId)
			yVal, ok2 := dataMap[key2]
			if !ok2 {
				continue
			}
			if !isNotFirst {
				xMinVal = xVal
				xMaxVal = xVal
				yMinVal = yVal
				yMaxVal = yVal
				isNotFirst = true
			} else {
				if xVal < xMinVal {
					xMinVal = xVal
				}
				if xVal > xMaxVal {
					xMaxVal = xVal
				}
				if yVal < yMinVal {
					yMinVal = yVal
				}
				if yVal > yMaxVal {
					yMaxVal = yVal
				}
			}
			coordinatePointList = append(coordinatePointList, CoordinatePoint{
				X:     xVal,
				Y:     yVal,
				XDate: dateMap[key1],
				YDate: dateMap[key2],
			})
		}

		dataList = append(dataList, SectionScatterSeriesItemResp{
			Name:                variety.ChartVarietyName,
			NameEn:              "",
			Color:               "",
			CoordinatePointData: coordinatePointList,
		})
	}

	dataResp = ChartInfoResp{
		XName:       xTagInfo.ChartTagName,
		XNameEn:     "",
		XUnitName:   "%",
		XUnitNameEn: "%",
		YName:       yTagInfo.ChartTagName,
		YNameEn:     "",
		YUnitName:   "%",
		YUnitNameEn: "%",
		XMinValue:   fmt.Sprint(xMinVal),
		XMaxValue:   fmt.Sprint(xMaxVal),
		YMinValue:   fmt.Sprint(yMinVal),
		YMaxValue:   fmt.Sprint(yMaxVal),
		DataList:    dataList,
	}

	return
}

// AddChartInfo
// @Description: AddChartInfo
// @author: Roc
// @datetime 2023-11-24 15:58:14
// @param req request.AddChartReq
// @param sysUser *system.Admin
// @return chartInfo *data_manage.ChartInfo
// @return err error
// @return errMsg string
// @return isSendEmail bool
func AddChartInfo(req request.AddChartReq, sysUser *system.Admin) (chartInfo *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
	isSendEmail = true

	source := utils.CHART_SOURCE_CROSS_HEDGING

	req.ChartName = strings.Trim(req.ChartName, " ")
	if req.ChartName == "" {
		errMsg = "请填写图表名称！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.TagX <= 0 {
		errMsg = "请选择X轴坐标的标签！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.TagY <= 0 {
		errMsg = "请选择Y轴坐标的标签！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.CalculateValue <= 0 {
		errMsg = "请设置时间长度！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.CalculateUnit == `` {
		errMsg = "请设置时间频度！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	// 品种配置
	if len(req.VarietyList) < 0 {
		errMsg = "请选择品种！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	// 日期配置
	dateConfigList := len(req.DateConfigList)
	if dateConfigList < 0 {
		errMsg = "请选择日期！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if dateConfigList > 5 {
		errMsg = "日期数量已达上限！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	// 基础配置转string
	extraConfigByte, err := json.Marshal(req)
	if err != nil {
		return
	}

	chartClassify, err := data_manage.GetCrossVarietyChartClassifyBySysUserId(sysUser.AdminId)
	if err != nil {
		if err.Error() != utils.ErrNoRow() {
			errMsg = "获取分类信息失败"
			err = errors.New("获取分类信息失败，Err:" + err.Error())
			return
		}

		timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
		chartClassify = &data_manage.ChartClassify{
			ChartClassifyId:   0,
			ChartClassifyName: sysUser.RealName,
			ParentId:          0,
			HasData:           0,
			CreateTime:        time.Now(),
			ModifyTime:        time.Now(),
			SysUserId:         sysUser.AdminId,
			SysUserRealName:   sysUser.RealName,
			Level:             1,
			UniqueCode:        utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
			Sort:              0,
			Source:            source,
		}
	}

	var chartInfoId int
	// 判断图表是否存在
	var condition string
	var pars []interface{}
	condition += " AND chart_name=? AND source = ? "
	pars = append(pars, req.ChartName, source)
	count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
	if err != nil {
		errMsg = "判断图表名称是否存在失败"
		err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
		return
	}

	if count > 0 {
		errMsg = "图表已存在，请重新填写"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	chartInfo = new(data_manage.ChartInfo)
	chartInfo.ChartName = req.ChartName
	//chartInfo.EdbInfoIds = edbInfoIdStr
	//chartInfo.ChartClassifyId = req.ChartClassifyId
	chartInfo.SysUserId = sysUser.AdminId
	chartInfo.SysUserRealName = sysUser.RealName
	chartInfo.CreateTime = time.Now()
	chartInfo.ModifyTime = time.Now()
	chartInfo.IsSetName = 0
	timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
	chartInfo.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
	chartInfo.ChartType = 9 // 相关性图
	chartInfo.Calendar = "公历"
	chartInfo.DateType = 6
	//chartInfo.StartDate = req.StartDate
	//chartInfo.EndDate = req.EndDate
	//chartInfo.SeasonStartDate = req.StartDate
	//chartInfo.SeasonEndDate = req.EndDate
	chartInfo.LeftMin = req.LeftMin
	chartInfo.LeftMax = req.LeftMax
	//chartInfo.RightMin = req.RightMin
	//chartInfo.RightMax = req.RightMax
	//chartInfo.Disabled = disableVal
	chartInfo.Source = source
	chartInfo.ExtraConfig = string(extraConfigByte)

	// 图表品种
	chartVarietyMappingList := make([]*cross_varietyModel.ChartVarietyMapping, 0)
	for _, varietyId := range req.VarietyList {
		chartVarietyMappingList = append(chartVarietyMappingList, &cross_varietyModel.ChartVarietyMapping{
			Id:             0,
			ChartInfoId:    0,
			ChartVarietyId: varietyId,
			ModifyTime:     time.Now(),
			CreateTime:     time.Now(),
		})
	}

	// 图表配置
	chartInfoCrossVariety := &cross_varietyModel.ChartInfoCrossVariety{
		Id:             0,
		ChartInfoId:    0,
		ChartXTagId:    req.TagX,
		ChartYTagId:    req.TagY,
		CalculateValue: req.CalculateValue,
		CalculateUnit:  req.CalculateUnit,
		ModifyTime:     time.Now(),
		CreateTime:     time.Now(),
	}

	// 新增图表和指标mapping
	chartInfoId, e := cross_varietyModel.CreateChart(chartInfo, chartClassify, chartVarietyMappingList, chartInfoCrossVariety)
	if e != nil {
		errMsg = "操作失败"
		err = errors.New("新增相关性图表失败, Err: " + e.Error())
		return
	}

	//添加es数据
	go data.EsAddOrEditChartInfo(chartInfoId)

	return
}

// EditChartInfo
// @Description: 编辑图表
// @author: Roc
// @datetime 2023-11-24 15:58:31
// @param req request.EditChartReq
// @param sysUser *system.Admin
// @return chartItem *data_manage.ChartInfo
// @return err error
// @return errMsg string
// @return isSendEmail bool
func EditChartInfo(req request.EditChartReq, sysUser *system.Admin) (chartItem *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
	isSendEmail = true

	chartItem, err = data_manage.GetChartInfoById(req.ChartInfoId)
	if err != nil {
		if err.Error() == utils.ErrNoRow() {
			errMsg = "图表已被删除，请刷新页面"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
		errMsg = "获取图表信息失败"
		err = errors.New("获取图表信息失败,Err:" + err.Error())
		return
	}

	if chartItem.Source != utils.CHART_SOURCE_CROSS_HEDGING {
		errMsg = "该图不是跨品种分析图表！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	req.ChartName = strings.Trim(req.ChartName, " ")
	if req.ChartName == "" {
		errMsg = "请填写图表名称！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.TagX <= 0 {
		errMsg = "请选择X轴坐标的标签！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.TagY <= 0 {
		errMsg = "请选择Y轴坐标的标签！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.CalculateValue <= 0 {
		errMsg = "请设置时间长度！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if req.CalculateUnit == `` {
		errMsg = "请设置时间频度！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	// 品种配置
	if len(req.VarietyList) < 0 {
		errMsg = "请选择品种！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	// 日期配置
	dateConfigList := len(req.DateConfigList)
	if dateConfigList < 0 {
		errMsg = "请选择日期！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if dateConfigList > 5 {
		errMsg = "日期数量已达上限！"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	// 基础配置转string
	extraConfigByte, err := json.Marshal(req)
	if err != nil {
		return
	}

	// 图表操作权限
	ok := data.CheckOpChartPermission(sysUser, chartItem.SysUserId)
	if !ok {
		errMsg = "没有该图表的操作权限"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	//判断图表是否存在
	var condition string
	var pars []interface{}
	condition += " AND chart_info_id <> ? "
	pars = append(pars, req.ChartInfoId)
	condition += " AND chart_name=? AND source = ? "
	pars = append(pars, req.ChartName, chartItem.Source)
	count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
	if err != nil {
		errMsg = "判断图表名称是否存在失败"
		err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
		return
	}
	if count > 0 {
		errMsg = "图表已存在，请重新填写"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	chartItem.ChartName = req.ChartName
	chartItem.ExtraConfig = string(extraConfigByte)
	chartItem.ModifyTime = time.Now()
	chartUpdateCols := []string{"ChartName", "ExtraConfig", "ModifyTime"}

	// 跨品种分析配置
	chartInfoCrossVariety, err := cross_varietyModel.GetChartInfoCrossVarietyByChartInfoId(chartItem.ChartInfoId)
	if err != nil {
		return
	}
	chartInfoCrossVariety.ChartXTagId = req.TagX
	chartInfoCrossVariety.ChartYTagId = req.TagY
	chartInfoCrossVariety.CalculateValue = req.CalculateValue
	chartInfoCrossVariety.CalculateUnit = req.CalculateUnit
	chartInfoCrossVariety.ModifyTime = time.Now()
	chartInfoCrossVarietyUpdateCols := []string{"ChartXTagId", "ChartYTagId", "CalculateValue", "CalculateUnit", "ModifyTime"}

	// 图表品种
	chartVarietyMappingList := make([]*cross_varietyModel.ChartVarietyMapping, 0)
	for _, varietyId := range req.VarietyList {
		chartVarietyMappingList = append(chartVarietyMappingList, &cross_varietyModel.ChartVarietyMapping{
			Id:             0,
			ChartInfoId:    0,
			ChartVarietyId: varietyId,
			ModifyTime:     time.Now(),
			CreateTime:     time.Now(),
		})
	}

	err = cross_varietyModel.EditChart(chartItem, chartVarietyMappingList, chartInfoCrossVariety,
		chartUpdateCols, chartInfoCrossVarietyUpdateCols)
	if err != nil {
		errMsg = "保存失败"
		err = errors.New("保存失败,Err:" + err.Error())
		return
	}

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

	//添加es数据
	go data.EsAddOrEditChartInfo(chartItem.ChartInfoId)
	//修改my eta es数据
	go data.EsAddOrEditMyChartInfoByChartInfoId(chartItem.ChartInfoId)

	return
}