test-version22
/
MeterSphere
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

perf(系统管理): 只有开启的资源池才获取剩余并发数

This commit is contained in:
guoyuqi 2024-10-17 12:56:03 +08:00 committed by Craftsman
parent c03a7f181b
commit d4655268f1
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
backend/services/system-setting/src/main/java/io/metersphere/system/service/TestResourcePoolService.java
View File

@ -170,10 +170,12 @@ public class TestResourcePoolService {
if (StringUtils.equalsIgnoreCase(pool.getType(), ResourcePoolTypeEnum.NODE.getName())) {
nodeMap.put(pool.getId(), testResourceDTO.getNodesList());
poolDTOMap.put(pool.getId(), testResourcePoolDTO);
if (pool.getEnable()) {
Set<String> nodeSet = testResourceDTO.getNodesList().stream()
.map(node -> node.getIp() + ":" + node.getPort())
.collect(Collectors.toSet());
nodeSets.addAll(nodeSet);
}
} else {
//处理k8s资源池
testResourcePoolDTO.setMaxConcurrentNumber(testResourceDTO.getConcurrentNumber());
@ -181,7 +183,7 @@ public class TestResourcePoolService {
}
});
//处理node资源池
Map<String,Integer>lastConcurrentNumberMap = new HashMap<>();
Map<String, Integer> lastConcurrentNumberMap = new HashMap<>();
List<Future<Map<String, Integer>>> futures = new ArrayList<>();
for (String nodeSet : nodeSets) {
@ -189,7 +191,7 @@ public class TestResourcePoolService {
String[] split = nodeSet.split(":");
ResourcePoolNodeMetric nodeMetric = getNodeMetric(split[0], split[1]);
Map<String, Integer> resultMap = new HashMap<>();
if (nodeMetric!=null) {
if (nodeMetric != null) {
resultMap.put(nodeSet, nodeMetric.getConcurrentNumber() - nodeMetric.getOccupiedConcurrentNumber());
}
return resultMap;
@ -201,20 +203,22 @@ public class TestResourcePoolService {
lastConcurrentNumberMap.putAll(future.get());
} catch (InterruptedException | ExecutionException e) {
// 处理异常
LogUtils.error("获取剩余并发数失败："+ e);
LogUtils.error("获取剩余并发数失败：" + e);
}
}
nodeMap.forEach((poolId,nodeList)->{
nodeMap.forEach((poolId, nodeList) -> {
int lastConcurrentNumber = 0;
int maxConcurrentNumber = 0;
for (TestResourceNodeDTO testResourceNodeDTO : nodeList) {
if (lastConcurrentNumberMap.get(testResourceNodeDTO.getIp() + ":" + testResourceNodeDTO.getPort())!=null) {
lastConcurrentNumber = lastConcurrentNumber+lastConcurrentNumberMap.get(testResourceNodeDTO.getIp() + ":" + testResourceNodeDTO.getPort());
if (lastConcurrentNumberMap.get(testResourceNodeDTO.getIp() + ":" + testResourceNodeDTO.getPort()) != null) {
lastConcurrentNumber = lastConcurrentNumber + lastConcurrentNumberMap.get(testResourceNodeDTO.getIp() + ":" + testResourceNodeDTO.getPort());
}
maxConcurrentNumber = maxConcurrentNumber + testResourceNodeDTO.getConcurrentNumber();
}
TestResourcePoolDTO testResourcePoolDTO = poolDTOMap.get(poolId);
if (testResourcePoolDTO.getEnable()) {
testResourcePoolDTO.setLastConcurrentNumber(lastConcurrentNumber);
}
testResourcePoolDTO.setMaxConcurrentNumber(maxConcurrentNumber);
testResourcePoolDTOS.add(testResourcePoolDTO);
});
@ -353,7 +357,7 @@ public class TestResourcePoolService {
}
@PreDestroy
private void shutdownExecutor() {
public void shutdownExecutor() {
executor.shutdown();
}
}