fix(接口测试): 修复任务中心状态查询问题

backend/services/system-setting/src/main/java/io/metersphere/system/dto/taskhub/ResourcePoolStatusDTO.java

@@ -1,6 +1,7 @@
 package io.metersphere.system.dto.taskhub;
 
 import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 
@@ -12,6 +13,7 @@ import java.io.Serializable;
  */
 @Data
 @EqualsAndHashCode(callSuper = false)
+@AllArgsConstructor
 public class ResourcePoolStatusDTO implements Serializable {
 
     @Serial
@@ -22,4 +24,5 @@ public class ResourcePoolStatusDTO implements Serializable {
 
     @Schema(description = "状态 (true 正常 false 异常)")
     private boolean status;
+
 }

backend/services/system-setting/src/main/java/io/metersphere/system/service/BaseTaskHubService.java

@@ -533,29 +533,51 @@ public class BaseTaskHubService {
      * @return
      */
     public List<ResourcePoolStatusDTO> getResourcePoolStatus(List<String> ids) {
-        List<ResourcePoolStatusDTO> statusDTOS = new ArrayList<>();
-        List<ExecTaskItem> itemList = extExecTaskItemMapper.selectPoolNodeByIds(ids);
-        Map<String, List<ExecTaskItem>> poolNodeMap = itemList.stream().collect(Collectors.groupingBy(ExecTaskItem::getResourcePoolNode));
-        poolNodeMap.forEach((k, v) -> {
-            String[] split = k.split(":");
-            TestResourceNodeDTO node = new TestResourceNodeDTO();
-            boolean status = false;
-            try {
+        List<ExecTaskItem> items = extExecTaskItemMapper.selectPoolNodeByIds(ids);
+
+        return items.stream()
+                .collect(Collectors.groupingBy(ExecTaskItem::getResourcePoolNode))
+                .entrySet()
+                .stream()
+                .map(entry -> {
+                    String key = entry.getKey();
+                    List<ExecTaskItem> itemGroup = entry.getValue();
+
+                    // Asynchronously determine the status
+                    CompletableFuture<Boolean> statusFuture = CompletableFuture.supplyAsync(() -> determineStatus(key, itemGroup));
+
+                    return statusFuture.thenApply(status -> itemGroup.stream()
+                            .map(item -> new ResourcePoolStatusDTO(item.getId(), status))
+                            .collect(Collectors.toList()));
+                })
+                .toList()
+                .stream()
+                .flatMap(future -> future.join().stream()) // Wait for all futures to complete
+                .collect(Collectors.toList());
+    }
+
+    /**
+     * Determine the status of a resource pool node
+     */
+    private boolean determineStatus(String key, List<ExecTaskItem> items) {
+        try {
+            String[] split = key.split(":");
+            if (split.length == 2) {
+                var node = new TestResourceNodeDTO();
                 node.setIp(split[0]);
                 node.setPort(split[1]);
-                status = nodeResourcePoolService.validateNode(node);
-            } catch (Exception e) {
-                LogUtils.error(e);
+                return nodeResourcePoolService.validateNode(node);
+            } else if (!items.isEmpty()) {
+                var testResourceDTO = new TestResourceDTO();
+                var returnDTO = testResourcePoolService.getTestResourcePoolDetail(items.getFirst().getResourcePoolId());
+                BeanUtils.copyBean(testResourceDTO, returnDTO.getTestResourceReturnDTO());
+                testResourceDTO.setDeployName(key);
+                return EngineFactory.validateNamespaceExists(testResourceDTO);
             }
-            boolean finalStatus = status;
-            v.forEach(item -> {
-                ResourcePoolStatusDTO poolStatusDTO = new ResourcePoolStatusDTO();
-                poolStatusDTO.setId(item.getId());
-                poolStatusDTO.setStatus(finalStatus);
-                statusDTOS.add(poolStatusDTO);
-            });
-        });
-        return statusDTOS;
+        } catch (Exception e) {
+            // Log the exception if needed
+        }
+        return false;
     }