| 提交 | 用户 | 时间 | ||
| 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 | |
| 8bf553 | 45 | * @param planMap |
| 7fd198 | 46 | * @return |
| 潘 | 47 | */ |
| a4891a | 48 | public List<DataValueVO> completionData(int length, List<DataValueVO> dataEntityList, Date startTime, Date endTime, |
| 8bf553 | 49 | String paramId, String paramType, Map<String, ApiPointDTO> pointMap, Map<String, ApiPlanItemDTO> planMap) { |
| b2aca2 | 50 | if (CollectionUtils.isEmpty(dataEntityList) || length == dataEntityList.size()) { |
| 7fd198 | 51 | return dataEntityList; |
| a4891a | 52 | } else if (length < dataEntityList.size()) { |
| 潘 | 53 | return dataEntityList.subList(dataEntityList.size() - length, dataEntityList.size()); |
| 7fd198 | 54 | } |
| 潘 | 55 | |
| b2aca2 | 56 | List<DataValueVO> completionDataEntityList = new ArrayList<>(); |
| D | 57 | long oneMin = 0L; |
| 58 | ||
| 59 | long start = startTime.getTime(); | |
| 60 | long end = endTime.getTime(); | |
| 61 | long mins = 0L; | |
| 62 | ||
| 63 | switch (ModelParamType.getEumByCode(paramType)) { | |
| 19a4fb | 64 | case NORMALITEM: |
| D | 65 | case MERGEITEM: |
| b2aca2 | 66 | // 预测值 |
| c9dd12 | 67 | oneMin = PredGranularityEnum.MIN1.getCode() * 1000L; |
| D | 68 | start = start - (start % oneMin) + oneMin; |
| 69 | end = end - (end % oneMin) + oneMin; | |
| b2aca2 | 70 | mins = ((end - start) / oneMin); |
| D | 71 | break; |
| 72 | case DATAPOINT: | |
| 73 | // 测点值 | |
| 8bf553 | 74 | ApiPointDTO dataPoint = pointMap.get(paramId); |
| b2aca2 | 75 | oneMin = 1000L * DataPointFreqEnum.getEumByCode(dataPoint.getMinfreqid()).getValue(); |
| D | 76 | // 设置时间偏移量 |
| 77 | start = start - (start % oneMin) + oneMin; | |
| 78 | end = end - (end % oneMin) + oneMin; | |
| 79 | mins = ((end - start) / oneMin); | |
| 80 | break; | |
| 81 | case IND: | |
| 82 | // 指标数据 | |
| 83 | oneMin = 24 * 60 * 60 * 1000; | |
| 84 | Calendar calendar2 = Calendar.getInstance(); | |
| 85 | calendar2.setTime(startTime); | |
| a4891a | 86 | calendar2.set(Calendar.HOUR_OF_DAY, 0); |
| 潘 | 87 | calendar2.set(Calendar.MINUTE, 0); |
| 88 | calendar2.set(Calendar.SECOND, 0); | |
| b2aca2 | 89 | start = calendar2.getTime().getTime(); |
| D | 90 | |
| 91 | calendar2.setTime(endTime); | |
| a4891a | 92 | calendar2.set(Calendar.HOUR_OF_DAY, 0); |
| 潘 | 93 | calendar2.set(Calendar.MINUTE, 0); |
| 94 | calendar2.set(Calendar.SECOND, 0); | |
| b2aca2 | 95 | end = calendar2.getTime().getTime(); |
| D | 96 | mins = ((end - start) / oneMin); |
| 97 | break; | |
| 50084d | 98 | case PLAN: |
| D | 99 | // 计划数据 |
| 8bf553 | 100 | ApiPlanItemDTO planItem = planMap.get(paramId); |
| 50084d | 101 | oneMin = 1000L * TimeGranularitySecEnum.getEumByCode(planItem.getTimeGranularity()).getValue(); |
| D | 102 | // 设置时间偏移量 |
| 103 | start = start - (start % oneMin) + oneMin; | |
| 104 | end = end - (end % oneMin) + oneMin; | |
| 105 | mins = ((end - start) / oneMin); | |
| 106 | break; | |
| b2aca2 | 107 | default: |
| D | 108 | break; |
| 109 | } | |
| 7fd198 | 110 | Map<Long, Double> sourceDataMap = new HashMap<>(dataEntityList.size()); |
| 潘 | 111 | for (DataValueVO dataEntity : dataEntityList) { |
| 112 | sourceDataMap.put(dataEntity.getDataTime().getTime(), dataEntity.getDataValue()); | |
| 113 | } | |
| 114 | ||
| 115 | //找出缺少项 | |
| 116 | Map<Long, Double> dataMap = new LinkedHashMap<>(); | |
| a4891a | 117 | for (int i = 0; i < mins; i++) { |
| 7fd198 | 118 | Long key = start + oneMin * i; |
| 潘 | 119 | Double value = sourceDataMap.get(key); |
| 120 | dataMap.put(key, value); | |
| 121 | } | |
| 122 | ||
| 123 | //补充缺少项 | |
| 124 | int k = 0; | |
| 125 | Map.Entry<Long, Double> lastItem = null; | |
| 126 | for (Map.Entry<Long, Double> item : dataMap.entrySet()) { | |
| 127 | if (k == 0 && item.getValue() == null) { | |
| 128 | item.setValue(getFirstValue(dataMap)); | |
| 129 | } else if (item.getValue() == null) { | |
| 130 | item.setValue(lastItem.getValue()); | |
| 131 | } | |
| a4891a | 132 | k++; |
| 7fd198 | 133 | lastItem = item; |
| 潘 | 134 | |
| 135 | DataValueVO dataEntity = new DataValueVO(); | |
| e0b86e | 136 | dataEntity.setDataTime(new Date(item.getKey())); |
| 7fd198 | 137 | dataEntity.setDataValue(item.getValue()); |
| 潘 | 138 | completionDataEntityList.add(dataEntity); |
| 139 | } | |
| 140 | return completionDataEntityList; | |
| 141 | } | |
| 142 | ||
| 143 | /** | |
| 144 | * getFirstValue | |
| 145 | * | |
| 146 | * @param dataMap | |
| 147 | * @return | |
| 148 | */ | |
| 149 | private Double getFirstValue(Map<Long, Double> dataMap) { | |
| 150 | for (Map.Entry<Long, Double> item : dataMap.entrySet()) { | |
| 151 | if (item.getValue() != null) { | |
| 152 | return item.getValue(); | |
| 153 | } | |
| 154 | } | |
| 155 | return null; | |
| 156 | } | |
| 157 | ||
| 158 | } | |
