回环屏障CyclicBarrier详解-蒲公英云

CountDownLatch的计数器是一次性的，也就是等到计数器值变为0后，再调用CountDownLatch的await和countdown方法都会立刻返回，这就起不到线程同步的效果了。CyclicBarrier可以让一组线程全部达到一个状态后再全部同时执行。这里之所以叫做回环是因为所有等待线程执完毕，并重置CyclicBarrier的状态后它可以被重用。之所以叫做屏障是因为线程调用await方法后就会被阻塞，这个阻塞点就成为屏障点，等所有线程都调用了await方法后，线程就会冲破屏障，继续向下运行。

1、案例介绍

使用两个线程去执行一个被分解的任务A，当两个线程把自己的任务都执行完毕后再对它们的结果进行汇总处理。

public class CycleBarrierTest1 {
    // 创建一个CyclicBarrier实例，添加一个所有子线程全部到达屏障后执行的任务
    private static CyclicBarrier cyclicBarrier = new CyclicBarrier(2, new Runnable() {
        @Override
        public void run() {
            System.out.println(Thread.currentThread() + "task1 merge result");
        }
    });
    public static void main(String[] args) throws InterruptedException {
        // 创建一个线程数固定为2的线程池
        ExecutorService executorService = Executors.newFixedThreadPool(2);
        // 将线程A添加到线程池
        executorService.submit(new Runnable() {
            @Override
            public void run() {
                try {
                    System.out.println(Thread.currentThread() + "task1-1");
                    System.out.println(Thread.currentThread() + "enter in barrier");
                    cyclicBarrier.await();
                    System.out.println(Thread.currentThread() + "enter out barrier");
                }catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        // 将线程B添加到线程池
        executorService.submit(new Runnable() {
            @Override
            public void run() {
                try {
                    System.out.println(Thread.currentThread() + "task1-2");
                    System.out.println(Thread.currentThread() + "enter in barrier");
                    cyclicBarrier.await();
                    System.out.println(Thread.currentThread() + "enter out barrier");
                }catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        // 关闭线程池
        executorService.shutdown();
    }
}

执行程序：

Thread[pool-1-thread-1,5,main]task1-1
Thread[pool-1-thread-2,5,main]task1-2
Thread[pool-1-thread-1,5,main]enter in barrier
Thread[pool-1-thread-2,5,main]enter in barrier
Thread[pool-1-thread-2,5,main]task1 merge result
Thread[pool-1-thread-2,5,main]enter out barrier
Thread[pool-1-thread-1,5,main]enter out barrier

如上代码创建了一个CyclicBarrier对象，其第一个参数为计数器初始值，第二个参数Runable是当计数器值为0时需要执行的任务。
在main函数里面首先创建了一个大小为2的线程池，然后添加两个子任务到线程池，每个子任务在执行自己的逻辑后会调用await方法。
一开始计数器值为2，当第一个线程调用await方法时，计数器值会递减为1.由于计数器值不为0，所以当前线程就到了屏障点而被阻塞。然后第二个线程调用await时，会进入屏障，计数器值也会递减，现在计数器值为0，这时就会去执行CyclicBarrier构造函数中的任务，执行完毕后退出屏障点，并且唤醒被阻塞的第二个线程，这时候第一个线程也会退出屏障点继续向下执行。

2、案例2

假设一个任务由阶段1、阶段2和阶段3组成，每个线程要串行地执行阶段1、阶段2和阶段3，当多个线程执行该任务时，必须要保证所有线程的阶段1全部完成后才能进入阶段2执行，当所有线程的阶段2全部完成后才能进入阶段3执行：

import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class CycleBarrierTest2 {
    // 创建一个CyclicBarrier实例
    private static CyclicBarrier cyclicBarrier = new CyclicBarrier(2);
    public static void main(String[] args) throws InterruptedException {
        ExecutorService executorService = Executors.newFixedThreadPool(2);
        // 将线程A添加到线程池
        executorService.submit(new Runnable() {
            @Override
            public void run() {
                try {
                    System.out.println(Thread.currentThread() + "step1");
                    cyclicBarrier.await();
                    System.out.println(Thread.currentThread() + "step2");
                    cyclicBarrier.await();
                    System.out.println(Thread.currentThread() + "step3");
                }catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        // 将线程B添加到线程池
        executorService.submit(new Runnable() {
            @Override
            public void run() {
                try {
                    System.out.println(Thread.currentThread() + "step1");
                    cyclicBarrier.await();
                    System.out.println(Thread.currentThread() + "step2");
                    cyclicBarrier.await();
                    System.out.println(Thread.currentThread() + "step3");
                }catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        // 关闭线程池
        executorService.shutdown();
    }
}

执行程序：

Thread[pool-1-thread-2,5,main]step1
Thread[pool-1-thread-1,5,main]step1
Thread[pool-1-thread-1,5,main]step2
Thread[pool-1-thread-2,5,main]step2
Thread[pool-1-thread-2,5,main]step3
Thread[pool-1-thread-1,5,main]step3

如上代码，每个子线程在执行完阶段1后都调用了await方法，等到所有线程都到达屏障点后才会一块往下执行，这就保证了所有线程都完成了阶段1后才会开始执行阶段2.然后在阶段2后面调用了await方法，这保证了所有线程都完成了阶段2后，才开始阶段3的执行。这个功能使用单个CountDownLatch是无法完成的。