背景:最近做测试需要准备大量的数据,用单线程去连续调几个接口耗时N秒,而数据量大这个时间消耗就成倍数增长,单线程效率太低,决定用多线程跑数据。
使用多线程,首先实现一个线程,再开辟一个线程池,循环调用。但是线程什么时候结束是个问题,这里使用到了CountDownLatch。
Latch:英[lætʃ], 门闩。
CountDownLatch 是一个同步工具类,它允许一个或多个线程一直等待,直到其他线程的操作执行完成后再执行。
CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值达到0时,表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务了。
CountDownLatch的伪代码如下所示:
1. //Main thread start
2. //Create CountDownLatch for N threads
3. //Create and start N threads
4. //Main thread wait on latch
5. //N threads completes there tasks are returns
6. //Main thread resume execution
CountDownLatch如何使用?
CountDownLatch.java类中定义的构造函数:
1. //Constructs a CountDownLatch initialized with the given count.
2. public void CountDownLatch(int count) {...}
构造器中的计数值实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,且不能被修改。
主线程必须在启动其他线程后立即调用CountDownLatch.await()方法。这样主线程的操作就会在这个方法上阻塞,直到其他线程完成各自的任务。
其他N个线程必须引用闭锁对象,因为他们需要通知CountDownLatch对象,他们已经完成了各自的任务。这种通知通过CountDownLatch.countDown()方法来完成;每一次调用这个方法,在构造函数中初始化的count值就减1.所以当所有N个线程都调用了这个方法,count=0,主线程就能通过await()方法,恢复执行自己的任务。
代码Demo:
public void bizDealCode(){
Long startTime = System.currentTimeMillis();
intsize =10000;
ExecutorService threadPool = Executors.newFixedThreadPool(15);
CountDownLatch countDownLatch =newCountDownLatch(size);
for(intj =0; j < size; ++j) {
threadPool.submit(newMakeOrderThread(countDownLatch));
}
System.out.println("finished submit");
try{
countDownLatch.await();
}catch(Exception e) {
}
System.out.println("finished process");
threadPool.shutdown();
System.out.println("costTime is "+ (System.currentTimeMillis() - startTime) +"ms");
}
static classMakeOrderThreadextendsThread {
privateCountDownLatchcountDownLatch;
publicMakeOrderThread(CountDownLatch countDownLatch) {
this.countDownLatch= countDownLatch;
}
@Override
public voidrun() {
try{
bizCode();
}catch(Exception e) {
}finally{
countDownLatch.countDown();
}
}
}
}