1、简介
CountDownLatch是一个同步工具类,用来携调多个线程之间的同步,它是是使用一个计数器进行实现的,计数器初始值为线程数量。当每一个线程完成自己任务后,计数器的值就会减1。当计数器的值为0时,表示所有的线程都已经完成一些任务,然后再CountDownLatch上等待的线程就可以恢复执行下来的任务。
CountDownLatch主要有两个方法:countDown()和await();
countDown()方法用于使计数器减一,其一般是执行任务的线程调用,
await()方法是调用该方法的线程处于等待状态,其一般是主线程调用。
countDown()方法并没有规定一个线程只能调用一次,当同一个线程调用多次countDown()方法时,每次都会使计数器减一,
await()方法也并没有规定只能有一个线程执行该方法,如果多个线程同时执行await()方法,那么这几个线程都将处于等待状态,并且以共享模式享有同一个锁。
2、方法API
方法 | 说明 |
---|---|
await() | 使当前线程进入同步队列进行等待,直到latch 的值被减到0 或者当前线程被中断,当前线程就会被唤醒。 |
await(long timeout, TimeUnit unit) | 等待timeout时间后,count的值还不是0,不再等待,那么将继续执行 |
countDown() | 使latch 的值减1 ,如果减到了0 ,则会唤醒所有等待在这个latch 上的线程。 |
getCount() | 获得latch 的数值。 |
3、使用
public class Test {
public static void main(String[] args) throws InterruptedException {
//当前执行的线程数量
Integer poolNum = 10;
long start = System.currentTimeMillis();
//初始化线程运行的数量
CountDownLatch countDownLatch = new CountDownLatch(poolNum);
Map<Integer,Integer> map = new HashMap<>();
for (int i = 1; i <= poolNum; i++) {
//耗时
Integer time = i*1000;
ThreadPoolUtils.execute(()->{
try {
//假设逻辑执行的耗时
Thread.sleep(time);
map.put(time,time);
} catch (InterruptedException e) {
e.printStackTrace();
}finally {
countDownLatch.countDown();
}
});
}
// 4分45秒 countDownLatch.getCount()还不是,不再等待,那么将继续执行
countDownLatch.await(4 * 60 + 45);
System.out.println("总耗时:"+(System.currentTimeMillis()-start));
for (Integer key: map.keySet()){
//打印出来的key值是没有顺序的,因为竞争的CPU资源不一样
System.out.println("map的key为:"+key+" map的值为:"+ map.get(key));
}
}
}