工业互联网平台
查看 | 历史 | 提交 | 提交对比
潘志宝
2024-11-29 56dba6294342aa571197298669d04618b4f7258f
提交 | 用户 | 时间
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;
12 import lombok.extern.slf4j.Slf4j;
13 import org.springframework.beans.factory.annotation.Autowired;
14 import org.springframework.stereotype.Component;
15 import org.springframework.util.CollectionUtils;
16
17 import java.math.BigDecimal;
18 import java.util.*;
19
20 /**
21  * 累计点处理
22  *
23  * @author PanZhibao
24  * @Description
25  * @createTime 2024年11月29日
26  */
27 @Slf4j
28 @Component
29 public class CumulateHandle {
30
31     @Autowired
32     private DataPointApi dataPointApi;
33
34     public List<InfluxPointValuePOJO> handle(Date collectTime, List<DaPointDTO> dtos) {
35         List<InfluxPointValuePOJO> result = new ArrayList<>();
36         try {
37             log.info("累计点处理开始");
38             if (CollectionUtils.isEmpty(dtos)) {
39                 return result;
40             }
41             dtos.forEach(dto -> {
42                 try {
43                     Object value = singleCompute(dto, collectTime);
44                     InfluxPointValuePOJO pojo = GenInfluxPointValueUtils.getByPoint(dto, value);
45                     pojo.setTimestamp(collectTime.toInstant());
46                     result.add(pojo);
47                 } catch (Exception ex) {
48                     ex.printStackTrace();
49                     log.info("累计点异常！PointNo=" + dto.getPointNo());
50                 }
51             });
52
53         } catch (Exception ex) {
54             ex.printStackTrace();
55             log.info("累计点处理异常！");
56         }
57         return result;
58     }
59
60
61     private Object singleCompute(DaPointDTO dto, Date collectTime) {
62         ApiPointDTO pointDTO = dataPointApi.getInfoByNo(dto.getMomentPoint());
63         if (pointDTO == null) {
64             return CommonConstant.BAD_VALUE;
65         }
66         Calendar calendar = Calendar.getInstance();
67         calendar.setTime(collectTime);
68         calendar.add(Calendar.MINUTE, -1);
69         Date endTime = calendar.getTime();
70         calendar.add(Calendar.MINUTE, dto.getLength() * -1);
71         Date startTime = calendar.getTime();
72         ApiPointValueQueryDTO queryDto = new ApiPointValueQueryDTO();
73         queryDto.setStart(startTime);
74         queryDto.setEnd(endTime);
75         queryDto.setPointNo(dto.getMomentPoint());
76
77         List<ApiPointValueDTO> dataList = dataPointApi.queryPointHistoryValue(queryDto);
78         if (CollectionUtils.isEmpty(dataList)) {
79             return BigDecimal.ZERO;
80         } else if (dataList.size() < dto.getLength()) {
81             // 补全数据
82             dataList = completionData(dto.getLength(), dataList, startTime, endTime, pointDTO);
83         }
84         double total = dataList.stream().mapToDouble(ApiPointValueDTO::getV).sum();
85         return new BigDecimal(total).divide(new BigDecimal(dto.getDivisor()), 2, BigDecimal.ROUND_HALF_UP);
86     }
87
88     private List<ApiPointValueDTO> completionData(int length, List<ApiPointValueDTO> dataList, Date startTime, Date endTime, ApiPointDTO pointDTO) {
89         if (CollectionUtils.isEmpty(dataList) || length == dataList.size()) {
90             return dataList;
91         } else if (length < dataList.size()) {
92             return dataList.subList(dataList.size() - length, dataList.size());
93         }
94
95         List<ApiPointValueDTO> result = new ArrayList<>();
96         long start = startTime.getTime();
97         long end = endTime.getTime();
98         long oneMin = 1000L * DataPointFreqEnum.getEumByCode(pointDTO.getMinfreqid()).getValue();
99         long mins = (end - start) / oneMin;
100
101         //找出缺少项
102         Map<Long, Double> sourceDataMap = new HashMap<>(dataList.size());
103         Map<Long, Double> dataMap = new LinkedHashMap<>();
104         for (int i = 0; i < mins; i++) {
105             Long key = start + oneMin * i;
106             Double value = sourceDataMap.get(key);
107             dataMap.put(key, value);
108         }
109
110         //补充缺少项
111         int k = 0;
112         Map.Entry<Long, Double> lastItem = null;
113         for (Map.Entry<Long, Double> item : dataMap.entrySet()) {
114             if (k == 0 && item.getValue() == null) {
115                 item.setValue(getFirstValue(dataMap));
116             } else if (item.getValue() == null) {
117                 item.setValue(lastItem.getValue());
118             }
119             k++;
120             lastItem = item;
121
122             ApiPointValueDTO dataEntity = new ApiPointValueDTO();
123             dataEntity.setT(new Date(item.getKey()));
124             dataEntity.setV(item.getValue());
125             result.add(dataEntity);
126         }
127         return result;
128     }
129
130     private Double getFirstValue(Map<Long, Double> dataMap) {
131         for (Map.Entry<Long, Double> item : dataMap.entrySet()) {
132             if (item.getValue() != null) {
133                 return item.getValue();
134             }
135         }
136         return null;
137     }
138 }