| 提交 | 用户 | 时间 | ||
| 7fd198 | 1 | package com.iailab.module.model.mdk.sample; |
| 潘 | 2 | |
| 75a848 | 3 | import com.iailab.module.data.api.ind.IndItemApi; |
| D | 4 | import com.iailab.module.data.api.ind.dto.ApiIndItemDTO; |
| 8bf553 | 5 | import com.iailab.module.data.api.plan.PlanItemApi; |
| D | 6 | import com.iailab.module.data.api.plan.dto.ApiPlanItemDTO; |
| 07890e | 7 | import com.iailab.module.data.api.point.DataPointApi; |
| D | 8 | import com.iailab.module.data.api.point.dto.ApiPointDTO; |
| 7fd198 | 9 | import com.iailab.module.model.mcs.pre.entity.MmModelParamEntity; |
| 91343d | 10 | import com.iailab.module.model.mcs.pre.service.MmModelParamService; |
| 潘 | 11 | import com.iailab.module.model.mcs.pre.service.MmPredictItemService; |
| 7fd198 | 12 | import com.iailab.module.model.mcs.pre.service.MmPredictModelService; |
| 07890e | 13 | import com.iailab.module.model.mdk.common.enums.ModelParamType; |
| 7fd198 | 14 | import com.iailab.module.model.mdk.sample.dto.ColumnItem; |
| 潘 | 15 | import com.iailab.module.model.mdk.sample.dto.ColumnItemPort; |
| 50084d | 16 | import com.iailab.module.model.mdk.sample.dto.SampleInfo; |
| 7fd198 | 17 | import org.springframework.beans.factory.annotation.Autowired; |
| 潘 | 18 | import org.springframework.stereotype.Component; |
| 19 | import org.springframework.util.CollectionUtils; | |
| 20 | ||
| 50084d | 21 | import java.util.*; |
| 07890e | 22 | import java.util.function.Function; |
| D | 23 | import java.util.stream.Collectors; |
| 7fd198 | 24 | |
| 潘 | 25 | /** |
| 26 | * @author PanZhibao | |
| 27 | * @Description | |
| 28 | * @createTime 2024年09月03日 | |
| 29 | */ | |
| 30 | @Component | |
| 31 | public class PredictSampleInfoConstructor extends SampleInfoConstructor { | |
| 32 | ||
| 33 | @Autowired | |
| 34 | private MmPredictModelService mmPredictModelService; | |
| 35 | ||
| 36 | @Autowired | |
| 91343d | 37 | private MmModelParamService mmModelParamService; |
| 7fd198 | 38 | |
| 潘 | 39 | @Autowired |
| 91343d | 40 | private MmPredictItemService mmPredictItemService; |
| 07890e | 41 | |
| D | 42 | @Autowired |
| 43 | private DataPointApi dataPointApi; | |
| 7fd198 | 44 | |
| 8bf553 | 45 | @Autowired |
| D | 46 | private PlanItemApi planItemApi; |
| 47 | ||
| 75a848 | 48 | @Autowired |
| D | 49 | private IndItemApi indItemApi; |
| 50 | ||
| 7fd198 | 51 | /** |
| 潘 | 52 | * 返回样本矩阵的列数 |
| 53 | * | |
| 54 | * @param modelId | |
| 55 | * @return | |
| 56 | */ | |
| 57 | @Override | |
| 58 | protected Integer getSampleColumn(String modelId) { | |
| 59 | return mmPredictModelService.getSampleLength(modelId).intValue(); | |
| 60 | } | |
| 61 | ||
| 62 | /** | |
| 63 | * 样本的列信息 | |
| 64 | * | |
| 65 | * @param modelId | |
| 66 | * @param predictTime | |
| 67 | * @return | |
| 68 | */ | |
| 69 | @Override | |
| bab433 | 70 | protected SampleInfo getColumnInfo(String modelId, Date predictTime, Map<Integer, Integer> dynamicDataLength) { |
| 50084d | 71 | SampleInfo sampleInfo = new SampleInfo(); |
| 7fd198 | 72 | List<ColumnItemPort> resultList = new ArrayList<>(); |
| 潘 | 73 | List<ColumnItem> columnItemList = new ArrayList<>(); |
| 74 | ColumnItem columnInfo = new ColumnItem(); | |
| 75 | ColumnItemPort curPort = new ColumnItemPort(); //当前端口 | |
| 45520a | 76 | List<MmModelParamEntity> modelInputParamEntityList = mmModelParamService.getByModelidFromCache(modelId); |
| 7fd198 | 77 | if (CollectionUtils.isEmpty(modelInputParamEntityList)) { |
| 潘 | 78 | return null; |
| 79 | } | |
| 80 | //设置当前端口号，初始值为最小端口（查询结果按端口号从小到达排列） | |
| 81 | int curPortOrder = modelInputParamEntityList.get(0).getModelparamportorder(); | |
| 82 | //设置当前查询数据长度，初始值为最小端口数据长度 | |
| bab433 | 83 | int curDataLength = super.getDataLength(dynamicDataLength, curPortOrder, modelInputParamEntityList.get(0).getDatalength()); |
| 07890e | 84 | // 统一获取测点的信息 |
| 50084d | 85 | Set<String> pointIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.DATAPOINT)).map(MmModelParamEntity::getModelparamid).collect(Collectors.toSet()); |
| 07890e | 86 | List<ApiPointDTO> points = dataPointApi.getInfoByIds(pointIds); |
| bab433 | 87 | Map<String, ApiPointDTO> pointMap = points.stream().collect(Collectors.toMap(ApiPointDTO::getId, Function.identity(), (e1, e2) -> e1)); |
| 8bf553 | 88 | // 统一获取计划数据的信息 |
| D | 89 | Set<String> planIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.PLAN)).map(MmModelParamEntity::getModelparamid).collect(Collectors.toSet()); |
| 90 | List<ApiPlanItemDTO> plans = planItemApi.getInfoByIds(planIds); | |
| bab433 | 91 | Map<String, ApiPlanItemDTO> planMap = plans.stream().collect(Collectors.toMap(ApiPlanItemDTO::getId, Function.identity(), (e1, e2) -> e1)); |
| 75a848 | 92 | // 统一获取指标数据的信息 |
| 1ab119 | 93 | Set<String> indIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.IND) || ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.IND_ASCII)).map(MmModelParamEntity::getModelparamid).collect(Collectors.toSet()); |
| 75a848 | 94 | List<ApiIndItemDTO> inds = indItemApi.getInfoByIds(indIds); |
| D | 95 | Map<String, ApiIndItemDTO> indMap = inds.stream().collect(Collectors.toMap(ApiIndItemDTO::getId, Function.identity(), (e1, e2) -> e1)); |
| 07890e | 96 | |
| 7fd198 | 97 | for (MmModelParamEntity entry : modelInputParamEntityList) { |
| 潘 | 98 | columnInfo.setParamType(entry.getModelparamtype()); |
| 1a2b62 | 99 | columnInfo.setParamId(entry.getModelparamid()); |
| bab433 | 100 | columnInfo.setDataLength(super.getDataLength(dynamicDataLength, entry.getModelparamportorder(), entry.getDatalength())); |
| 7fd198 | 101 | columnInfo.setModelParamOrder(entry.getModelparamorder()); |
| 潘 | 102 | columnInfo.setModelParamPortOrder(entry.getModelparamportorder()); |
| 75a848 | 103 | columnInfo.setGranularity(super.getGranularity(columnInfo, pointMap, planMap,indMap)); |
| e691b9 | 104 | columnInfo.setStartTime(getStartTime(columnInfo, predictTime)); |
| D | 105 | columnInfo.setEndTime(getEndTime(columnInfo, predictTime)); |
| 7fd198 | 106 | |
| 潘 | 107 | //对每一个爪进行数据项归并 |
| bab433 | 108 | if (curPortOrder != entry.getModelparamportorder()) { |
| 7fd198 | 109 | //当数据项端口号不为当前端口号时，封装上一个端口类，操作下一个端口类 |
| 潘 | 110 | curPort.setColumnItemList(columnItemList); |
| 111 | curPort.setDataLength(curDataLength); | |
| 112 | curPort.setPortOrder(curPortOrder); | |
| 113 | resultList.add(curPort); | |
| 114 | curPort = new ColumnItemPort(); //对象重新初始化，防止引用拷贝导致数据覆盖 | |
| 115 | //封装上一个端口类后更新当前的各个参数 | |
| 116 | columnItemList = new ArrayList<>(); | |
| 117 | curDataLength = entry.getDatalength(); | |
| 118 | curPortOrder = entry.getModelparamportorder(); | |
| 119 | } | |
| 120 | columnItemList.add(columnInfo); | |
| 121 | columnInfo = new ColumnItem(); //对象重新初始化，防止引用拷贝导致数据覆盖 | |
| 122 | } | |
| 123 | //当迭代到最后一个项的时候，封装最后一个端口的信息 | |
| 124 | curPort.setColumnItemList(columnItemList); | |
| 125 | curPort.setDataLength(curDataLength); | |
| 126 | curPort.setPortOrder(curPortOrder); | |
| 127 | resultList.add(curPort); | |
| 50084d | 128 | sampleInfo.setColumnInfo(resultList); |
| D | 129 | sampleInfo.setPointMap(pointMap); |
| 8bf553 | 130 | sampleInfo.setPlanMap(planMap); |
| 75a848 | 131 | sampleInfo.setIndMap(indMap); |
| 50084d | 132 | return sampleInfo; |
| 7fd198 | 133 | } |
| 潘 | 134 | |
| 135 | /** | |
| 136 | * 样本的采样周期 | |
| 137 | * | |
| 138 | * @param modelId | |
| 139 | * @return | |
| 140 | */ | |
| 141 | @Override | |
| 142 | protected Integer getSampleCycle(String modelId) { | |
| b82ba2 | 143 | return mmPredictItemService.getItemByIdFromCache(mmPredictModelService.getInfoFromCatch(modelId).getItemid()).getGranularity(); |
| 7fd198 | 144 | } |
| 潘 | 145 | |
| 146 | ||
| 147 | } | |
