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