在Java的多线程中,有的时候我们需要等某些多线程的方法全部都执行完毕后,才继续往下执行。这个该怎么控制呢?用CountDownLatch类。这个类是java.util.concurrent包里面,也即是Java的并发工具包,这个包里面有很多多线程的工具类,在多线程编程中很有用。

下面就是一个CountDownLatch类的一个用法。比如我们有五个人(五个线程),他们都赶往同一目的地,但是我们不知道他们什么时候才能赶来,或许1秒,或许9秒,或许更长。然后我们接下来的事情需要五个人都到齐了才能继续往下做。这就需要进行多线程控制。

在Java语言里,也就是有N个线程,我们接下来要多线程调度这N个线程,然后这N个线程都全部执行完毕后,代码才接着往下执行。应用场景已经说明,接下来我们用一个例子来说明CountDownLatch是如何控制代码等多个线程执行完毕后才继续往下执行。假设我们有五个线程并发执行,我们通知了他们,然后他们多线程执行。等到他们都执行完了,又接着往下做。以下示例代码可以作为参考。读者可以复制代码到自己的IDE进行调试运行:

import java.util.Random;
import java.util.concurrent.CountDownLatch;

/**
 * 多个线程都全部执行完毕后,再往下执行
 * @Author: LiYang
 * @Date: 2019/9/10 19:53
 */
public class WaitForUs {

    /**
     * 多线程工具类CountDownLatch的一个用法示例
     * @param args
     */
    public static void main(String[] args) {
        //执行多线程之前的代码,本例中就是通知大家到目的地
        System.out.println("通知,大家快来这个地方");
        System.out.println("=========== 可以理解为多线程之前 ===========");

        //总的线程数为5(大家可以换这个变量,进行调试)
        int threadNum = 5;

        //创建一个CountDownLatch类,构造入参线程数
        CountDownLatch countDownLatch = new CountDownLatch(threadNum);

        //创建5个线程,并发同步执行
        for (int i = 0; i < threadNum; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    //10000以内的随机数,作为等待时间
                    int waitMillis = new Random().nextInt(10000);

                    try {
                        //模拟线程执行,耗费一定的时间(本例是10秒内)
                        Thread.sleep(waitMillis);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }

                    //线程执行完毕,报告结果
                    System.out.println(Thread.currentThread().getName() + "已到达目的地,所用毫秒数:" + waitMillis);

                    //CountDownLatch类计数减一
                    //注意,CountDownLatch类的实例要在新建多线程之前,然后入参需要
                    //统一完成后才往下执行的线程数。然后每个线程执行完后,或者
                    //部分执行完后,调用之前创建的CountDownLatch类的实例的countDown方法
                    countDownLatch.countDown();
                }
            }).start();
        }

        try {
            //下面这句代码,CountDownLatch就阻塞在这里了
            //直到countDown()到0了(从构造入参的线程数开始减)
            //也即是所有线程都countDown了,
            //则解除阻塞,代码继续往下执行
            //注意,这句代码放在多线程countDown之后,多线程全部完成后
            //继续往下执行的代码之前,起一个分界线的作用
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        //由于CountDownLatch发挥的作用,等到上面的5个线程都执行完毕了,
        //也就是都打印了"已到达目的地",接下来才会往下执行,打印下面这句话
        System.out.println("=========== 可以理解为多线程之后 ===========");
        System.out.println("大家都到齐了,那就开始吧");
    }

}

运行程序,控制台输出以下内容,完全符合我们的预期,多线程得到了很好的控制:

通知,大家快来这个地方
=========== 可以理解为多线程之前 ===========
Thread-0已到达目的地,所用毫秒数:2060
Thread-2已到达目的地,所用毫秒数:3080
Thread-3已到达目的地,所用毫秒数:4757
Thread-4已到达目的地,所用毫秒数:7331
Thread-1已到达目的地,所用毫秒数:7643
=========== 可以理解为多线程之后 ===========
大家都到齐了,那就开始吧