JAVA多线程-CyclicBarrier同步屏障

一、概述

CyclicBarrier,也是位于java.util.concurrent包下的一个并发同步工具类,光从字面上理解的意思是可循环使用(Cyclic)的屏障(Barrier)。

主要的作用是:可以让一组线程到达一个屏障(也可以叫同步点)时被阻塞,直到这组线程中的最后一个线程到达屏障时,屏障才会被移除,然后就可以使得所有被屏障拦截的线程能够继续往下干活。我们暂且把这个状态就叫做barrier,当调用await()方法之后,线程就处于barrier了。

二、应用场景

CyclicBarrier可以用于多线程计算数据,最后合并计算结果的应用场景。比如我们用一个Excel文件保存了用户的所有银行流水,然后每个sheet工作簿保存一个账户最近一年的每笔流水记录,现在需要统计用户的日均银行流水,那么就可以使用多线程处理每个sheet工作簿里的银行流水,全部执行完之后,得到每个sheet的日均银行流水,最后,再一起统计这些线程的计算结果,从而最终计算出整个Excel的日均银行流水。

三、使用方法

3.1、默认构造器

public CyclicBarrier(int parties, Runnable barrierAction) {
    if (parties <= 0) throw new IllegalArgumentException();
    this.parties = parties;
    this.count = parties;
    this.barrierCommand = barrierAction;
}

public CyclicBarrier(int parties) {
    this(parties, null);
}

其中,parties这个参数表示屏障拦截的线程数量,每个线程调用await()方法告诉CyclicBarrier我已经到达了屏障,然后当前线程被阻塞,而barrierAction这个参数的作用则是:当这些线程都到达了屏障时,可以额外做某些事情的,其本身是传入了一个Runnable类型的参数,然后可以在run()方法中可以额外做一些事情。

3.2、重要方法

public int await() throws InterruptedException, BrokenBarrierException {
    try {
        return dowait(false, 0L);
    } catch (TimeoutException toe) {
        throw new Error(toe); // cannot happen
    }
}

通常比较常用,可以用于挂起一组线程,直至所有的线程都到达屏障了再同时执行后续任务。

public int await(long timeout, TimeUnit unit)
        throws InterruptedException,
               BrokenBarrierException,
               TimeoutException {
    return dowait(true, unit.toNanos(timeout));
}

可以让这些那些先到达屏障状态的线程等待一定的时间,如果此时还有线程没有到达屏障则直接让已经到达屏障的线程继续执行后续任务。

四、代码实例

  1. 假若有若干个线程都要进行写数据操作,并且只有所有线程都完成写数据操作之后,这些线程才能继续做后面的事情,此时就可以利用CyclicBarrier了。
package com.feizi.java.concurrency.tool;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;

/**
 * Created by feizi on 2018/6/4.
 */
public class TestCyclicBarrier {
    public static void main(String[] args) {
        int N = 4;
        CyclicBarrier barrier = new CyclicBarrier(N);
        for (int i = 0; i < N; i++){
            new Writer(barrier).start();
        }
    }

    static class Writer extends Thread{
        private CyclicBarrier barrier;

        public Writer(CyclicBarrier barrier) {
            this.barrier = barrier;
        }

        @Override
        public void run() {
            System.out.println("线程" + Thread.currentThread().getName() + "正在写入数据...");
            try {
                //模拟写入耗时操作
                Thread.sleep(5000);
                System.out.println("线程" + Thread.currentThread().getName() + "写入数据完毕,等待其他线程写入完毕...");

                barrier.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
            System.out.println("所有线程写入完毕,继续处理其他任务...");
        }
    }
}

控制台输出结果:

线程Thread-1正在写入数据...
线程Thread-0正在写入数据...
线程Thread-3正在写入数据...
线程Thread-2正在写入数据...
线程Thread-1写入数据完毕,等待其他线程写入完毕...
线程Thread-2写入数据完毕,等待其他线程写入完毕...
线程Thread-3写入数据完毕,等待其他线程写入完毕...
线程Thread-0写入数据完毕,等待其他线程写入完毕...
所有线程写入完毕,继续处理其他任务...
所有线程写入完毕,继续处理其他任务...
所有线程写入完毕,继续处理其他任务...
所有线程写入完毕,继续处理其他任务...

Process finished with exit code 0

从上面输出结果可以看到,每个线程执行完毕之后,都在等待其他线程执行完毕,当所有线程都执行完毕之后,则他们就可以继续进行后续的任务了。

  1. 如果想在所有线程都到达屏障后,还需要进行一些额外的操作,那么这个时候可以使用上面说的第二个构造函数了,传入一个Runnable参数进行处理。
package com.feizi.java.concurrency.tool;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;

/**
 * Created by feizi on 2018/6/5.
 */
public class TestCyclicBarrierDemo {
    public static void main(String[] args) {
        int N = 4;

        CyclicBarrier barrier = new CyclicBarrier(N, new Runnable() {
            @Override
            public void run() {
                System.out.println("当前线程" + Thread.currentThread().getName() + "被翻牌了...");
            }
        });

        for (int i = 0; i < N; i++){
            new Writer(barrier).start();
        }
    }

