JUC之AQS—Cyclicbarrier

导读:这篇文章介绍的是java并发组件aqs之CyclicBarrier

Cyclicbarrier概念:
  • Cyclicbarrier是一个同步辅助类,它允许一组线程相互等待直到到达某个公共的屏障点,通过它可以完成多个线程之间相互等待,直有当每个线程都准备就绪后才能各自继续往下执行后面的操作,和countdownlatch有相似的地方,都是通过计数器来实现的,当某个线程调用了await()方法之后该线程就进入了等待状态而且计数器执行的是加一的操作,当计数器的值达到我们设置的初始值的时候,因为调用了await()方法进入等待状态的线程会被唤醒继续执行它们后续的操作。由于CyclicBarrier释放等待线程后可以重用所以我们又称为循环屏障。
    cyclicbarrier
Cyclicbarrier使用场景:
  • Cyclicbarrier的使用场景跟Countdownlatch很相似,可以用于多线程计算数据最后合并计算结果的应用场景。比如我们需要统计多个Excel中的数据,然后等到一个总结果。我们可以通过多线程处理每一个Excel,执行完成后得到相应的结果,最后通过Barrieraction来计算这些线程的计算结果,得到所有Excel的总和。


    CyclicBarrier中的方法

下面通过实例来展示下Cyclicbarrier的使用

  • 1、示列一
@Slf4j
public class CyclicBarrierExample1 {

    private static CyclicBarrier barrier = new CyclicBarrier(5); //定义5个线程同步等待

    public static void main(String[] args) throws Exception {

        ExecutorService executor = Executors.newCachedThreadPool();
        for (int i = 0; i < 10; i++) {
            final int threadNum = i;
            Thread.sleep(1000);
            executor.execute(() -> {
                try {
                    race(threadNum);
                } catch (Exception e) {
                    log.error("exception", e);
                }
            });
        }
        executor.shutdown();
    }

    private static void race(int threadNum) throws Exception {
        Thread.sleep(1000);
        log.info("{} is ready", threadNum);
        barrier.await();  //线程调用await()告诉当前有个线程ok了
        log.info("{} continue", threadNum); //到达定义的数目时,await()方法之后的操作就可以执行
    }
}

返回结果:可以看出通过下面的返回结果。代码中定义了5个线程同步等待,当rece()中每个线程调用await()告诉当前线程ok了,当累计到我们之前定义的线程数时,await()方法后的操作开始执行。

示列一返回结果
  • 2、示列二
@Slf4j
public class CyclicBarrierExample2 {

    private static CyclicBarrier barrier = new CyclicBarrier(5);

    public static void main(String[] args) throws Exception {

        ExecutorService executor = Executors.newCachedThreadPool();

        for (int i = 0; i < 5; i++) {
            final int threadNum = i;
            Thread.sleep(1000);
            executor.execute(() -> {
                try {
                    race(threadNum);
                } catch (Exception e) {
                    log.error("exception", e);
                }
            });
        }
        executor.shutdown();
    }

    private static void race(int threadNum) throws Exception {
        Thread.sleep(1000);
        log.info("{} is ready", threadNum);
        try {
            // 由于状态可能会改变,所以会抛出BarrierException异常,如果想继续往下执行,需要加上try-catch
            barrier.await(2000, TimeUnit.MILLISECONDS); //用于做超时处理
        } catch (Exception e) {
            log.warn("BarrierException", e);
        }
        log.info("{} continue", threadNum);
    }
}

返回结果:注意查看下返回结果的时间,示例二和示例一的代码差不多,我们加了个await的超时处理,
这个过程中如果超时状态可能会改变,所以会抛出BarrierException异常,这时候如果要让后续的操作继续执行,得捕捉异常。

示例二返回结果
  • 3、示列三
@Slf4j
public class CyclicBarrierExample3 {

    private static CyclicBarrier barrier = new CyclicBarrier(5, () -> {
        // 当线程全部到达屏障时,优先执行这里的runnable
        log.info("callback is running");
    });

    public static void main(String[] args) throws Exception {

        ExecutorService executor = Executors.newCachedThreadPool();

        for (int i = 0; i < 10; i++) {
            final int threadNum = i;
            Thread.sleep(1000);
            executor.execute(() -> {
                try {
                    race(threadNum);
                } catch (Exception e) {
                    log.error("exception", e);
                }
            });
        }
        executor.shutdown();
    }

    private static void race(int threadNum) throws Exception {
        Thread.sleep(1000);
        log.info("{} is ready", threadNum);
        barrier.await();
        log.info("{} continue", threadNum);
    }
}

返回结果:这里我们定义CyclicBarrier(int parties, Runnable barrierAction),指定了一个runnable就是在线程到达屏障的时候优先执行runnable。

示列三返回示列

总结:CyclicBarrier的内部是使用重入锁ReentrantLock和Condition,在CyclicBarrier中最重要的方法莫过于await()方法,表示在此的所有线程都已经在此barrier上调用了await()方法之前,将一直等待。同时await()方法内部调用了dowait(boolean timed, long nanos)方法,更多详细的可以看CyclicBarrier的类源码(目前先不记录了),await()的处理逻辑还是比较简单的:如果该线程不是到达的最后一个线程,则他会一直处于等待状态,除非发生以下情况:
1、最后一个线程到达,即index == 0
2、超出了指定时间(超时等待)
3、其他的某个线程中断当前线程
4、其他的某个线程中断另一个等待的线程
5、其他的某个线程在等待barrier超时
6、其他的某个线程在此barrier调用reset()方法。reset()方法用于将屏障重置为初始状态。

你可能感兴趣的:(JUC之AQS—Cyclicbarrier)