iailab-plat.git

潘志宝

2024-12-23 5bf42aa9950058f391805e6fb8d7376f4378924b

提交 \| 用户 \| 时间
56dba6	1	package com.iailab.module.data.point.collection.handler;
潘	2
	3	import com.iailab.module.data.api.point.DataPointApi;
	4	import com.iailab.module.data.api.point.dto.ApiPointDTO;
	5	import com.iailab.module.data.api.point.dto.ApiPointValueDTO;
	6	import com.iailab.module.data.api.point.dto.ApiPointValueQueryDTO;
	7	import com.iailab.module.data.common.enums.CommonConstant;
	8	import com.iailab.module.data.enums.DataPointFreqEnum;
	9	import com.iailab.module.data.influxdb.pojo.InfluxPointValuePOJO;
	10	import com.iailab.module.data.point.collection.utils.GenInfluxPointValueUtils;
	11	import com.iailab.module.data.point.dto.DaPointDTO;
5bf42a	12	import com.iailab.module.data.point.service.DaPointService;
56dba6	13	import lombok.extern.slf4j.Slf4j;
潘	14	import org.springframework.beans.factory.annotation.Autowired;
f25963	15	import org.springframework.context.annotation.Lazy;
56dba6	16	import org.springframework.stereotype.Component;
潘	17	import org.springframework.util.CollectionUtils;
	18
5bf42a	19	import javax.annotation.Resource;
56dba6	20	import java.math.BigDecimal;
潘	21	import java.util.*;
	22
	23	/**
	24	* 累计点处理
	25	*
	26	* @author PanZhibao
	27	* @Description
	28	* @createTime 2024年11月29日
	29	*/
	30	@Slf4j
	31	@Component
	32	public class CumulateHandle {
5bf42a	33
潘	34	@Resource
	35	private DaPointService daPointService;
56dba6	36
潘	37	@Autowired
f25963	38	@Lazy
56dba6	39	private DataPointApi dataPointApi;
潘	40
	41	public List<InfluxPointValuePOJO> handle(Date collectTime, List<DaPointDTO> dtos) {
	42	List<InfluxPointValuePOJO> result = new ArrayList<>();
	43	try {
	44	log.info("累计点处理开始");
	45	if (CollectionUtils.isEmpty(dtos)) {
	46	return result;
	47	}
	48	dtos.forEach(dto -> {
	49	try {
	50	Object value = singleCompute(dto, collectTime);
	51	InfluxPointValuePOJO pojo = GenInfluxPointValueUtils.getByPoint(dto, value);
	52	pojo.setTimestamp(collectTime.toInstant());
	53	result.add(pojo);
	54	} catch (Exception ex) {
	55	ex.printStackTrace();
	56	log.info("累计点异常！PointNo=" + dto.getPointNo());
	57	}
	58	});
	59
	60	} catch (Exception ex) {
	61	ex.printStackTrace();
	62	log.info("累计点处理异常！");
	63	}
	64	return result;
	65	}
	66
5bf42a	67	public Map<String, Object> getCurrent(List<String> pointNos) {
潘	68	Map<String, Object> data = new HashMap<>();
	69	List<DaPointDTO> pointMathList = daPointService.getCumulatePoint(pointNos);
	70	if (CollectionUtils.isEmpty(pointMathList)) {
	71	return data;
	72	}
	73	Calendar calendar = Calendar.getInstance();
	74	calendar.set(Calendar.MILLISECOND, 0);
	75	pointMathList.forEach(item -> {
	76	data.put(item.getPointNo(), singleCompute(item, calendar.getTime()));
	77	});
	78	return data;
	79	}
	80
56dba6	81
潘	82	private Object singleCompute(DaPointDTO dto, Date collectTime) {
	83	ApiPointDTO pointDTO = dataPointApi.getInfoByNo(dto.getMomentPoint());
	84	if (pointDTO == null) {
	85	return CommonConstant.BAD_VALUE;
	86	}
	87	Calendar calendar = Calendar.getInstance();
	88	calendar.setTime(collectTime);
	89	calendar.add(Calendar.MINUTE, -1);
	90	Date endTime = calendar.getTime();
	91	calendar.add(Calendar.MINUTE, dto.getLength() * -1);
	92	Date startTime = calendar.getTime();
	93	ApiPointValueQueryDTO queryDto = new ApiPointValueQueryDTO();
	94	queryDto.setStart(startTime);
	95	queryDto.setEnd(endTime);
	96	queryDto.setPointNo(dto.getMomentPoint());
	97
	98	List<ApiPointValueDTO> dataList = dataPointApi.queryPointHistoryValue(queryDto);
	99	if (CollectionUtils.isEmpty(dataList)) {
	100	return BigDecimal.ZERO;
	101	} else if (dataList.size() < dto.getLength()) {
	102	// 补全数据
	103	dataList = completionData(dto.getLength(), dataList, startTime, endTime, pointDTO);
	104	}
	105	double total = dataList.stream().mapToDouble(ApiPointValueDTO::getV).sum();
	106	return new BigDecimal(total).divide(new BigDecimal(dto.getDivisor()), 2, BigDecimal.ROUND_HALF_UP);
	107	}
	108
	109	private List<ApiPointValueDTO> completionData(int length, List<ApiPointValueDTO> dataList, Date startTime, Date endTime, ApiPointDTO pointDTO) {
	110	if (CollectionUtils.isEmpty(dataList) \|\| length == dataList.size()) {
	111	return dataList;
	112	} else if (length < dataList.size()) {
	113	return dataList.subList(dataList.size() - length, dataList.size());
	114	}
	115
	116	List<ApiPointValueDTO> result = new ArrayList<>();
	117	long start = startTime.getTime();
	118	long end = endTime.getTime();
	119	long oneMin = 1000L * DataPointFreqEnum.getEumByCode(pointDTO.getMinfreqid()).getValue();
	120	long mins = (end - start) / oneMin;
	121
	122	//找出缺少项
	123	Map<Long, Double> sourceDataMap = new HashMap<>(dataList.size());
c844cd	124	for (ApiPointValueDTO pv : dataList) {
潘	125	sourceDataMap.put(pv.getT().getTime(), pv.getV());
	126	}
	127
56dba6	128	Map<Long, Double> dataMap = new LinkedHashMap<>();
潘	129	for (int i = 0; i < mins; i++) {
	130	Long key = start + oneMin * i;
	131	Double value = sourceDataMap.get(key);
	132	dataMap.put(key, value);
	133	}
	134
	135	//补充缺少项
	136	int k = 0;
	137	Map.Entry<Long, Double> lastItem = null;
	138	for (Map.Entry<Long, Double> item : dataMap.entrySet()) {
	139	if (k == 0 && item.getValue() == null) {
	140	item.setValue(getFirstValue(dataMap));
	141	} else if (item.getValue() == null) {
	142	item.setValue(lastItem.getValue());
	143	}
	144	k++;
	145	lastItem = item;
	146
	147	ApiPointValueDTO dataEntity = new ApiPointValueDTO();
	148	dataEntity.setT(new Date(item.getKey()));
	149	dataEntity.setV(item.getValue());
	150	result.add(dataEntity);
	151	}
	152	return result;
	153	}
	154
	155	private Double getFirstValue(Map<Long, Double> dataMap) {
	156	for (Map.Entry<Long, Double> item : dataMap.entrySet()) {
	157	if (item.getValue() != null) {
	158	return item.getValue();
	159	}
	160	}
	161	return null;
	162	}
	163	}