TASK:generate 修复逻辑Bug

src/main/java/com/ai/da/common/utils/AsyncCallerUtil.java

@@ -30,21 +30,20 @@ public class AsyncCallerUtil {
     public List<String> generate(GenerateToPythonDTO generateToPython) {
         ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(5);
         String taskId = generateToPython.getTasks_id();
-        waitingStatus.put(taskId, true);
         ScheduledFuture<?> timeoutTask = null;
+        if (!waitingStatus.containsKey(taskId)) waitingStatus.put(taskId, true);
 
         try {
             CompletableFuture<List<String>> generateResult = callGenerateAsync(generateToPython);
-            // 10秒后第一次确认，之后每隔10秒确认一次用户选择结果
+            // 5秒后第一次确认，之后每隔10秒确认一次用户选择结果
             timeoutTask = scheduledExecutorService.scheduleAtFixedRate(() -> {
                 // 调用另一个接口获取用户的选择
                 if (!waitingStatus.get(taskId)) {
-                    // 如果用户选择取消，则取消对generate的调用，cancel判断是否成功取消
+                    // 如果用户选择取消，则取消对generate的调用
                     generateResult.cancel(true);
                     waitingStatus.remove(taskId);
-                }
-                log.info("===============持续等待===============");
-            }, 10, 10, TimeUnit.SECONDS);
+                } else log.info("===============持续等待===============");
+            }, 5, 10, TimeUnit.SECONDS);
 
             log.info("阻塞等待结果...");
             // 阻塞，等待结果
@@ -53,6 +52,10 @@ public class AsyncCallerUtil {
             timeoutTask.cancel(true);
             waitingStatus.remove(taskId);
             return result;
+        } catch (CancellationException e) {
+            // generateResult.cancel(true);通过抛出异常取消该任务
+            log.info("==========成功取消generate任务==========");
+            return null;
         } catch (InterruptedException | ExecutionException | BusinessException e) {
             // 处理异常
             log.error("发生错误 ： " + e);