基于CyclicBarrier和CountDownLatch的使用区别说明

Z时代
2024-01-10
分类：综合

2018.12.12更新

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


public static void main(String[] args) {
  ExecutorService pool = Executors.newCachedThreadPool();
  int size = 3;
  // 设置参数时，线程实际执行数size+1，将main线程也加到等待队列中
  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 + "位运动员都准备好了，可以起跑!");
 }
}