CountDownLatch用法:
CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值到达0时,它表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务。
构造器中的计数值(count)实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,而且CountDownLatch没有提供任何机制去重新设置这个计数值。
与CountDownLatch的第一次交互是主线程等待其他线程。主线程必须在启动其他线程后立即调用CountDownLatch.await()方法。这样主线程的操作就会在这个方法上阻塞,直到其他线程完成各自的任务。
其他N个线程必须引用闭锁对象,因为他们需要通知CountDownLatch对象,他们已经完成了各自的任务。这种通知机制是通过 CountDownLatch.countDown()方法来完成的;每调用一次这个方法,在构造函数中初始化的count值就减1。所以当N个线程都调 用了这个方法,count的值等于0,然后主线程就能通过await()方法,恢复执行自己的任务。
比如:要完成一项任务,需要多个初始化步骤,等所有初始化步骤完成了以后才能执行业务方法。
实例:
初始化线程类
public class InitThread implements Runnable{
private CountDownLatch countDownLatch;
InitThread(CountDownLatch countDownLatch){
this.countDownLatch=countDownLatch;
}
@Override
public void run() {
System.out.println("线程"+Thread.currentThread().getName()+"开始初始化工作");
Random random = new Random();
//随机休眠2秒
if(random.nextBoolean()){
try {
System.out.println("线程"+Thread.currentThread().getName()+"进入休眠");
Thread.sleep(5000l);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
countDownLatch.countDown();
System.out.println("线程"+Thread.currentThread().getName()+"完成初始化工作");
}
}
测试方法
public static void main(String [] args) throws InterruptedException {
CountDownLatch countDownLatch = new CountDownLatch(3);
for(int i=0;i<3;i++){
new Thread(new InitThread(countDownLatch),"Thread"+i).start();
}
try {
countDownLatch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
Thread.sleep(500l);
System.out.println("所有线程均已就绪可以开始执行doBussiness方法");
}
执行结果:
线程Thread1开始初始化工作
线程Thread1进入休眠
线程Thread0开始初始化工作
线程Thread0进入休眠
线程Thread2开始初始化工作
线程Thread2完成初始化工作
线程Thread1完成初始化工作
线程Thread0完成初始化工作
所有线程均已就绪可以开始执行doBussiness方法
ps:await()方法可以在任何位置,也可以在子线程中增加countDown.await()方法,当countdown到0的时候,所有在await()处等待的线程都可以得到释放