有些场景下,你可能要检查代码有没有并发问题,验证幂等或者做一下简单的压力测试, 这时候需要写一个并发请求的程序,那么怎么做呢?
先上代码
final long time_start = System.currentTimeMillis();
ExecutorService service = Executors.newFixedThreadPool(10);
int client_num = 10;
final CountDownLatch countDownLatch = new CountDownLatch(1);
for (int i = 0; i < client_num; i++) {
Runnable job = new Runnable() {
@Override
public void run() {
Worker worker = new Worker();
try {
countDownLatch.await();
worker.doWork();
}catch(Exception e){
e.printStackTrace();
}
}
};
service.submit(job);
}
countDownLatch.countDown();
service.shutdown();
while (!service.isTerminated()) {
}
System.out.println("time consuming is " + (System.currentTimeMillis() - time_start));
我们来拆解下这个Case,看看都做了些什么:
1. 容器准备
首先,你需要一个线程池来调用任务处理,虽然你可以每次new一个线程来做,但是无论是在生产环境还是测试,都不太建议这样做。本示例用的是
ExecutorService service = Executors.newFixedThreadPool(10)
其效果是构造一个固定线程数的线程池,这个线程池的coreSize和maxSize都是10, 也就是说,最多可以同时处理10个任务,多了的就要排队。
线程池装载完任务后,执行submit , 但是此时任务并未执行,具体原因看下面。
2. 并发执行
这里是比较tricky的一个点,因为要模拟完全并发的情况,就要保证所有任务同一时间触发,这里我们用了一个java.util.concurrent的工具类 CountDownLatch, 从名字大概可以猜测到,这个是一个像栅栏一样的工具,先把羊圈在里面,等count数到0的时候,打开栅栏,放出群羊。。
使用步骤: 首先,初始化countDownLatch计数为1。 然后, 在业务代码里在执行业务逻辑之前我们嵌入了一行 countDownLatch.await()
。最后, 当所有任务都装载完毕提交后,执行countDownLatch.countDown()
,如同发令枪一般,所有的任务同时触发,并发量达到client_num的数量10。
3. 任务结束
任务结束的时候调用的是线程池的shutdown方法, 看一下这个方法的注释
Attempts to stop all actively executing tasks, halts the processing of waiting tasks, and returns a list of the tasks that were awaiting execution.
意思是这个方法只是尝试去停止所有任务,但是并不保证都能结束,并且没有结束的任务是以list返回的。 所以后面又跟了一个
while (!service.isTerminated()) {
}
循环一直等待所有的任务结束完毕,最后再统计所有任务的执行时间。
结语
这是一段比较简单的并发测试代码,单元测试足够用,但是如果用做规模化的压力测试,还需要考虑压力机与测试机的连接以及数据热点等一系列问题,这又是个很大的话题了。