一、场景描述

在多线程程序设计中,经常会遇到一个线程等待一个或多个线程的场景

例如:百米赛跑,十名运动员同时起跑,由于速度的快慢,肯定有先到达和后到达的,而终点有个统计成绩的仪器,当所有选手到达终点时,它会统计所有人的成绩并进行排序,然后把结果发送到汇报成绩的系统。

再例如:当我们需要解析一个Excel里多个sheet的数据时,可以考虑使用多线程,每个线程解析一个sheet里的数据,等到所有的sheet都解析完之后,程序需要提示解析完成。在这个需求中,要实现主线程等待所有线程完成sheet的解析操作。

以上场景是一个线程等待多个线程,则就可以使用CountDownLatch来实现比较好的控制

二、CountDownLatch如何工作的

正如Java文档所描述的那样,CountDownLatch是一个同步工具类,它允许一个或多个线程一直等待,直到其他线程的操作执行完后再执行。CountDownLatch是在Java1.5被引入的,跟它一起被引入的并发工具类还有CyclicBarrier、Semaphore、ConcurrentHashMap和BlockingQueue,它们都存在于java.util.concurrent包下。

CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值到达0时,它表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务。

三、方法说明

public void CountDownLatch(int count) {…}

构造器中的计数值(count)实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,而且CountDownLatch没有提供任何机制去重新设置这个计数值

public void countDown()

其他线程必须引用闭锁对象,因为他们需要通知CountDownLatch对象,他们已经完成了各自的任务。这种通知机制是通过CountDownLatch.countDown()方法来完成的,每调用一次这个方法,在构造函数中初始化的count值就减1.所以当N个线程都调用了这个方法,count的值等于0,然后主线程就能通过await()方法,恢复执行自己的任务。

public void await()

当线程调用了 await(),则这个线程就等待这个计数器变为0,当这个计数器变为0时,这个线程继续自己下面的工作。

代码示例:

package com.itmyhome;

import java.util.concurrent.CountDownLatch;

/**
 * 运动员
 * @author itmyhome
 *
 */
class Athlete implements Runnable {
    private String name; // 姓名
    private CountDownLatch beginSignal;
    private CountDownLatch endSignal;

    public Athlete(String name, CountDownLatch begin, CountDownLatch end) {
        this.name = name;
        this.beginSignal = begin;
        this.endSignal = end;
    }

    @Override
    public void run() {
        try {
            beginSignal.await();
            System.out.println("运动员:" + name + " 到达终点...");
            endSignal.countDown();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

/**
 * 终点统计类
 * @author itmyhome
 *
 */
public class CountDownLatchTest {
    public static void main(String[] args) {
        CountDownLatch begSignal = new CountDownLatch(1);
        CountDownLatch endSignal = new CountDownLatch(10);

        for (int i = 0; i < 10; i++) {
            new Thread(new Athlete("name" + i, begSignal, endSignal)).start();
        }

        try {
            begSignal.countDown(); // 统一起跑
            endSignal.await(); // 等待运动员到达终点
            // Thread.sleep(1);
            System.out.println("结果发送到汇报成绩的系统");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

输出结果:

运动员:name0 到达终点...
运动员:name4 到达终点...
运动员:name9 到达终点...
运动员:name5 到达终点...
运动员:name3 到达终点...
运动员:name6 到达终点...
运动员:name1 到达终点...
运动员:name7 到达终点...
运动员:name8 到达终点...
运动员:name2 到达终点...
结果发送到汇报成绩的系统