CyclicBarrier简介和使用

115 阅读2分钟

​本文已参与「新人创作礼」活动,一起开启掘金创作之路。

 CyclicBarrier简介

和CountDowmLatch相似,当所有线程都到达barrier栏栅时,所有线程再一起向下执行。

CyclicBarrier使用场景

开启多个线程去执行任务最后汇总

CountDownLaatch的实现如下

CountDownLatch理论实现

  • 因为任务是一份,线程是多份。在任务节点上发出线程能感知的信号,即可控制每个来执行此任务的线程。
  • 例如:统一吃饭:停住就要用countdownlatch的await方法。所有执行此任务的线程,走到这里都会停住。等待远方一声countdown命令才继续执行。
  • 例如:大家都吃完:每个线程吃完饭都countdown想继续,在远方有个await收集都有谁想继续呀?收集够了所有人的,那就大家一起继续睡觉吧!

代码实现

@Slf4j
public class TestCountDownLatch implements Runnable {
    //有多少干饭的兄弟?
    private static final int HOW_MANY_BROTHER = 3;
    //有多少老大管事呢?
    private static final int HOW_MANY_BOSS = 1;
    //等待同一时间开饭的停住口令,(由主线程下达开动命令)
    private static CountDownLatch startEatingTogether = new CountDownLatch(HOW_MANY_BOSS);
    //等待所有人都吃完,这口令创建时候就要定义好需要收集几次(由主线程下达开动命令)
    private static CountDownLatch  waitAllFinish= new CountDownLatch(HOW_MANY_BROTHER);

    //所有线程都要做这件事,定义任务
    @SneakyThrows
    @Override
    public void run() {
        //停在这等口令
        startEatingTogether.await();
        //如果得到命令,继续向下执行,干饭3秒钟
        log.info("兄弟--{}--号--干饭!",Thread.currentThread().getName());
        Thread.sleep(3000);
        //告诉老大干完饭,想继续,等老大收集齐大家的信号,就会继续向下执行
        log.info("兄弟--{}--号--干完饭想睡觉!",Thread.currentThread().getName());
        waitAllFinish.countDown();
        //结束
    }
    //老大(主线程)来统一安排
    public static void main(String[] args) throws InterruptedException {
        ExecutorService executorService = Executors.newFixedThreadPool(HOW_MANY_BROTHER);
        //来三个线程跑起来
        for (int i = 0; i < 3; i++) {
            executorService.execute(new TestCountDownLatch());
        }
        //程序执行到这里,即使上面三个线程都跑起来,但仍然卡在等口令那里
        log.info("老大--{}--下达命令干饭!",Thread.currentThread().getName());
        startEatingTogether.countDown();
        waitAllFinish.await();
        log.info("老大--{}--大家都吃完了!",Thread.currentThread().getName());

    }
}

CyclicBarrier使用demo

对比 juejin.cn/post/711321… countdownlatch中的同一个例子实现如下:

@Slf4j
public class TestCyclicBarrier implements Runnable {
    private CyclicBarrier all;
    //有多少干饭的兄弟?
    private static final int HOW_MANY_BROTHER = 3;
    //有多少老大管事呢?
    private static final int HOW_MANY_BOSS = 1;
    //构造方法
    public TestCyclicBarrier(CyclicBarrier all){
       this.all= all;
    }
    //所有线程都要做这件事,定义任务
    @SneakyThrows
    @Override
    public void run() {
        //如果得到命令,继续向下执行,干饭3秒钟
        log.info("兄弟--{}--号--开始干饭!",Thread.currentThread().getName());
        Thread.sleep(new Random().nextInt(1000));
        //告诉老大干完饭,想继续,等老大收集齐大家的信号,就会继续向下执行
        log.info("兄弟--{}--号--干完饭!",Thread.currentThread().getName());
        //结束
        all.await();
    }
    //老大(主线程)来统一安排
    public static void main(String[] args) throws InterruptedException, BrokenBarrierException {
        CyclicBarrier all =new CyclicBarrier(HOW_MANY_BROTHER + HOW_MANY_BOSS, new Runnable() {
            @Override
            public void run() {
                log.info("老大--{}--都吃完了!",Thread.currentThread().getName());
            }
        });
        ExecutorService executorService = Executors.newFixedThreadPool(HOW_MANY_BROTHER);
        //来三个线程跑起来
        for (int i = 0; i < HOW_MANY_BROTHER; i++) {
            executorService.execute(new TestCyclicBarrier(all));
        }
        all.await();
    }
}