From 0a2b23ad3f30dfb01c5d590fb98f39e93bfe1932 Mon Sep 17 00:00:00 2001
From: dongyukun <1208714201@qq.com>
Date: 星期二, 31 十二月 2024 18:35:17 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java |   35 ++++++++++++++++-------------------
 1 files changed, 16 insertions(+), 19 deletions(-)

diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java
index f593686..221133c 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java
@@ -1,15 +1,13 @@
 package com.iailab.module.model.api;
 
 import com.alibaba.fastjson.JSON;
-import com.iailab.framework.common.util.object.ConvertUtils;
 import com.iailab.module.model.api.mdk.MdkApi;
 import com.iailab.module.model.api.mdk.dto.*;
 import com.iailab.module.model.mcs.pre.entity.DmModuleEntity;
 import com.iailab.module.model.mcs.pre.service.DmModuleService;
 import com.iailab.module.model.mcs.pre.service.MmPredictItemService;
-import com.iailab.module.model.mdk.factory.ItemEntityFactory;
-import com.iailab.module.model.mdk.factory.PredictItemFactory;
-import com.iailab.module.model.mdk.predict.PredictItemHandler;
+import com.iailab.module.model.mcs.sche.service.StScheduleRecordService;
+import com.iailab.module.model.mcs.sche.service.StScheduleSchemeService;
 import com.iailab.module.model.mdk.predict.PredictModuleHandler;
 import com.iailab.module.model.mdk.predict.PredictResultHandler;
 import com.iailab.module.model.mdk.schedule.ScheduleModelHandler;
@@ -25,9 +23,6 @@
 
 import java.util.*;
 import java.util.stream.Collectors;
-
-import static com.iailab.framework.common.pojo.CommonResult.error;
-import static com.iailab.framework.common.pojo.CommonResult.success;
 
 /**
  * @author PanZhibao
@@ -49,16 +44,16 @@
     private PredictModuleHandler predictModuleHandler;
 
     @Autowired
-    private ItemEntityFactory itemEntityFactory;
-
-    @Autowired
-    private PredictItemFactory predictItemFactory;
-
-    @Autowired
     private PredictResultHandler predictResultHandler;
 
     @Autowired
     private ScheduleModelHandler scheduleModelHandler;
+
+    @Autowired
+    private StScheduleRecordService stScheduleRecordService;
+
+    @Autowired
+    private StScheduleSchemeService stScheduleSchemeService;
 
     /**
      * 按模块预测
@@ -101,10 +96,10 @@
                 // 分组,先运行normal预测项,再将结果传递给merge预测项
                 List<ItemVO> normalItems = predictItemList.stream().filter(e -> e.getItemType().equals("NormalItem")).collect(Collectors.toList());
                 if (!CollectionUtils.isEmpty(normalItems)) {
-                    predictModuleHandler.predict(normalItems, reqDTO.getPredictTime(), intervalTime,predictResultMap);
-                    List<ItemVO> mergeItem = predictItemList.stream().filter(e -> e.getItemType().equals("MergeItem")).collect(Collectors.toList());
-                    if (!CollectionUtils.isEmpty(mergeItem)) {
-                        predictModuleHandler.predict(mergeItem, reqDTO.getPredictTime(), intervalTime,predictResultMap);
+                    predictModuleHandler.predict(normalItems, reqDTO.getPredictTime(), intervalTime, predictResultMap);
+                    List<ItemVO> mergeItems = predictItemList.stream().filter(e -> e.getItemType().equals("MergeItem")).collect(Collectors.toList());
+                    if (!CollectionUtils.isEmpty(mergeItems)) {
+                        predictModuleHandler.predict(mergeItems, reqDTO.getPredictTime(), intervalTime, predictResultMap);
                     }
                 }
                 // 更新Module时间
@@ -156,7 +151,7 @@
             List<ItemVO> predictItemList = new ArrayList<>();
             predictItemList.add(itemByItemNo);
             Map<String, PredictResultVO> predictResultMap = new HashMap<>(predictItemList.size());
-            predictModuleHandler.predict(predictItemList, reqDTO.getPredictTime(), 0,predictResultMap);
+            predictModuleHandler.predict(predictItemList, reqDTO.getPredictTime(), 0, predictResultMap);
 
             Map<String, List<MdkPredictDataDTO>> itemPredictData = new HashMap<>();
 
@@ -208,10 +203,12 @@
             log.info("调度计算开始: " + System.currentTimeMillis());
             ScheduleResultVO scheduleResult = scheduleModelHandler.doSchedule(reqDTO.getScheduleCode(), reqDTO.getScheduleTime());
             resp.setResult(scheduleResult.getResult());
+            stScheduleRecordService.create(scheduleResult);
+            stScheduleSchemeService.updateTime(scheduleResult.getSchemeId(), scheduleResult.getScheduleTime());
             log.info("预测计算结束: " + System.currentTimeMillis());
         } catch (Exception ex) {
             log.info("调度计算异常: " + System.currentTimeMillis());
-//            ex.printStackTrace();
+            ex.printStackTrace();
             return resp;
         }
         return resp;

--
Gitblit v1.9.3