    static class Writer extends Thread{
        private CyclicBarrier barrier;

        public Writer(CyclicBarrier barrier) {
            this.barrier = barrier;
        }

        @Override
        public void run() {
            System.out.println("线程" + Thread.currentThread().getName() + "正在写入数据...");
            try {
                //模拟耗时操作
                Thread.sleep(5000);

                System.out.println("线程" + Thread.currentThread().getName() + "写入数据完毕...");
                barrier.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
            System.out.println("所有线程写入完毕,继续处理其他任务...");
        }
    }
}

控制台输出结果:

线程Thread-0正在写入数据...
线程Thread-1正在写入数据...
线程Thread-2正在写入数据...
线程Thread-3正在写入数据...
线程Thread-1写入数据完毕...
线程Thread-2写入数据完毕...
线程Thread-3写入数据完毕...
线程Thread-0写入数据完毕...
当前线程Thread-0被翻牌了...
所有线程写入完毕,继续处理其他任务...
所有线程写入完毕,继续处理其他任务...
所有线程写入完毕,继续处理其他任务...
所有线程写入完毕,继续处理其他任务...

Process finished with exit code 0

从上面的输出结果我们可以看到,当4个线程都到达屏障后,CyclicBarrier会从这组线程中选择一个线程(随机选择,跟CPU调度有关,谁被分配了时间片,就谁去执行)去执行Runnable中的内容,也就是额外的操作了。我们看到上面选择的是Thread-0这个线程。

  1. 如果想让那些先到达屏障状态的线程等待一定的时间,然后如果此时还有线程没有到达屏障则直接让已经到达屏障的线程继续执行后续任务。那么我们可以使用上面介绍的这个public int await(long timeout, TimeUnit unit)方法。
package com.feizi.java.concurrency.tool;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

/**
 * Created by feizi on 2018/6/5.
 */
public class TestCyclicBarrierDemo1 {

    public static void main(String[] args) {
        int N = 4;
        CyclicBarrier barrier = new CyclicBarrier(N);

        for (int i = 0; i < N; i++){
            if(i < N -1){
                new Writer(barrier).start();
            }else {
                try {
                    Thread.sleep(5000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                new Writer(barrier).start();
            }
        }
    }

    static class Writer extends Thread{
        private CyclicBarrier barrier;

        public Writer(CyclicBarrier barrier) {
            this.barrier = barrier;
        }

        @Override
        public void run() {
            System.out.println("线程 " + Thread.currentThread().getName() + "正在写入数据...");

            try {
                //模拟耗时操作
                Thread.sleep(1000);

                System.out.println("线程 " + Thread.currentThread().getName() + "写入数据完毕, 等待其他线程写入完毕...");

                try {
                    barrier.await(5000, TimeUnit.MILLISECONDS);
                } catch (BrokenBarrierException e) {
                    e.printStackTrace();
                } catch (TimeoutException e) {
                    e.printStackTrace();
                }
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

            System.out.println(Thread.currentThread().getName() + ", 所有线程写入完毕,继续处理其他任务...");
        }
    }
}

控制台输出结果:

线程 Thread-0正在写入数据...
线程 Thread-2正在写入数据...
线程 Thread-1正在写入数据...
线程 Thread-0写入数据完毕, 等待其他线程写入完毕...
线程 Thread-2写入数据完毕, 等待其他线程写入完毕...
线程 Thread-1写入数据完毕, 等待其他线程写入完毕...
线程 Thread-3正在写入数据...
java.util.concurrent.BrokenBarrierException
    at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:250)
    at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
    at com.feizi.java.concurrency.tool.TestCyclicBarrierDemo1$Writer.run(TestCyclicBarrierDemo1.java:49)
java.util.concurrent.TimeoutException
    at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:257)
    at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
    at com.feizi.java.concurrency.tool.TestCyclicBarrierDemo1$Writer.run(TestCyclicBarrierDemo1.java:49)
java.util.concurrent.BrokenBarrierException
    at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:250)
    at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
    at com.feizi.java.concurrency.tool.TestCyclicBarrierDemo1$Writer.run(TestCyclicBarrierDemo1.java:49)
java.util.concurrent.BrokenBarrierException
    at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:207)
    at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
    at com.feizi.java.concurrency.tool.TestCyclicBarrierDemo1$Writer.run(TestCyclicBarrierDemo1.java:49)
线程 Thread-3写入数据完毕, 等待其他线程写入完毕...
Thread-2, 所有线程写入完毕,继续处理其他任务...
Thread-0, 所有线程写入完毕,继续处理其他任务...
Thread-1, 所有线程写入完毕,继续处理其他任务...
Thread-3, 所有线程写入完毕,继续处理其他任务...

Process finished with exit code 0

从上面输出结果,我们可以看到,因为在main方法测试的时候,我们特意指定了让最后一个线程延迟启动,由于在前面三个线程都到达了屏障之后,在等待了指定的时间之后发现第4个线程还没有到达屏障,然后就会抛出异常而去继续执行后面的任务了。

五、CyclicBarrier和CountDownLatch的区别

