CountDownLatch
概念
Java源代码中写的很清楚CountDownLatch是一个同步工具,用来允许一个或多个线程等待直到其他线程中的一系列操作完成。
CountDownLatch简单演示
package com.example.multi_thread;
import java.util.Random;
import java.util.concurrent.CountDownLatch;
public class CountDownLatchTest01 {
class SearchTask implements Runnable {
Integer id;
CountDownLatch latch;
public SearchTask(Integer id, CountDownLatch latch) {
this.id = id;
this.latch = latch;
}
@Override
public void run() {
System.out.println(Thread.currentThread().getName() + "开始寻找" + id + "号球员");
int seconds = new Random().nextInt(10);
try {
Thread.sleep(seconds * 1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(Thread.currentThread().getName() + "花了" + seconds + "s, 找到了" + id + "号球员");
latch.countDown();
}
}
public static void main(String[] args) {
// 用十一个元素的数组代表一个足球队
Integer[] array = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11};
// 传入的参数即为任务数量
CountDownLatch latch = new CountDownLatch(array.length);
for (Integer id : array) {
// 循环创建线程执行任务
new Thread(new CountDownLatchTest01().new SearchTask(id, latch)).start();
}
try {
// 执行await方法,使主线程等待
latch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
// 这句话会在十一个线程全部结束之后再执行
System.out.println("已集齐球员可以开始比赛");
}
}
执行结果
源码分析
内部属性
只有一个内部属性Sync,是CountDownLatch类的一个静态内部类
内部方法
-
await()
-
调用await
可以调用含参数await()方法设置timeout,不让等待线程阻塞时间过长
-
调用acquireSharedInterruptibly
-
调用Latch的tryAcquired
父类AQS中的tryAcquireShared方法对return值的要求
-
调用doAcquireSharedInterruptibly
-
调用setHeadAndPropagate
-
-
countDown()
用于告知当前任务子任务已完成
调用AQS的releaseShared
调用Latch的tryReleaseShared
调用doReleaseShared唤醒主线程 -
getCount()
由于继承自AQS,故通过state属性来看子任务数量,每当一个子任务完成,state - 1,当state为0时,则可判断所有子任务完成,当前线程不再阻塞