iailab-plat.git

潘志宝

2024-12-23 5bf42aa9950058f391805e6fb8d7376f4378924b

提交 \| 用户 \| 时间
7fd198	1	package com.iailab.module.model.mdk.sample;
潘	2
50084d	3	import com.iailab.module.data.api.plan.PlanItemApi;
D	4	import com.iailab.module.data.api.plan.dto.ApiPlanItemDTO;
b2aca2	5	import com.iailab.module.data.api.point.DataPointApi;
D	6	import com.iailab.module.data.api.point.dto.ApiPointDTO;
	7	import com.iailab.module.data.enums.DataPointFreqEnum;
50084d	8	import com.iailab.module.data.enums.TimeGranularitySecEnum;
b2aca2	9	import com.iailab.module.model.mdk.common.enums.ModelParamType;
7fd198	10	import com.iailab.module.model.mdk.sample.dto.SampleData;
潘	11	import com.iailab.module.model.mdk.sample.dto.SampleInfo;
	12	import com.iailab.module.model.mdk.vo.DataValueVO;
	13	import org.slf4j.Logger;
	14	import org.slf4j.LoggerFactory;
b2aca2	15	import org.springframework.beans.factory.annotation.Autowired;
7fd198	16	import org.springframework.util.CollectionUtils;
潘	17
	18	import java.util.*;
	19
	20	abstract class SampleDataConstructor {
	21
	22	private Logger logger = LoggerFactory.getLogger(getClass());
b2aca2	23
D	24	@Autowired
	25	private DataPointApi dataPointApi;
50084d	26	@Autowired
D	27	private PlanItemApi planItemApi;
7fd198	28
潘	29	/**
	30	* prepareSampleData
	31	*
	32	* @param sampleInfo
	33	* @return
	34	*/
50084d	35	public abstract List<SampleData> prepareSampleData(SampleInfo sampleInfo) throws Exception;
7fd198	36
潘	37	/**
	38	* 补全数据
	39	*
	40	* @param length
	41	* @param dataEntityList
	42	* @param startTime
	43	* @param endTime
8bf553	44	* @param planMap
7fd198	45	* @return
潘	46	*/
a4891a	47	public List<DataValueVO> completionData(int length, List<DataValueVO> dataEntityList, Date startTime, Date endTime,
8bf553	48	String paramId, String paramType, Map<String, ApiPointDTO> pointMap, Map<String, ApiPlanItemDTO> planMap) {
b2aca2	49	if (CollectionUtils.isEmpty(dataEntityList) \|\| length == dataEntityList.size()) {
7fd198	50	return dataEntityList;
a4891a	51	} else if (length < dataEntityList.size()) {
潘	52	return dataEntityList.subList(dataEntityList.size() - length, dataEntityList.size());
7fd198	53	}
潘	54
b2aca2	55	List<DataValueVO> completionDataEntityList = new ArrayList<>();
D	56	long oneMin = 0L;
	57
	58	long start = startTime.getTime();
	59	long end = endTime.getTime();
	60	long mins = 0L;
	61
	62	switch (ModelParamType.getEumByCode(paramType)) {
19a4fb	63	case NORMALITEM:
D	64	case MERGEITEM:
b2aca2	65	// 预测值
D	66	Calendar calendar = Calendar.getInstance();
	67	calendar.setTime(startTime);
a4891a	68	calendar.set(Calendar.HOUR_OF_DAY, 0);
潘	69	calendar.set(Calendar.MINUTE, 0);
	70	calendar.set(Calendar.SECOND, 0);
	71	calendar.add(Calendar.DAY_OF_YEAR, 1);
b2aca2	72	startTime = calendar.getTime();
D	73	start = startTime.getTime();
	74
	75	calendar.setTime(endTime);
a4891a	76	calendar.set(Calendar.HOUR_OF_DAY, 0);
潘	77	calendar.set(Calendar.MINUTE, 0);
	78	calendar.set(Calendar.SECOND, 0);
b2aca2	79	endTime = calendar.getTime();
D	80	end = endTime.getTime();
	81
50084d	82	oneMin = 60 * 1000L;
b2aca2	83	mins = ((end - start) / oneMin);
D	84	break;
	85	case DATAPOINT:
	86	// 测点值
8bf553	87	ApiPointDTO dataPoint = pointMap.get(paramId);
b2aca2	88	oneMin = 1000L * DataPointFreqEnum.getEumByCode(dataPoint.getMinfreqid()).getValue();
D	89	// 设置时间偏移量
	90	start = start - (start % oneMin) + oneMin;
	91	end = end - (end % oneMin) + oneMin;
	92	mins = ((end - start) / oneMin);
	93	break;
	94	case IND:
	95	// 指标数据
	96	oneMin = 24 * 60 * 60 * 1000;
	97	Calendar calendar2 = Calendar.getInstance();
	98	calendar2.setTime(startTime);
a4891a	99	calendar2.set(Calendar.HOUR_OF_DAY, 0);
潘	100	calendar2.set(Calendar.MINUTE, 0);
	101	calendar2.set(Calendar.SECOND, 0);
b2aca2	102	start = calendar2.getTime().getTime();
D	103
	104	calendar2.setTime(endTime);
a4891a	105	calendar2.set(Calendar.HOUR_OF_DAY, 0);
潘	106	calendar2.set(Calendar.MINUTE, 0);
	107	calendar2.set(Calendar.SECOND, 0);
b2aca2	108	end = calendar2.getTime().getTime();
D	109	mins = ((end - start) / oneMin);
	110	break;
50084d	111	case PLAN:
D	112	// 计划数据
8bf553	113	ApiPlanItemDTO planItem = planMap.get(paramId);
50084d	114	oneMin = 1000L * TimeGranularitySecEnum.getEumByCode(planItem.getTimeGranularity()).getValue();
D	115	// 设置时间偏移量
	116	start = start - (start % oneMin) + oneMin;
	117	end = end - (end % oneMin) + oneMin;
	118	mins = ((end - start) / oneMin);
	119	break;
b2aca2	120	default:
D	121	break;
	122	}
7fd198	123	Map<Long, Double> sourceDataMap = new HashMap<>(dataEntityList.size());
潘	124	for (DataValueVO dataEntity : dataEntityList) {
	125	sourceDataMap.put(dataEntity.getDataTime().getTime(), dataEntity.getDataValue());
	126	}
	127
	128	//找出缺少项
	129	Map<Long, Double> dataMap = new LinkedHashMap<>();
a4891a	130	for (int i = 0; i < mins; i++) {
7fd198	131	Long key = start + oneMin * i;
潘	132	Double value = sourceDataMap.get(key);
	133	dataMap.put(key, value);
	134	}
	135
	136	//补充缺少项
	137	int k = 0;
	138	Map.Entry<Long, Double> lastItem = null;
	139	for (Map.Entry<Long, Double> item : dataMap.entrySet()) {
	140	if (k == 0 && item.getValue() == null) {
	141	item.setValue(getFirstValue(dataMap));
	142	} else if (item.getValue() == null) {
	143	item.setValue(lastItem.getValue());
	144	}
a4891a	145	k++;
7fd198	146	lastItem = item;
潘	147
	148	DataValueVO dataEntity = new DataValueVO();
e0b86e	149	dataEntity.setDataTime(new Date(item.getKey()));
7fd198	150	dataEntity.setDataValue(item.getValue());
潘	151	completionDataEntityList.add(dataEntity);
	152	}
	153	return completionDataEntityList;
	154	}
	155
	156	/**
	157	* getFirstValue
	158	*
	159	* @param dataMap
	160	* @return
	161	*/
	162	private Double getFirstValue(Map<Long, Double> dataMap) {
	163	for (Map.Entry<Long, Double> item : dataMap.entrySet()) {
	164	if (item.getValue() != null) {
	165	return item.getValue();
	166	}
	167	}
	168	return null;
	169	}
	170
	171	}