并发编程（五）AbstractQueuedSynchronizer(AQS)-CountDownLatch、CyclicBarrier

CountDownLatch

CountDownLatch是什么？

CountDownLatch这个类能够使一个线程等待其他线程完成各自的工作后再执行。例如，应用程序的主线程希望在负责启动框架服务的线程已经启动所有的框架服务之后再执行。

CountDownLatch如何工作？

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

CountDownLatch是一次性的，计算器的值只能在构造方法中初始化一次，之后没有任何机制再次对其设置值，当CountDownLatch使用完毕后，它不能再次被使用。

特性

• CountDownLatch相当于一个“门栓”，一个“闸门”，只有它开启了，代码才能继续往下执行。通常情况下，如果当前线程需要等其他线程执行完成后才能执行，我们就可以使用CountDownLatch。
• 使用CountDownLatch#await方法阻塞等待一个“闸门”的开启。
• 使用CountDownLatch#countDown方法减少闸门所等待的任务数。
• CountDownLatch基于共享锁实现。
• CountDownLatch是一次性的，“闸门”开启后，无法再重复使用（同一个CountDownLatch任务不可重复执行），如果想重复使用，应该用CyclicBarrier

CountDownLatch应用场景例子

比如2个人去游乐场玩，假设想玩过山车和海盗船，买完票后一起玩，但是都排了好长好长的队，预计过山车排队需要20分钟，海盗船排队需要40分钟，如果这两个人分别去排队，同时买2个人的票，排队仅需40分钟，如果一个人分别排队的话排队需要60分钟

启动2个线程同时干2件事情，2个线程都完成后，在进行主线程的流程，否则主线程处于等待中

比如有1个excel表格，需要把表格内的多个sheet页内数据做统计，最终做汇总

示例

public class CountDownLaunchSample {

    public static void main(String[] args) throws InterruptedException {
        long now = System.currentTimeMillis();
        // 设置信号量
        CountDownLatch countDownLatch = new CountDownLatch(2);
        // 海盗船
        new Thread(new PirateShipTask(countDownLatch)).start();
        // 过山车
        new Thread(new RollerCoasterTask(countDownLatch)).start();
        //等待线程池中的2个任务执行完毕，否则一直等待
        countDownLatch.await();
        System.out.println("开始玩:"+(System.currentTimeMillis()-now));
    }
}

/**
 * 海盗船
 */
public class PirateShipTask implements Runnable {
    private CountDownLatch countDownLatch;

    public PirateShipTask(CountDownLatch countDownLatch){
        this.countDownLatch = countDownLatch;
    }

    public void run() {
        try {
            System.out.println("开始排队海盗船");
            Thread.sleep(2000);
            System.out.println("海盗船排队完成");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            if (countDownLatch != null)
                countDownLatch.countDown();
        }
    }

}

/**
 * 过山车
 */
public class RollerCoasterTask implements Runnable {

    private CountDownLatch countDownLatch;

    public RollerCoasterTask(CountDownLatch countDownLatch){
        this.countDownLatch = countDownLatch;
    }
    public void run() {
        try {
            System.out.println("开始排队过山车");
            Thread.sleep(5000);
            System.out.println("过山车排队成功");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            if (countDownLatch != null)
                countDownLatch.countDown();
        }
    }
}

当线程都执行完成后，才继续运行主线程代码

源码流程图

CyclicBarrier

栅栏屏障，让一组线程到达一个屏障（也可以叫同步点）时被阻塞，直到最后一个线程到达屏障时，屏障才会开门，所有被屏障拦截的线程才会继续运行。 CyclicBarrier默认的构造方法是CyclicBarrier（int parties），其参数表示屏障拦截的线程数量，每个线程调用await方法告CyclicBarrier我已经到达了屏障，然后当前线程被阻塞。

• 可重复的使用
• 与CountDownLatch区别是可以重复运行
• 它的作用就是会让所有线程都等待完成后才会继续下一步行动。

场景

可以用于多线程计算数据，最后合并计算结果的场景。

示例

public class CyclicBarrierTest implements Runnable {
    private CyclicBarrier cyclicBarrier;
    private int index ;

    public CyclicBarrierTest(CyclicBarrier cyclicBarrier, int index) {
        this.cyclicBarrier = cyclicBarrier;
        this.index = index;
    }

    public void run() {
        try {
            System.out.println("index: " + index);
            index--;
            cyclicBarrier.await();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws Exception {
        // 例如俄罗斯方块，当满一行后才会清除（当线程都结束了后才运行主线程）
        // 与CountDownLatch区别是可以重复运行
        CyclicBarrier cyclicBarrier = new CyclicBarrier(5, new Runnable() {
            public void run() {
                System.out.println("准备开始执行任务");
            }
        });
        for (int i = 0; i < 4; i++) {
            new Thread(new CyclicBarrierTest(cyclicBarrier, i)).start();
        }
        cyclicBarrier.await();
        System.out.println("清除一行");

        for (int i = 0; i < 4; i++) {
            new Thread(new CyclicBarrierTest(cyclicBarrier, i)).start();
        }
        cyclicBarrier.await();
        System.out.println("清除下一行");
    }

}

CyclicBarrier 与 CountDownLatch 区别

• CountDownLatch 是一次性的，CyclicBarrier 是可循环利用的
• CountDownLatch 参与的线程的职责是不一样的，有的在倒计时，有的在等待倒计时结束。CyclicBarrier 参与的线程职责是一样的。