From b19bbf91f21a523fecd93f510126521ab7c3c6a7 Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Tue, 22 Apr 2025 11:24:28 +0800
Subject: [PATCH] =?UTF-8?q?TASK:=E6=A8=A1=E5=9D=97=E5=8C=96=EF=BC=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../da/common/RabbitMQ/GenerateConsumer.java  | 32 +++++++++----------
 src/main/resources/application-dev.properties |  2 +-
 2 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java b/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
index 5c267721..37c1978d 100644
--- a/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
+++ b/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
@@ -380,7 +380,7 @@ public class GenerateConsumer {
 
         try {
             log.info("task_id : {} start ", generateResult.get("task_id"));
-            if (generateResult.get("status").equals("SUCCESS")) {
+            if (!StringUtils.isEmpty(generateResult.getString("progress"))) {
                 String progress = generateResult.getString("progress");
                 JSONArray result = generateResult.getJSONArray("result");
                 String url = null;
@@ -499,20 +499,20 @@ public class GenerateConsumer {
     public void getPoseTransformationResult(Message msg, Channel channel) {
         processPoseTransformResult(msg, channel);
     }
-//    @RabbitListener(queues = "#{rabbitMQProperties.queues.designBatch}")
-//    @RabbitHandler
-//    public void getDesignBatchResult(Message msg, Channel channel) {
-//        processDesignBatchResult(msg, channel);
-//    }
-//    @RabbitListener(queues = "#{rabbitMQProperties.queues.toProductImageBatch}")
-//    @RabbitHandler
-//    public void getToProductImageBatchResult(Message msg, Channel channel) {
-//        processToProductImageBatchResult(msg, channel);
-//    }
+    @RabbitListener(queues = "#{rabbitMQProperties.queues.designBatch}")
+    @RabbitHandler
+    public void getDesignBatchResult(Message msg, Channel channel) {
+        processDesignBatchResult(msg, channel);
+    }
+    @RabbitListener(queues = "#{rabbitMQProperties.queues.toProductImageBatch}")
+    @RabbitHandler
+    public void getToProductImageBatchResult(Message msg, Channel channel) {
+        processToProductImageBatchResult(msg, channel);
+    }
 
-//    @RabbitListener(queues = "#{rabbitMQProperties.queues.relightBatch}")
-//    @RabbitHandler
-//    public void getRelightBatchResult(Message msg, Channel channel) {
-//        processRelightBatchResult(msg, channel);
-//    }
+    @RabbitListener(queues = "#{rabbitMQProperties.queues.relightBatch}")
+    @RabbitHandler
+    public void getRelightBatchResult(Message msg, Channel channel) {
+        processRelightBatchResult(msg, channel);
+    }
 }
diff --git a/src/main/resources/application-dev.properties b/src/main/resources/application-dev.properties
index 2b3eef21..44aa807c 100644
--- a/src/main/resources/application-dev.properties
+++ b/src/main/resources/application-dev.properties
@@ -107,7 +107,7 @@ rabbitmq.queues.toProductImageResult=ToProductImage-dev
 rabbitmq.queues.relightResult=Relight-dev
 rabbitmq.queues.poseTransform=PoseTransform-dev
 rabbitmq.exchange.generate=generate-exchange
-rabbitmq.queues.designBatch=BatchRelight-dev
+rabbitmq.queues.designBatch=DesignBatch-dev
 rabbitmq.queues.relightBatch=BatchRelight-dev
 rabbitmq.queues.toProductImageBatch=BatchToProductImage-dev