eta_api/services/data/excel/mixed_table.go

package excel

import (
	"encoding/json"
	"errors"
	"eta/eta_api/models/data_manage"
	"eta/eta_api/models/data_manage/excel/request"
	"eta/eta_api/services/data"
	"eta/eta_api/utils"
	"fmt"
	"strings"
	"time"
)

// GetMixedTableCellData 获取混合表格数据
func GetMixedTableCellData(cellRelationConf string, config [][]request.MixedTableCellDataReq) (newMixedTableCellDataList [][]request.MixedTableCellDataReq, err error) {
	newMixedTableCellDataList = config

	// 单元格关系配置x信息
	cellRelationConfMap := make(map[string]request.CellRelationConf)
	cellRelationConfList := make([]request.CellRelationConf, 0)
	if cellRelationConf != `` {
		err = json.Unmarshal([]byte(cellRelationConf), &cellRelationConfList)
		if err != nil {
			return
		}

		for _, v := range cellRelationConfList {
			cellRelationConfMap[v.Key] = v
		}
	}

	// 找出所有的关联指标id
	config, edbInfoIdList, _ := handleConfig(config)

	edbInfoList, err := data_manage.GetEdbInfoByIdList(edbInfoIdList)
	if err != nil {
		return
	}

	// 指标信息map
	edbInfoMap := make(map[int]*data_manage.EdbInfo)
	// 日度指标数据map
	edbDataListMap := make(map[int]map[string]float64)
	// 月度指标数据map
	edbMonthDataListMap := make(map[int]map[string]float64)
	for _, edbInfo := range edbInfoList {
		edbInfoMap[edbInfo.EdbInfoId] = edbInfo

		dataList := make([]*data_manage.EdbDataList, 0)
		switch edbInfo.EdbInfoType {
		case 0:
			dataList, _ = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.EdbInfoId, ``, ``)
		case 1:
			_, dataList, _, _, _, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
		default:
			err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
		}

		dateValMap := make(map[string]float64)
		monthValMap := make(map[string]float64)
		for _, data := range dataList {
			// 日度数据
			dateValMap[data.DataTime] = data.Value
			// 月度数据（取该月份的第一个数据）
			yearMonth := strings.Join(strings.Split(data.DataTime, "-")[0:2], "-")
			if _, ok := monthValMap[yearMonth]; !ok {
				monthValMap[yearMonth] = data.Value
			}
		}
		edbDataListMap[edbInfo.EdbInfoId] = dateValMap
		edbMonthDataListMap[edbInfo.EdbInfoId] = monthValMap
	}

	for k, row := range newMixedTableCellDataList {
		for i, cell := range row {
			if cell.DataType == request.EdbDT {
				if edbInfo, ok := edbInfoMap[cell.EdbInfoId]; ok {
					cell.ShowValue = edbInfo.EdbName
				}
			} else if utils.InArrayByInt([]int{request.InsertDataDT, request.PopInsertDataDT}, cell.DataType) {
				tmpDateList := strings.Split(cell.DataTime, "-")
				tmpDateValMap := make(map[string]float64)
				if len(tmpDateList) == 2 {
					//月度数据
					if dateValMap, ok := edbMonthDataListMap[cell.EdbInfoId]; ok {
						tmpDateValMap = dateValMap
					}
				} else {
					// 日度数据
					if dateValMap, ok := edbDataListMap[cell.EdbInfoId]; ok {
						tmpDateValMap = dateValMap
					}

				}
				if val, ok2 := tmpDateValMap[cell.DataTime]; ok2 {
					//cell.ShowValue = fmt.Sprint(val)
					cell.ShowValue = utils.FormatTableDataShowValue(val)
				}
			}

			row[i] = cell
		}
		newMixedTableCellDataList[k] = row
	}

	return
}

func handleConfig(config [][]request.MixedTableCellDataReq) ([][]request.MixedTableCellDataReq, []int, []int) {
	edbInfoIdList := make([]int, 0)
	dataEdbInfoIdList := make([]int, 0)
	for _, row := range config {
		for _, cell := range row {
			switch cell.DataType {
			case request.EdbDT: // 指标信息
				edbInfoIdList = append(edbInfoIdList, cell.EdbInfoId)
			case request.InsertDataDT, request.PopInsertDataDT: // 插值、弹框插值
				dataEdbInfoIdList = append(dataEdbInfoIdList, cell.EdbInfoId)
			case request.DateDT: // 日期类型
			}
		}
	}

	return config, edbInfoIdList, dataEdbInfoIdList
}

func HandleDate(dataTimeType int, val string) (date string, err error, errMsg string) {
	return handleDate(dataTimeType, val)
}

func handleDate(dataTimeType int, val string) (date string, err error, errMsg string) {
	if val == `` {
		errMsg = "错误的日期数据"
		err = errors.New(errMsg)
		return
	}
	switch dataTimeType {
	case request.CustomDateT: //手动输入日期
		date = val
	case request.SystemDateT: // 系统日期
		date = time.Now().Format(utils.FormatDate)
	case request.SystemCalculateDateT: // 系统日期计算后的日期
		date, err, errMsg = handleSystemCalculateDateT(val)
	case request.SystemAppointDateT: // 处理系统日期相关的指定频率（所在周/旬/月/季/半年/年的最后/最早一天）
		date, err, errMsg = handleSystemAppointDateT(val)
	case request.EdbDateDT: // 导入指标日期（指标库的最新日期）
	default:
		errMsg = "错误的日期类型"
		err = errors.New(errMsg)
		return
	}

	return
}

//	 handleSystemCalculateDateT
//		@Description: 处理系统日期计算后的日期
//		@author: Roc
//		@datetime2023-10-26 11:12:56
//		@param val string
//		@return date string
//		@return err error
//		@return errMsg string
func handleSystemCalculateDateT(val string) (date string, err error, errMsg string) {
	type SystemCalculateVal struct {
		Num       int
		Frequency string
	}
	var config SystemCalculateVal
	err = json.Unmarshal([]byte(val), &config)
	if err != nil {
		return
	}
	currDate := time.Now()
	switch config.Frequency {
	case "", "日":
		date = currDate.AddDate(0, 0, config.Num).Format(utils.FormatDate)
	default:
		errMsg = "错误的日期频度：" + config.Frequency
		err = errors.New(errMsg)
		return
	}

	return
}

//	 handleSystemAppointDateT
//		@Description: 处理系统日期相关的指定频率（所在周/旬/月/季/半年/年的最后/最早一天）
//		@author: Roc
//		@datetime2023-10-26 11:13:52
//		@param val string
//		@return date string
//		@return err error
//		@return errMsg string
func handleSystemAppointDateT(val string) (date string, err error, errMsg string) {
	type SystemCalculateVal struct {
		Frequency string
		Day       string
	}
	var config SystemCalculateVal
	err = json.Unmarshal([]byte(val), &config)
	if err != nil {
		return
	}
	currDate := time.Now()
	switch config.Frequency {
	case "本周":
		day := currDate.Day()
		num := 0
		switch config.Day {
		case "周一":
			num = 1
		case "周二":
			num = 2
		case "周三":
			num = 3
		case "周四":
			num = 4
		case "周五":
			num = 5
		case "周六":
			num = 6
		case "周日":
			num = 0
		}
		day = day - num
	default:
		errMsg = "错误的日期频度：" + config.Frequency
		err = errors.New(errMsg)
		return
	}

	return
}