| | |
| | | resp.setScheduleTime(reqDTO.getScheduleTime()); |
| | | try { |
| | | log.info("调度计算开始: " + System.currentTimeMillis()); |
| | | log.info("reqDTO=" + JSON.toJSONString(reqDTO)); |
| | | ScheduleResultVO scheduleResult = scheduleModelHandler.doSchedule(reqDTO.getScheduleCode(), reqDTO.getScheduleTime(), |
| | | reqDTO.getDynamicDataLength(), reqDTO.getDynamicSettings()); |
| | | resp.setStatusCode(scheduleResult.getResultCode()); |
| | |
| | | * @return |
| | | */ |
| | | @Override |
| | | public List<SampleData> prepareSampleData(SampleInfo sampleInfo) throws Exception { |
| | | public List<SampleData> prepareSampleData(SampleInfo sampleInfo) throws Exception { |
| | | List<SampleData> sampleDataList = new ArrayList<>(); |
| | | Map<String, ApiPointDTO> pointMap = sampleInfo.getPointMap(); |
| | | Map<String, ApiPlanItemDTO> planMap = sampleInfo.getPlanMap(); |
| | |
| | | //对每一项依次进行数据查询,然后将查询出的值赋给matrix对应的位置 |
| | | for (int i = 0; i < entry.getColumnItemList().size(); i++) { |
| | | try { |
| | | List<DataValueVO> dataEntityList = getData(entry.getColumnItemList().get(i),pointMap,planMap); |
| | | List<DataValueVO> dataEntityList = getData(entry.getColumnItemList().get(i), pointMap, planMap); |
| | | //补全数据 |
| | | ColumnItem columnItem = entry.getColumnItemList().get(i); |
| | | dataEntityList = super.completionData(matrix.length, dataEntityList, columnItem.startTime, columnItem.endTime, |
| | | columnItem.paramId, columnItem.getParamType(),pointMap,planMap); |
| | | columnItem.paramId, columnItem.getParamType(), pointMap, planMap); |
| | | |
| | | /** 如果数据取不满,把缺失的数据点放在后面 */ |
| | | if (dataEntityList != null && dataEntityList.size() != 0) { |
| | |
| | | * @return |
| | | */ |
| | | @Override |
| | | protected SampleInfo getColumnInfo(String modelId, Date predictTime) { |
| | | protected SampleInfo getColumnInfo(String modelId, Date predictTime, Map<Integer, Integer> dynamicDataLength) { |
| | | SampleInfo sampleInfo = new SampleInfo(); |
| | | List<ColumnItemPort> resultList = new ArrayList<>(); |
| | | List<ColumnItem> columnItemList = new ArrayList<>(); |
| | |
| | | //设置当前端口号,初始值为最小端口(查询结果按端口号从小到达排列) |
| | | int curPortOrder = modelInputParamEntityList.get(0).getModelparamportorder(); |
| | | //设置当前查询数据长度,初始值为最小端口数据长度 |
| | | int curDataLength = modelInputParamEntityList.get(0).getDatalength(); |
| | | int curDataLength = super.getDataLength(dynamicDataLength, curPortOrder, modelInputParamEntityList.get(0).getDatalength()); |
| | | // 统一获取测点的信息 |
| | | Set<String> pointIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.DATAPOINT)).map(MmModelParamEntity::getModelparamid).collect(Collectors.toSet()); |
| | | List<ApiPointDTO> points = dataPointApi.getInfoByIds(pointIds); |
| | | Map<String, ApiPointDTO> pointMap = points.stream().collect(Collectors.toMap(ApiPointDTO::getId, Function.identity(), (e1,e2) -> e1)); |
| | | Map<String, ApiPointDTO> pointMap = points.stream().collect(Collectors.toMap(ApiPointDTO::getId, Function.identity(), (e1, e2) -> e1)); |
| | | // 统一获取计划数据的信息 |
| | | Set<String> planIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.PLAN)).map(MmModelParamEntity::getModelparamid).collect(Collectors.toSet()); |
| | | List<ApiPlanItemDTO> plans = planItemApi.getInfoByIds(planIds); |
| | | Map<String, ApiPlanItemDTO> planMap = plans.stream().collect(Collectors.toMap(ApiPlanItemDTO::getId, Function.identity(), (e1,e2) -> e1)); |
| | | Map<String, ApiPlanItemDTO> planMap = plans.stream().collect(Collectors.toMap(ApiPlanItemDTO::getId, Function.identity(), (e1, e2) -> e1)); |
| | | |
| | | for (MmModelParamEntity entry : modelInputParamEntityList) { |
| | | columnInfo.setParamType(entry.getModelparamtype()); |
| | | columnInfo.setParamId(entry.getModelparamid()); |
| | | columnInfo.setDataLength(entry.getDatalength()); |
| | | columnInfo.setDataLength(super.getDataLength(dynamicDataLength, entry.getModelparamportorder(), entry.getDatalength())); |
| | | columnInfo.setModelParamOrder(entry.getModelparamorder()); |
| | | columnInfo.setModelParamPortOrder(entry.getModelparamportorder()); |
| | | columnInfo.setStartTime(getStartTime(columnInfo, predictTime,pointMap,planMap)); |
| | | columnInfo.setEndTime(getEndTime(columnInfo, predictTime,pointMap,planMap)); |
| | | columnInfo.setGranularity(super.getGranularity(columnInfo,pointMap,planMap)); |
| | | columnInfo.setStartTime(getStartTime(columnInfo, predictTime, pointMap, planMap)); |
| | | columnInfo.setEndTime(getEndTime(columnInfo, predictTime, pointMap, planMap)); |
| | | columnInfo.setGranularity(super.getGranularity(columnInfo, pointMap, planMap)); |
| | | |
| | | //对每一个爪进行数据项归并 |
| | | if (curPortOrder != entry.getModelparamportorder()){ |
| | | if (curPortOrder != entry.getModelparamportorder()) { |
| | | //当数据项端口号不为当前端口号时,封装上一个端口类,操作下一个端口类 |
| | | curPort.setColumnItemList(columnItemList); |
| | | curPort.setDataLength(curDataLength); |
| | |
| | | @Autowired |
| | | private SampleFactory sampleFactory; |
| | | |
| | | public List<SampleData> constructSample(String typeA, String modelId, Date runTime,String itemName, Map<Integer, Integer> dynamicDataLength) throws ModelInvokeException { |
| | | public List<SampleData> constructSample(String typeA, String modelId, Date runTime,String itemName, |
| | | Map<Integer, Integer> dynamicDataLength) throws ModelInvokeException { |
| | | try { |
| | | SampleInfoConstructor sampleInfoConstructor = sampleFactory.createSampleInfo(typeA, modelId); |
| | | SampleInfo sampleInfo = sampleInfoConstructor.prepareSampleInfo(modelId, runTime); |
| | | if (!CollectionUtils.isEmpty(dynamicDataLength) && sampleInfo.getColumnInfo() != null) { |
| | | for(ColumnItemPort columnInfo : sampleInfo.getColumnInfo()) { |
| | | if (!dynamicDataLength.containsKey(columnInfo.getPortOrder())) { |
| | | continue; |
| | | } |
| | | columnInfo.setDataLength(dynamicDataLength.get(columnInfo.getPortOrder())); |
| | | columnInfo.getColumnItemList().forEach(columnItem -> { |
| | | columnItem.setDataLength(columnInfo.getDataLength()); |
| | | }); |
| | | } |
| | | } |
| | | SampleInfo sampleInfo = sampleInfoConstructor.prepareSampleInfo(modelId, runTime, dynamicDataLength); |
| | | SampleDataConstructor sampleDataConstructor = sampleFactory.createSampleData(typeA); |
| | | return sampleDataConstructor.prepareSampleData(sampleInfo); |
| | | } catch (Exception e) { |
| | |
| | | import com.iailab.module.model.mdk.sample.dto.ColumnItem; |
| | | import com.iailab.module.model.mdk.sample.dto.SampleInfo; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.util.CollectionUtils; |
| | | |
| | | import java.util.Calendar; |
| | | import java.util.Date; |
| | |
| | | * @param predictTime |
| | | * @return |
| | | */ |
| | | protected SampleInfo prepareSampleInfo(String modelId, Date predictTime) { |
| | | protected SampleInfo prepareSampleInfo(String modelId, Date predictTime, Map<Integer, Integer> dynamicDataLength) { |
| | | //样本的列信息 |
| | | return getColumnInfo(modelId, predictTime); |
| | | return getColumnInfo(modelId, predictTime, dynamicDataLength); |
| | | } |
| | | |
| | | /** |
| | |
| | | * @param predictTime |
| | | * @return |
| | | */ |
| | | protected abstract SampleInfo getColumnInfo(String modelId, Date predictTime); |
| | | protected abstract SampleInfo getColumnInfo(String modelId, Date predictTime, Map<Integer, Integer> dynamicDataLength); |
| | | |
| | | /** |
| | | * 样本的采样周期 |
| | |
| | | calendar.add(Calendar.SECOND, timeLength * granularity); |
| | | return calendar.getTime(); |
| | | } |
| | | |
| | | protected int getDataLength(Map<Integer, Integer> dynamicDataLength, Integer port, Integer dataLength) { |
| | | if (CollectionUtils.isEmpty(dynamicDataLength)) { |
| | | return dataLength; |
| | | } |
| | | if (dynamicDataLength.containsKey(port)) { |
| | | return dynamicDataLength.get(port); |
| | | } |
| | | return dataLength; |
| | | } |
| | | } |
| | |
| | | } |
| | | |
| | | @Override |
| | | protected SampleInfo getColumnInfo(String modelId, Date predictTime) { |
| | | protected SampleInfo getColumnInfo(String modelId, Date predictTime, Map<Integer, Integer> dynamicDataLength) { |
| | | SampleInfo sampleInfo = new SampleInfo(); |
| | | List<ColumnItemPort> resultList = new ArrayList<>(); |
| | | List<ColumnItem> columnItemList = new ArrayList<>(); |
| | |
| | | //设置当前端口号,初始值为最小端口(查询结果按端口号从小到达排列) |
| | | int curPortOrder = modelInputParamEntityList.get(0).getModelparamportorder(); |
| | | //设置当前查询数据长度,初始值为最小端口数据长度 |
| | | int curDataLength = modelInputParamEntityList.get(0).getDatalength(); |
| | | int curDataLength = super.getDataLength(dynamicDataLength, curPortOrder, modelInputParamEntityList.get(0).getDatalength()); |
| | | // 统一获取测点的信息 |
| | | Set<String> pointIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.DATAPOINT)).map(StScheduleModelParamEntity::getModelparamid).collect(Collectors.toSet()); |
| | | List<ApiPointDTO> points = dataPointApi.getInfoByIds(pointIds); |
| | |
| | | for (StScheduleModelParamEntity entry : modelInputParamEntityList) { |
| | | columnInfo.setParamType(entry.getModelparamtype()); |
| | | columnInfo.setParamId(entry.getModelparamid()); |
| | | columnInfo.setDataLength(entry.getDatalength()); |
| | | columnInfo.setDataLength(super.getDataLength(dynamicDataLength, entry.getModelparamportorder(), entry.getDatalength())); |
| | | columnInfo.setModelParamOrder(entry.getModelparamorder()); |
| | | columnInfo.setModelParamPortOrder(entry.getModelparamportorder()); |
| | | columnInfo.setStartTime(getStartTime(columnInfo, predictTime,pointMap,planMap)); |
| | |
| | | return null; |
| | | } |
| | | |
| | | |
| | | |
| | | } |