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

generate

Browse Source
This commit is contained in:
xupei
2024-04-18 18:04:02 +08:00
parent 3be1bf3c14
commit e832d5e4d4
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -109,12 +109,18 @@ public class GenerateConsumer {
long start = System.currentTimeMillis(); long start = System.currentTimeMillis();
Map<String, String> generateResult = JSONObject.parseObject(msg.getBody(), Map.class); Map<String, String> generateResult = JSONObject.parseObject(msg.getBody(), Map.class);
// 2.1 手动确认该消息
try {
channel.basicAck(msg.getMessageProperties().getDeliveryTag(), false);
} catch (IOException e) {
throw new RuntimeException(e);
}
log.info("tasks_id : {} start ",generateResult.get("tasks_id")); log.info("tasks_id : {} start ",generateResult.get("tasks_id"));
// log.info("tasks_id : {}, message : {}",generateResult.get("tasks_id"), generateResult.get("message") ); // log.info("tasks_id : {}, message : {}",generateResult.get("tasks_id"), generateResult.get("message") );
if (generateResult.get("status").equals("SUCCESS")){ if (generateResult.get("status").equals("SUCCESS")){
String url = generateResult.get("data"); String url = generateResult.get("data");
String taskId = generateResult.get("tasks_id"); String taskId = generateResult.get("tasks_id");
generateService.processGenerateResult(taskId, url); // generateService.processGenerateResult(taskId, url);
}else { }else {
// 修改redis中的数据状态为exception // 修改redis中的数据状态为exception
String key = generateResultKey + ":" + generateResult.get("tasks_id"); String key = generateResultKey + ":" + generateResult.get("tasks_id");