缓存线程池+ExecutorCompletionService 实现异步压力测试执行

妖狐艹你老母 2022-12-29 12:48 160阅读 0赞

整体逻辑比较简单，Springboot提供接口用以前端开启压测，controller层调用异步任务开始执行压测，Springboot使用异步接口可参考[《SpringBoot：实现异步(Async)接口》][SpringBoot_Async]。  
然后在异步任务中通过`缓存线程池(CachedThreadPool)`启用多线程，通过`ExecutorCompletionService`进行执行操作。

#### controller ####

Springboot提供`/execute`接口用以前端调用压测，在这里以接口压力测试为例

@PostMapping("/execute")
    public String execute(@RequestBody PressureExecuteDTO pressureExecuteDTO) { 
    	// 取得对应测试接口的数据
        HttpCaseDO httpCaseDO = httpCaseService.getById(pressureExecuteDTO.getCaseId());
    	// 传入压测数据和接口数据
        pressureStart.testStart(pressureExecuteDTO, httpCaseDTO);
    
        return "开始执行";
    
    }

#### PressureExecuteDTO ####

/** * 并发线程数 */
    private Integer threads;
    
    /** * 重复执行次数 */
    private Integer times;
    
    /** * httpCaseId */
    private Integer caseId;

#### PressureStart ####

@Component
    public class PressureStart { 
    
    	// 创建缓存线程池
        ExecutorService es = Executors.newCachedThreadPool();
        CompletionService<List<AutoTestResponseVO>> completionService = new ExecutorCompletionService<>(es);
    
    	@Async
        public void testStart(PressureExecuteDTO pressureExecuteDTO, HttpCaseDTO httpCaseDTO) { 
    
            int threads = pressureExecuteDTO.getThreads();
            int times = pressureExecuteDTO.getTimes();
            
            // 保存回调结果，如果不考虑重复执行次数，这里保存为Future<?>即可
            List<Future<List<ResponseVO>>> respList = new ArrayList<>();
            // 启动多线程
            for (int i=0;i<threads;i++){ 
            	// 在submit内拿到回调结果
                respList.add(completionService.submit(()->{ 
                    List<ResponseVO> resList = new ArrayList<>(times);
                    // 线层内重复执行测试
                    for (int j=0;j<times;j++) { 
                    	// 测试执行
                        ResponseVO executeResponse = httpCaseService.caseDoExecute(httpCaseDTO);
                        resList.add(executeResponse);
                    }
                    return resList;
                }));
            }
    		es.shutdown();
    		// 拿到所有结果后的后续处理
            for (Future<List<ResponseVO>> o : respList) { 
                try { 
                    List<ResponseVO> l = o.get();
                    for (AutoTestResponseVO res:l){ 
                        System.out.println(JSON.toJSONString(res.getResponseTime()));
                    }
                } catch (InterruptedException | ExecutionException e) { 
                    e.printStackTrace();
                }
            }
        }
    }

[SpringBoot_Async]: https://blog.csdn.net/lt326030434/article/details/108658212