From 5bf42aa9950058f391805e6fb8d7376f4378924b Mon Sep 17 00:00:00 2001 From: 潘志宝 <979469083@qq.com> Date: 星期一, 23 十二月 2024 16:16:55 +0800 Subject: [PATCH] cumulateHandle.getCurrent --- iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictResultHandler.java | 86 ++++++++++++++++++++++++++---------------- 1 files changed, 53 insertions(+), 33 deletions(-) diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictResultHandler.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictResultHandler.java index b0ab71d..ee15741 100644 --- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictResultHandler.java +++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictResultHandler.java @@ -1,11 +1,9 @@ package com.iailab.module.model.mdk.predict; -import com.iailab.module.data.api.point.DataPointApi; -import com.iailab.module.data.api.point.dto.ApiPointDTO; -import com.iailab.module.data.enums.DataPointFreq; +import com.baomidou.dynamic.datasource.annotation.DSTransactional; +import com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity; import com.iailab.module.model.mcs.pre.service.MmItemResultService; import com.iailab.module.model.mdk.factory.ItemEntityFactory; -import com.iailab.module.model.mdk.vo.MmItemOutputVO; import com.iailab.module.model.mdk.vo.DataValueVO; import com.iailab.module.model.mdk.vo.PredictResultVO; import org.springframework.beans.factory.annotation.Autowired; @@ -23,9 +21,6 @@ @Autowired private ItemEntityFactory itemEntityFactory; - @Autowired - private DataPointApi dataPointApi; - /** * convertToPredictData * @@ -34,36 +29,56 @@ */ public Map<String, List<DataValueVO>> convertToPredictData(PredictResultVO predictResult) { Map<String, List<DataValueVO>> resultMap = new HashMap<>(); - List<MmItemOutputVO> itemOutPutList = itemEntityFactory.getOutPutByItemId(predictResult.getPredictId()); - - if (!CollectionUtils.isEmpty(predictResult.getPredictList())) { - resultMap.put(itemOutPutList.get(0).getId(), predictResult.getPredictList()); - return resultMap; - } - ApiPointDTO point = dataPointApi.getInfoById(itemOutPutList.get(0).getPointId()); - List<Date> dateTimeList = new ArrayList<>(); - Integer rows = predictResult.getPredictMatrix().length; - Integer columns = predictResult.getPredictMatrix()[0].length; - Date tempTime = predictResult.getPredictTime(); - for (Integer i = 0; i < rows; i++) { - dateTimeList.add(tempTime); - Calendar calendar = Calendar.getInstance(); - calendar.setTime(tempTime); - calendar.add(Calendar.SECOND, DataPointFreq.getEumByCode(point.getMinfreqid()).getValue()); - tempTime = calendar.getTime(); - } - for (Integer i = 0; i < columns; i++) { +// List<MmItemOutputEntity> itemOutPutList = itemEntityFactory.getOutPutByItemId(predictResult.getPredictId()); +// +// if (!CollectionUtils.isEmpty(predictResult.getPredictList())) { +// resultMap.put(itemOutPutList.get(0).getId(), predictResult.getPredictList()); +// return resultMap; +// } + Map<com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity, double[]> predictMatrixs = predictResult.getPredictMatrixs(); + HashMap<String,List<DataValueVO>> predictLists = new HashMap<>(); + for (Map.Entry<com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity, double[]> entry : predictMatrixs.entrySet()) { + Integer rows = entry.getValue().length; List<DataValueVO> predictDataList = new ArrayList<>(); - for (Integer j = 0; j < rows; j++) { + Calendar calendar = Calendar.getInstance(); + calendar.setTime(predictResult.getPredictTime()); + for (Integer i = 0; i < rows; i++) { DataValueVO predictData = new DataValueVO(); - predictData.setDataTime(dateTimeList.get(j)); - predictData.setDataValue(Double.valueOf(predictResult.getPredictMatrix()[j][i])); + predictData.setDataTime(calendar.getTime()); + predictData.setDataValue(Double.valueOf(entry.getValue()[i])); predictDataList.add(predictData); + + calendar.add(Calendar.SECOND, predictResult.getGranularity()); } - resultMap.put(itemOutPutList.get(i).getId(), predictDataList); - predictResult.setPredictList(predictDataList); + resultMap.put(entry.getKey().getId(), predictDataList); + predictLists.put(entry.getKey().getResultstr(), predictDataList); } + predictResult.setPredictLists(predictLists); return resultMap; + } + + public Map<String, List<DataValueVO>> convertToPredictData2(PredictResultVO predictResult) { + Map<String, List<DataValueVO>> predictLists = new HashMap<>(); + if (!CollectionUtils.isEmpty(predictResult.getPredictList())) { + return predictLists; + } + Map<com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity, double[]> predictMatrixs = predictResult.getPredictMatrixs(); + for (Map.Entry<com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity, double[]> entry : predictMatrixs.entrySet()) { + Integer rows = entry.getValue().length; + List<DataValueVO> predictDataList = new ArrayList<>(); + Calendar calendar = Calendar.getInstance(); + calendar.setTime(predictResult.getPredictTime()); + for (Integer i = 0; i < rows; i++) { + DataValueVO predictData = new DataValueVO(); + predictData.setDataTime(calendar.getTime()); + predictData.setDataValue(Double.valueOf(entry.getValue()[i])); + predictDataList.add(predictData); + + calendar.add(Calendar.SECOND, predictResult.getGranularity()); + } + predictLists.put(entry.getKey().getResultstr(), predictDataList); + } + return predictLists; } /** @@ -72,19 +87,24 @@ * @param predictResult */ @Async + @DSTransactional public void savePredictResult(PredictResultVO predictResult) { Map<String, List<DataValueVO>> resultMap = convertToPredictData(predictResult); mmItemResultService.savePredictValue(resultMap, predictResult.getLt(), "n", predictResult.getPredictTime()); + // 存double类型输出 + if (!CollectionUtils.isEmpty(predictResult.getPredictDoubleValues())) { + mmItemResultService.savePredictValue(predictResult.getPredictDoubleValues(), predictResult.getPredictTime()); + } } public List<DataValueVO> getPredictValueByItemNo(String itemNo, Date start, Date end) { String itemId = itemEntityFactory.getItemByItemNo(itemNo).getId(); - List<MmItemOutputVO> outputList = itemEntityFactory.getOutPutByItemId(itemId); + List<MmItemOutputEntity> outputList = itemEntityFactory.getOutPutByItemId(itemId); return mmItemResultService.getPredictValue(outputList.get(0).getId(), start, end); } public List<DataValueVO> getPredictValueByItemId(String itemId, Date start, Date end) { - List<MmItemOutputVO> outputList = itemEntityFactory.getOutPutByItemId(itemId); + List<MmItemOutputEntity> outputList = itemEntityFactory.getOutPutByItemId(itemId); return mmItemResultService.getPredictValue(outputList.get(0).getId(), start, end); } } -- Gitblit v1.9.3