  1. CountDownLatch的计数器只能使用一次。而CyclicBarrier的计数器可以使用reset() 方法重置。所以CyclicBarrier能处理更为复杂的业务场景,比如如果计算发生错误,可以重置计数器,并让线程们重新执行一次。

  2. CyclicBarrier还提供其他有用的方法,比如getNumberWaiting方法可以获得CyclicBarrier阻塞的线程数量。isBroken方法用来知道阻塞的线程是否被中断。

除此之外,CyclicBarrier还可以进行重用,即重复使用:

package com.feizi.java.concurrency.tool;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;

/**
 * Created by feizi on 2018/6/5.
 */
public class TestCyclicBarrierDemo2 {
    public static void main(String[] args) {
        int N = 4;

        CyclicBarrier barrier = new CyclicBarrier(N);

        for (int i = 0; i < N; i++){
            new Writer(barrier).start();
        }

        try {
            Thread.sleep(10000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println("CyclicBarrierz重用1...");

        for (int i = 0; i < N; i++){
            new Writer(barrier).start();
        }

        try {
            Thread.sleep(10000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println("CyclicBarrierz重用2...");

        for (int i = 0; i < N; i++){
            new Writer(barrier).start();
        }
    }

    static class Writer extends Thread{
        private CyclicBarrier barrier;

        public Writer(CyclicBarrier barrier) {
            this.barrier = barrier;
        }

        @Override
        public void run() {
            System.out.println("线程 " + Thread.currentThread().getName() + " 正在写入数据...");
            try {
                Thread.sleep(5000);

                System.out.println("线程 " + Thread.currentThread().getName() + " 写入数据完毕,等待其他线程写入完毕...");
                barrier.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }

            System.out.println(Thread.currentThread().getName() + ", 所有线程写入完毕,继续处理其他任务...");
        }
    }
}

控制台输出结果:

线程 Thread-0 正在写入数据...
线程 Thread-1 正在写入数据...
线程 Thread-2 正在写入数据...
线程 Thread-3 正在写入数据...
线程 Thread-0 写入数据完毕,等待其他线程写入完毕...
线程 Thread-3 写入数据完毕,等待其他线程写入完毕...
线程 Thread-1 写入数据完毕,等待其他线程写入完毕...
线程 Thread-2 写入数据完毕,等待其他线程写入完毕...
Thread-3, 所有线程写入完毕,继续处理其他任务...
Thread-0, 所有线程写入完毕,继续处理其他任务...
Thread-2, 所有线程写入完毕,继续处理其他任务...
Thread-1, 所有线程写入完毕,继续处理其他任务...
CyclicBarrierz重用1...
线程 Thread-4 正在写入数据...
线程 Thread-5 正在写入数据...
线程 Thread-6 正在写入数据...
线程 Thread-7 正在写入数据...
线程 Thread-5 写入数据完毕,等待其他线程写入完毕...
线程 Thread-4 写入数据完毕,等待其他线程写入完毕...
线程 Thread-6 写入数据完毕,等待其他线程写入完毕...
线程 Thread-7 写入数据完毕,等待其他线程写入完毕...
Thread-5, 所有线程写入完毕,继续处理其他任务...
Thread-6, 所有线程写入完毕,继续处理其他任务...
Thread-4, 所有线程写入完毕,继续处理其他任务...
Thread-7, 所有线程写入完毕,继续处理其他任务...
CyclicBarrierz重用2...
线程 Thread-8 正在写入数据...
线程 Thread-9 正在写入数据...
线程 Thread-10 正在写入数据...
线程 Thread-11 正在写入数据...
线程 Thread-9 写入数据完毕,等待其他线程写入完毕...
线程 Thread-10 写入数据完毕,等待其他线程写入完毕...
线程 Thread-8 写入数据完毕,等待其他线程写入完毕...
线程 Thread-11 写入数据完毕,等待其他线程写入完毕...
Thread-11, 所有线程写入完毕,继续处理其他任务...
Thread-8, 所有线程写入完毕,继续处理其他任务...
Thread-10, 所有线程写入完毕,继续处理其他任务...
Thread-9, 所有线程写入完毕,继续处理其他任务...

Process finished with exit code 0

从上面输出结果,我们可以看到,在首次的4个线程穿过屏障之后,又可以继续重新一轮的使用,而CountDownLatch则无法做到重复使用。

六、原文参考

  1. http://ifeve.com/concurrency-cyclicbarrier/
  2. http://www.importnew.com/21889.html
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,233评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,013评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,030评论 0 241
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,827评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,221评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,542评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,814评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,513评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,225评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,497评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,998评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,342评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,986评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,812评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,560评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,461评论 2 266

推荐阅读更多精彩内容