
前面我们举了一个例子,说明 CountDownLatch 闭锁的运用,本文我们讲解一下 CyclicBarrier 的运用,这两个经常被面试官放在一起问区别,实际上记住和理解我举的这两个例子就基本掌握了它们的区别。
CyclicBarrier 类似于闭锁,与闭锁的关键区别在于,闭锁用于等待事件(上次我们的例子中,主线程等待其他 cache 加载完成才继续执行,没加载完成就阻塞主线程),栅栏用于等待其他线程,其作用是让一组线程到达一个屏障时被阻塞,直到最后一个线程到达屏障时,屏障才会开门,所有被屏障拦截的线程才会继续运行。CountDownLatch 倾向于一个线程等多个线程,CyclicBarrier 倾向于多个线程互相等待。
栅栏的作用就类似于运动员赛跑时候的发令枪,裁判没打发令枪之前,谁也不准跑
考虑这么一个问题,现在要对你们线上的数据库做压测。你可能想到这么几步:
第 2 步准备数据的是时候,你起了 1w 个线程去准备数据,但是 load 数据是需要时间的,这个时候如果有的线程先请求了数据库,有的线程后请求数据库,对于数据库的 QPS 就是 < 1w 的,所以需要他们同时并发的去请求。在 load 数据这个阶段我们用 CyclicBarrier 去拦住他们,当所有的线程都 load 好数据之后,同时请求数据库,对数据库进行压测。
public class CyclicBarrierTest { private static CyclicBarrier cyclicBarrier; public static class PrepareReadDataThread extends Thread { private int id; PrepareReadDataThread(int id) { this.id = id; } public void run() { try { System.out.println("线程:" + this.id + " 可以读数据啦!"); cyclicBarrier.await(); } catch (Exception e) { e.printStackTrace(); } System.out.println("所有任务完毕,继续处理其他任务... "+System.currentTimeMillis()); } } public static class PrepareWriteDataThread extends Thread { private int id; PrepareWriteDataThread(int id) { this.id = id; } public void run() { try { Thread.sleep(3000); System.out.println("线程:" + this.id + " 可以 load 写数据完成!"); cyclicBarrier.await(); } catch (Exception e) { e.printStackTrace(); } System.out.println("所有任务完毕,继续处理其他任务... "+System.currentTimeMillis()); } } public static void main(String[] args) { cyclicBarrier = new CyclicBarrier(5, () -> System.out.println("测试开始!")); for (int i = 0; i < 5; i++) { new PrepareReadDataThread(i).start(); } //重置一下 cyclicBarrier.reset(); for (int i = 0; i < 5; i++) { new PrepareWriteDataThread(i).start(); } } } 1.CountDownLatch 的计数器只能使用一次,而 CyclicBarrier 的计数器可以使用 reset()方法重置。所以 CyclicBarrier 能处理更为复杂的业务场景。例如,如果计算发生错误,可以重置计数器,并让线程重新执行一次
2.CyclicBarrier 还提供其他有用的方法,比如 getNumberWaiting 方法可以获得 Cyclic-Barrier 阻塞的线程数量。isBroken()方法用来了解阻塞的线程是否被中断
3.CountDownLatch 倾向于一个线程等多个线程,CyclicBarrier 倾向于多个线程互相等待