首页 > 代码库 > Java并发工具类(二)同步屏障CyclicBarrier
Java并发工具类(二)同步屏障CyclicBarrier
作用
CyclicBarrier是一个同步辅助类,它允许一组线程互相等待,直到到达某个公共屏障点 (common barrier point),才继续执行。
简介
CyclicBarrier 的字面意思是可循环使用(Cyclic)的屏障(Barrier)。它要做的事情是,让一组线程到达一个屏障(也可以叫同步点)时被阻塞,直到最后一个线程到达屏障时,屏障才会开门,所有被屏障拦截的线程才会继续干活。CyclicBarrier默认的构造方法是CyclicBarrier(int parties),其参数表示屏障拦截的线程数量,每个线程调用await方法告诉CyclicBarrier我已经到达了屏障,然后当前线程被阻塞。
应用场景
CyclicBarrier可以用于多线程计算数据,最后合并计算结果的应用场景。比如我们用一个Excel保存了用户所有银行流水,每个Sheet保存一个帐户近一年的每笔银行流水,现在需要统计用户的日均银行流水,先用多线程处理每个sheet里的银行流水,都执行完之后,得到每个sheet的日均银行流水,最后,再用barrierAction用这些线程的计算结果,计算出整个Excel的日均银行流水。
代码示例
实例代码如下:
1 public class CyclicBarrierTest { 2 3 static CyclicBarrier c = new CyclicBarrier(2); 4 5 public static void main(String[] args) { 6 new Thread(new Runnable() { 7 8 @Override 9 public void run() {10 try {11 c.await();12 } catch (Exception e) {13 14 }15 System.out.println(1);16 }17 }).start();18 19 try {20 c.await();21 } catch (Exception e) {22 23 }24 System.out.println(2);25 }26 }
输出
21
或者输出
12
如果把new CyclicBarrier(2)修改成new CyclicBarrier(3)则主线程和子线程会永远等待,因为没有第三个线程执行await方法,即没有第三个线程到达屏障,所以之前到达屏障的两个线程都不会继续执行。
CyclicBarrier还提供一个更高级的构造函数CyclicBarrier(int parties, Runnable barrierAction),用于在线程到达屏障时,优先执行barrierAction,方便处理更复杂的业务场景。barrierAction在CyclicBarrier的数目达到后,所有其他线程被唤醒前执行。
代码如下:
1 public class CyclicBarrierTest2 { 2 3 static CyclicBarrier c = new CyclicBarrier(2, new A()); 4 5 public static void main(String[] args) { 6 new Thread(new Runnable() { 7 8 @Override 9 public void run() {10 try {11 c.await();12 } catch (Exception e) {13 14 }15 System.out.println(1);16 }17 }).start();18 19 try {20 c.await();21 } catch (Exception e) {22 23 }24 System.out.println(2);25 }26 27 static class A implements Runnable {28 29 @Override30 public void run() {31 System.out.println(3);32 }33 34 }35 36 }
输出
312
CyclicBarrier和CountDownLatch的区别
- CountDownLatch的计数器只能使用一次。而CyclicBarrier的计数器可以使用reset() 方法重置。所以CyclicBarrier能处理更为复杂的业务场景,比如如果计算发生错误,可以重置计数器,并让线程们重新执行一次。
- CyclicBarrier还提供其他有用的方法,比如getNumberWaiting方法可以获得CyclicBarrier阻塞的线程数量。isBroken方法用来知道阻塞的线程是否被中断。比如以下代码执行完之后会返回true。
isBroken的使用代码如下:
1 import java.util.concurrent.BrokenBarrierException; 2 import java.util.concurrent.CyclicBarrier; 3 4 public class CyclicBarrierTest3 { 5 6 static CyclicBarrier c = new CyclicBarrier(2); 7 8 public static void main(String[] args) throws InterruptedException, BrokenBarrierException { 9 Thread thread = new Thread(new Runnable() {10 11 @Override12 public void run() {13 try {14 c.await();15 } catch (Exception e) {16 }17 }18 });19 thread.start();20 thread.interrupt();21 try {22 c.await();23 } catch (Exception e) {24 System.out.println(c.isBroken());25 }26 }27 }
输出
true
参考:《Java并发编程的艺术》
Java并发工具类(二)同步屏障CyclicBarrier