首页 > 代码库 > 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