aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:模块化；

Browse Source
This commit is contained in:
shahaibo
2025-04-22 11:24:28 +08:00
parent d9e4fb3022
commit b19bbf91f2
2 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
View File

@@ -380,7 +380,7 @@ public class GenerateConsumer {
try { try {
log.info("task_id : {} start ", generateResult.get("task_id")); 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"); String progress = generateResult.getString("progress");
JSONArray result = generateResult.getJSONArray("result"); JSONArray result = generateResult.getJSONArray("result");
String url = null; String url = null;
@@ -499,20 +499,20 @@ public class GenerateConsumer {
public void getPoseTransformationResult(Message msg, Channel channel) { public void getPoseTransformationResult(Message msg, Channel channel) {
processPoseTransformResult(msg, channel); processPoseTransformResult(msg, channel);
} }
// @RabbitListener(queues = "#{rabbitMQProperties.queues.designBatch}") @RabbitListener(queues = "#{rabbitMQProperties.queues.designBatch}")
// @RabbitHandler @RabbitHandler
// public void getDesignBatchResult(Message msg, Channel channel) { public void getDesignBatchResult(Message msg, Channel channel) {
// processDesignBatchResult(msg, channel); processDesignBatchResult(msg, channel);
// } }
// @RabbitListener(queues = "#{rabbitMQProperties.queues.toProductImageBatch}") @RabbitListener(queues = "#{rabbitMQProperties.queues.toProductImageBatch}")
// @RabbitHandler @RabbitHandler
// public void getToProductImageBatchResult(Message msg, Channel channel) { public void getToProductImageBatchResult(Message msg, Channel channel) {
// processToProductImageBatchResult(msg, channel); processToProductImageBatchResult(msg, channel);
// } }
// @RabbitListener(queues = "#{rabbitMQProperties.queues.relightBatch}") @RabbitListener(queues = "#{rabbitMQProperties.queues.relightBatch}")
// @RabbitHandler @RabbitHandler
// public void getRelightBatchResult(Message msg, Channel channel) { public void getRelightBatchResult(Message msg, Channel channel) {
// processRelightBatchResult(msg, channel); processRelightBatchResult(msg, channel);
// } }
} }

2
src/main/resources/application-dev.properties
View File

@@ -107,7 +107,7 @@ rabbitmq.queues.toProductImageResult=ToProductImage-dev
rabbitmq.queues.relightResult=Relight-dev rabbitmq.queues.relightResult=Relight-dev
rabbitmq.queues.poseTransform=PoseTransform-dev rabbitmq.queues.poseTransform=PoseTransform-dev
rabbitmq.exchange.generate=generate-exchange rabbitmq.exchange.generate=generate-exchange
rabbitmq.queues.designBatch=BatchRelight-dev rabbitmq.queues.designBatch=DesignBatch-dev
rabbitmq.queues.relightBatch=BatchRelight-dev rabbitmq.queues.relightBatch=BatchRelight-dev
rabbitmq.queues.toProductImageBatch=BatchToProductImage-dev rabbitmq.queues.toProductImageBatch=BatchToProductImage-dev