工业互联网平台
查看 | 历史 | 提交 | 提交对比
潘志宝
20 小时以前 271fed395d4d3e4ab8cdb44ccf466859ba485e46
提交 | 用户 | 时间
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;
c9dd12 9 import com.iailab.module.model.mcs.pre.enums.PredGranularityEnum;
b2aca2 10 import com.iailab.module.model.mdk.common.enums.ModelParamType;
7fd198 11 import com.iailab.module.model.mdk.sample.dto.SampleData;
潘 12 import com.iailab.module.model.mdk.sample.dto.SampleInfo;
13 import com.iailab.module.model.mdk.vo.DataValueVO;
14 import org.slf4j.Logger;
15 import org.slf4j.LoggerFactory;
b2aca2 16 import org.springframework.beans.factory.annotation.Autowired;
7fd198 17 import org.springframework.util.CollectionUtils;
潘 18
19 import java.util.*;
20
21 abstract class SampleDataConstructor {
22
23     private Logger logger = LoggerFactory.getLogger(getClass());
b2aca2 24
D 25     @Autowired
26     private DataPointApi dataPointApi;
50084d 27     @Autowired
D 28     private PlanItemApi planItemApi;
7fd198 29
潘 30     /**
31      * prepareSampleData
32      *
33      * @param sampleInfo
34      * @return
35      */
50084d 36     public abstract List<SampleData> prepareSampleData(SampleInfo sampleInfo) throws Exception;
7fd198 37
潘 38     /**
39      * 补全数据
40      *
41      * @param length
42      * @param dataEntityList
43      * @param startTime
44      * @param endTime
45      * @return
46      */
e691b9 47     public List<DataValueVO> completionData(int length, List<DataValueVO> dataEntityList, Date startTime, Date endTime, String paramType, Integer granularity) {
b2aca2 48         if (CollectionUtils.isEmpty(dataEntityList) || length == dataEntityList.size()) {
7fd198 49             return dataEntityList;
a4891a 50         } else if (length < dataEntityList.size()) {
潘 51             return dataEntityList.subList(dataEntityList.size() - length, dataEntityList.size());
7fd198 52         }
潘 53
b2aca2 54         List<DataValueVO> completionDataEntityList = new ArrayList<>();
D 55         long oneMin = 0L;
56
57         long start = startTime.getTime();
58         long end = endTime.getTime();
59         long mins = 0L;
60
61         switch (ModelParamType.getEumByCode(paramType)) {
19a4fb 62             case NORMALITEM:
D 63             case MERGEITEM:
b2aca2 64                 // 预测值
e691b9 65                 oneMin = granularity * 1000L;
9f09d4 66                 start = start - (start % oneMin);
D 67                 end = end - (end % oneMin);
2e0a3c 68                 mins = ((end - start) / oneMin);
b2aca2 69                 break;
D 70             case DATAPOINT:
e691b9 71             case PLAN:
b2aca2 72                 // 测点值
e691b9 73                 oneMin = 1000L * granularity;
b2aca2 74                 // 设置时间偏移量
D 75                 start = start - (start % oneMin) + oneMin;
76                 end = end - (end % oneMin) + oneMin;
77                 mins = ((end - start) / oneMin);
78                 break;
79             case IND:
80                 // 指标数据
81                 oneMin = 24 * 60 * 60 * 1000;
82                 Calendar calendar2 = Calendar.getInstance();
83                 calendar2.setTime(startTime);
a4891a 84                 calendar2.set(Calendar.HOUR_OF_DAY, 0);
潘 85                 calendar2.set(Calendar.MINUTE, 0);
86                 calendar2.set(Calendar.SECOND, 0);
b2aca2 87                 start = calendar2.getTime().getTime();
D 88
89                 calendar2.setTime(endTime);
a4891a 90                 calendar2.set(Calendar.HOUR_OF_DAY, 0);
潘 91                 calendar2.set(Calendar.MINUTE, 0);
92                 calendar2.set(Calendar.SECOND, 0);
b2aca2 93                 end = calendar2.getTime().getTime();
50084d 94                 mins = ((end - start) / oneMin);
D 95                 break;
b2aca2 96             default:
D 97                 break;
98         }
7fd198 99         Map<Long, Double> sourceDataMap = new HashMap<>(dataEntityList.size());
潘 100         for (DataValueVO dataEntity : dataEntityList) {
101             sourceDataMap.put(dataEntity.getDataTime().getTime(), dataEntity.getDataValue());
102         }
103
104         //找出缺少项
105         Map<Long, Double> dataMap = new LinkedHashMap<>();
a4891a 106         for (int i = 0; i < mins; i++) {
7fd198 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             }
a4891a 121             k++;
7fd198 122             lastItem = item;
潘 123
124             DataValueVO dataEntity = new DataValueVO();
e0b86e 125             dataEntity.setDataTime(new Date(item.getKey()));
7fd198 126             dataEntity.setDataValue(item.getValue());
潘 127             completionDataEntityList.add(dataEntity);
128         }
129         return completionDataEntityList;
130     }
131
132     /**
133      * getFirstValue
134      *
135      * @param dataMap
136      * @return
137      */
138     private Double getFirstValue(Map<Long, Double> dataMap) {
139         for (Map.Entry<Long, Double> item : dataMap.entrySet()) {
140             if (item.getValue() != null) {
141                 return item.getValue();
142             }
143         }
144         return null;
145     }
146
147 }