CountDownLatch是什么?
jdk1.5开始concurrent包里提供的,并发编程工具类。
CountDownLatch这个类能够使一个线程等待其他线程完成各自的工作后再执行。例如,应用程序的主线程希望在负责启动框架服务的线程已经启动所有的框架服务之后再执行。
CountDownLatch如何工作?
CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值到达0时,它表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务。
API
countDownLatch.countDown()
countDownLatch.await();
CountDownLatch应用场景例子
比如陪媳妇去看病。
医院里边排队的人很多,如果一个人的话,要先看大夫,看完大夫再去排队交钱取药。
现在我们是双核,可以同时做这两个事(多线程)。
假设看大夫花3秒钟,排队交费取药花5秒钟。我们同时搞的话,5秒钟我们就能完成,然后一起回家(回到主线程)。
代码如下:
/**
* 看大夫任务
* Created by jiapeng on 2018/1/7.
*/
public class SeeDoctorTask implements Runnable{
private CountDownLatch countDownLatch;
public SeeDoctorTask(CountDownLatch countDownLatch){
this.countDownLatch = countDownLatch;
}
@Override
public void run() {
try {
Thread.sleep(3000);
System.out.println("看大夫成功,大夫给开了些药单子");
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
if(countDownLatch != null) {
countDownLatch.countDown();
}
}
}
}
/**
* 排队的任务
* Created by jiapeng on 2018/1/7.
*/
public class QueueTask implements Runnable{
private CountDownLatch countDownLatch;
public QueueTask(CountDownLatch countDownLatch){
this.countDownLatch = countDownLatch;
}
@Override
public void run() {
try {
Thread.sleep(5000);
System.out.println("排队成功,可以开始交费");
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
if(countDownLatch != null) {
countDownLatch.countDown();
}
}
}
}
/**
* 配媳妇去看病,轮到媳妇看大夫时
* 我就开始去排队准备交钱了。
* Created by jiapeng on 2018/1/7.
*/
public class MainClient {
public static void main(String[] args) throws Exception{
long now = System.currentTimeMillis();
CountDownLatch countDownLatch = new CountDownLatch(2);
Executor executor = Executors.newFixedThreadPool(2);
executor.execute(new SeeDoctorTask(countDownLatch));
executor.execute(new QueueTask(countDownLatch));
countDownLatch.await();
System.out.println("over,回家 cost:"+(System.currentTimeMillis()-now));
}
}
CountDownLatch底层原理
CountDownLatch通过AQS(AbstractQueuedSynchronizer)里面的共享锁来实现的。
ReentrantLock也是使用AQS