From 1178da30ca701da465bf7bc1342b539b2df03c7d Mon Sep 17 00:00:00 2001
From: dengzedong <dengzedong@email>
Date: 星期五, 20 十二月 2024 13:52:35 +0800
Subject: [PATCH] predictItem MODELRESULTERROR

---
 iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictModuleHandler.java |   28 +++++++++++++++++++---------
 1 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictModuleHandler.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictModuleHandler.java
index bf5be31..034a6c8 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictModuleHandler.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/PredictModuleHandler.java
@@ -1,5 +1,6 @@
 package com.iailab.module.model.mdk.predict;
 
+import com.iailab.module.model.common.exception.ModelResultErrorException;
 import com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity;
 import com.iailab.module.model.mcs.pre.enums.ItemRunStatusEnum;
 import com.iailab.module.model.mcs.pre.enums.ItemStatus;
@@ -72,28 +73,37 @@
                 long start = System.currentTimeMillis();
                 try {
                     // 预测项开始预测
-                    predictResult = predictItemHandler.predict(predictTime, predictItem, predictValueMap,itemRunStatusEnum);
+                    predictResult = predictItemHandler.predict(predictTime, predictItem, predictValueMap);
+                } catch (ModelResultErrorException e) {
+                    itemRunStatusEnum = ItemRunStatusEnum.MODELRESULTERROR;
+                    continue;
                 } catch (Exception e) {
+                    itemRunStatusEnum = ItemRunStatusEnum.FAIL;
                     continue;
                 }
                 long end = System.currentTimeMillis();
                 Long drtPre = end - start;
                 log.info(MessageFormat.format("预测项:{0},预测时间:{1}ms", predictItem.getItemName(), drtPre));
                 totalDur = totalDur + drtPre;
-
                 predictResult.setGranularity(predictItem.getGranularity());
                 predictResult.setT(intervalTime);
                 predictResult.setSaveIndex(predictItem.getSaveIndex());
                 predictResult.setLt(1);
+                predictResultMap.put(predictItem.getItemNo(), predictResult);
 
                 // 保存预测结果
-                predictResultHandler.savePredictResult(predictResult);
-                long endSave = System.currentTimeMillis();
-                Long drtSave = endSave - end;
-                log.info(MessageFormat.format("预测项:{0},保存时间:{1}ms", predictItem.getItemName(),
-                        drtSave));
-                totalDur = totalDur + drtSave;
-                predictResultMap.put(predictItem.getItemNo(), predictResult);
+                try {
+                    predictResultHandler.savePredictResult(predictResult);
+                } catch (Exception e) {
+                    itemRunStatusEnum = ItemRunStatusEnum.MODELRESULTSAVEERROR;
+                    throw new RuntimeException("模型结果保存异常,result:" + predictResult);
+                }
+                itemRunStatusEnum = ItemRunStatusEnum.SUCCESS;
+//                long endSave = System.currentTimeMillis();
+//                Long drtSave = endSave - end;
+//                log.info(MessageFormat.format("预测项:{0},保存时间:{1}ms", predictItem.getItemName(),
+//                        drtSave));
+//                totalDur = totalDur + drtSave;
             } catch (Exception e) {
                 e.printStackTrace();
                 log.error(MessageFormat.format("预测项编号:{0},预测项名称:{1},预测失败:{2} 预测时刻:{3}",

--
Gitblit v1.9.3