概念
Latch(阀门)设计模式也叫做 Count Down 设计模式。当若干个线程并发执行完某个特 定的任务,然后等到所有的子任务都执行结束之后再统一汇总。
CountDownLatch 能够使一个线程在等待另外一些线程完成各自工作之后,再继续执 行。使用一个计数器进行实现。计数器初始值为线程的数量。当每一个线程完成自己任务后, 计数器的值就会减一。当计数器的值为 0 时,表示所有的线程都已经完成了任务,然后在
CountDownLatch 上等待的线程就可以恢复执行任务。示例代码如下:
/**
* 自定义CustomCountDownLatch
*/
public class CustomCountDownLatch {
//阀门值
private final int latchNum;
//计数器
private int count;
public CustomCountDownLatch(int latch_num) {
this.latchNum = latch_num;
this.count = latch_num;
}
//计数方法 进行count--
public synchronized void countDown(){
this.count--;
notifyAll();
}
//阻止线程继续执行 依据计数器判断线程能否执行
public synchronized void await() {
if(this.count!=0){
try {
this.wait();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
}