CyclicBarrier和CountDownLatch使用上的区别

2018.12.12更新
在学习了CyclicBarrier之后发现，CyclicBarrier也可以实现跟CountDownLatch类似的功能，只需要在它的parties中多设置一个数，将主线程加入等待队列就可以了：

public static void main(String[] args) {
        ExecutorService pool = Executors.newCachedThreadPool();
        int size = 3;
        CyclicBarrier cyclicBarrier = new CyclicBarrier(size + 1);
        for (int i = 0; i < size; i++) {
            int index = i;
            pool.submit(() -> {
                try {
                    TimeUnit.SECONDS.sleep(index);
                    System.out.println("第" + index + "位运动员准备好了");
                    cyclicBarrier.await();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
        }
        try {
        //主线程也加入等待
            cyclicBarrier.await();
        } catch (Exception e) {
            e.printStackTrace();
        }
        System.out.println(size + "位运动员都准备好了，可以起跑!");
    }

执行结果：
在这里插入图片描述

以下是原内容：

我在使用并发线程栅栏的时候发现了两种，分别是CyclicBarrier 和CountDownLatch。对于两者的对比的文章有很多，这里不再赘述。我来说下我的使用过程。

**需求：**有三位运动员，他们一起参加万米赛跑，但是他们准备的时间不同，要等他们都准备好了再开始一起跑。

使用CyclicBarrier 实现：

import java.util.concurrent.*;

public class RunTest {
    public static void main(String[] args) {
        ExecutorService pool = Executors.newCachedThreadPool();
        int size = 3;
        CyclicBarrier cyclicBarrier = new CyclicBarrier(size, () -> {
            System.out.println(size + "位运动员都准备好了，可以起跑！");
            pool.shutdownNow();
        });

        for (int i = 0; i < size; i++) {
            int index = i;
            pool.submit(() -> {
                try {
                    TimeUnit.SECONDS.sleep(index);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("第" + index + "位运动员准备好了");
                try {
                    cyclicBarrier.await();
                } catch (InterruptedException | BrokenBarrierException e) {
                    e.printStackTrace();
                }
            });
        }
    }
}

结果：
这里写图片描述
可以看到，三位运动员准备的时间分别是1s，2s，3s。系统等到他们都准备好了，再发出起跑的信号。在这里CyclicBarrier 做法是在自己的构造器中new了一个runnable，等待其他线程都执行完，再执行此runnable中的代码。

我们再看看CountDownLatch怎么实现：

import java.util.concurrent.*;

public class RunTest {
    public static void main(String[] args) throws InterruptedException {
        ExecutorService pool = Executors.newCachedThreadPool();
        CountDownLatch countDownLatch = new CountDownLatch(3);
        int size = 3;

        for (int i = 0; i < size; i++) {
            int index = i;
            pool.submit(() -> {
                try {
                    TimeUnit.SECONDS.sleep(index);
                    countDownLatch.countDown();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("第" + index + "位运动员准备好了");
            });
        }
        countDownLatch.await();
        System.out.println(size + "位运动员都准备好了，可以起跑!");
    }
}

结果同上：
这里写图片描述

我们可以看到，countDownLatch是采取阻塞主线程的方法实现了线程的统一。他内部有一个计数器，我们在执行完一次线程任务的时候需要手动的减一个数，在主线程中使用 **countDownLatch.await()**监控计数器的状态，知道计数器计到0为止，再执行主线程的代码。

在实际的开发中，我个人比较倾向于第二种方法，因为使用起来简单，完全满足我的需求。

CyclicBarrier和CountDownLatch使用上的区别

猜你喜欢