一文搞懂CountDownLatch！

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类的一个静态内部类

内部方法

1. await()

   • 调用await
     
     可以调用含参数await()方法设置timeout，不让等待线程阻塞时间过长
     

   • 调用acquireSharedInterruptibly
     

   • 调用Latch的tryAcquired
     
     父类AQS中的tryAcquireShared方法对return值的要求
     

   • 调用doAcquireSharedInterruptibly
     

   • 调用setHeadAndPropagate
     

2. countDown()
   用于告知当前任务子任务已完成
   
   调用AQS的releaseShared
   
   调用Latch的tryReleaseShared
   
   调用doReleaseShared唤醒主线程

3. getCount()
   由于继承自AQS，故通过state属性来看子任务数量，每当一个子任务完成，state - 1，当state为0时，则可判断所有子任务完成，当前线程不再阻塞

调用时序