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 546de399..9324ef97 100644 --- a/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java +++ b/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java @@ -382,7 +382,11 @@ public class GenerateConsumer { log.info("task_id : {} start ", generateResult.get("task_id")); if (generateResult.get("status").equals("SUCCESS")) { String progress = generateResult.getString("progress"); - String url = generateResult.getJSONArray("result").getString(0); + JSONArray result = generateResult.getJSONArray("result"); + String url = null; + if (!StringUtils.isEmpty(result)) { + url = result.getString(0); + } String taskId = generateResult.getString("task_id"); userLikeGroupService.relightBatch(taskId, url, progress); } else { @@ -506,9 +510,9 @@ public class GenerateConsumer { 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); + } }