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