From 17a6e42c685b7aee80a953f182ab514db2643fe0 Mon Sep 17 00:00:00 2001
From: 潘志宝 <979469083@qq.com>
Date: 星期三, 11 十二月 2024 18:07:11 +0800
Subject: [PATCH] Merge branch 'master' of http://dlindusit.com:53929/r/shasteel

---
 shasteel-biz/src/main/java/com/iailab/module/shasteel/job/task/RunPredictModuleTask.java |   20 ++++++++++++--------
 1 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/shasteel-biz/src/main/java/com/iailab/module/shasteel/job/task/RunPredictModuleTask.java b/shasteel-biz/src/main/java/com/iailab/module/shasteel/job/task/RunPredictModuleTask.java
index f6e5f46..4675dd5 100644
--- a/shasteel-biz/src/main/java/com/iailab/module/shasteel/job/task/RunPredictModuleTask.java
+++ b/shasteel-biz/src/main/java/com/iailab/module/shasteel/job/task/RunPredictModuleTask.java
@@ -1,9 +1,12 @@
 package com.iailab.module.shasteel.job.task;
 
 import com.iailab.module.model.api.mdk.MdkApi;
+import com.iailab.module.model.api.mdk.dto.MdkPredictModuleRespDTO;
 import com.iailab.module.model.api.mdk.dto.MdkPredictReqDTO;
+import com.iailab.module.shasteel.mq.common.RoutingConstant;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.amqp.rabbit.core.RabbitTemplate;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
@@ -11,30 +14,31 @@
 
 /**
  * 执行模块预测定时任务
- *
  */
 @Component("runPredictModuleTask")
 public class RunPredictModuleTask implements ITask {
     private Logger logger = LoggerFactory.getLogger(getClass());
 
     @Autowired
-    MdkApi mdkApi;
+    private MdkApi mdkApi;
 
+    @Autowired
+    private RabbitTemplate rabbitTemplate;
 
     @Override
     public void run(String params) {
-        logger.info("runPredictModuleTask定时任务正在执行,参数为:{}",params );
+        logger.info("runPredictModuleTask定时任务正在执行,参数为:{}", params);
         try {
             Calendar calendar = Calendar.getInstance();
-            calendar.set(Calendar.SECOND,0);
-            calendar.set(Calendar.MILLISECOND,0);
+            calendar.set(Calendar.SECOND, 0);
+            calendar.set(Calendar.MILLISECOND, 0);
 
             MdkPredictReqDTO dto = new MdkPredictReqDTO();
             dto.setPredictTime(calendar.getTime());
             dto.setModuleType(params);
-            mdkApi.predictModule(dto);
-            logger.info(params + "模块预测完成");
-
+            MdkPredictModuleRespDTO mdkPredictModuleRespDTO = mdkApi.predictModule(dto);
+            logger.info(params + "模块预测完成," + mdkPredictModuleRespDTO);
+            rabbitTemplate.convertAndSend(RoutingConstant.EXCHANGE, RoutingConstant.Iailab_Model_PredictFinish, mdkPredictModuleRespDTO);
 
         } catch (Exception ex) {
             logger.error("runPredictModuleTask运行异常");

--
Gitblit v1.9.3