【并发编程】CyclicBarrier

最新推荐文章于 2023-01-22 19:58:40 发布

程序员卤蛋

最新推荐文章于 2023-01-22 19:58:40 发布

阅读量108

点赞数

分类专栏：并发编程

本文链接：https://blog.csdn.net/qq_34980601/article/details/107302798

版权

并发编程专栏收录该内容

4 篇文章 1 订阅

订阅专栏

字面意思回环栅栏，通过它可以实现让一组线程等待至某个状态之后再全部同时执行。

CyclicBarrier类位于java.util.concurrent包下，CyclicBarrier提供2个构造器：

public CyclicBarrier(int parties, Runnable barrierAction) {

}

public CyclicBarrier(int parties) {

}

然后CyclicBarrier中最重要的方法就是await方法，它有2个重载版本：参数parties指让多少个线程或者任务等待至barrier状态；参数barrierAction为当这些线程都达到barrier状态时会执行的内容。

public int await() throws InterruptedException, BrokenBarrierException { };

public int await(long timeout, TimeUnit unit)throws InterruptedException,BrokenBarrierException,TimeoutException { };

第二个版本是让这些线程等待至一定的时间，如果还有线程没有到达barrier状态就直接让到达barrier的线程执行后续任务。第一个版本比较常用，用来挂起当前线程，直至所有线程都到达barrier状态再同时执行后续任务；

举例：

假若有若干个线程都要进行写数据操作，并且只有所有线程都完成写数据操作之后，这些线程同时执行后面的事情，此时就可以利用CyclicBarrier了：

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 cyclicBarrier;

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

        @Override
        public void run() {
            System.out.println("线程" + Thread.currentThread().getName() + "正在写入数据...");
            try {
                Thread.sleep(5000);      //以睡眠来模拟写入数据操作
                System.out.println("线程" + Thread.currentThread().getName() + "写入数据完毕，等待其他线程写入完毕");
                cyclicBarrier.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
            System.out.println("并发处理任务...");
        }
    }

执行结果：


线程Thread-0正在写入数据...
线程Thread-2正在写入数据...
线程Thread-1正在写入数据...
线程Thread-3正在写入数据...
线程Thread-2写入数据完毕，等待其他线程写入完毕
线程Thread-3写入数据完毕，等待其他线程写入完毕
线程Thread-1写入数据完毕，等待其他线程写入完毕
线程Thread-0写入数据完毕，等待其他线程写入完毕
并发处理任务...
并发处理任务...
并发处理任务...
并发处理任务...

进阶1 Runnable

如果说想在所有线程写入操作完之后，进行额外的其他操作可以为CyclicBarrier提供Runnable参数：


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();
}

执行结果：当四个线程都到达barrier状态后，会从四个线程中选择一个线程去执行Runnable。


线程Thread-0正在写入数据...
线程Thread-3正在写入数据...
线程Thread-2正在写入数据...
线程Thread-1正在写入数据...
线程Thread-2写入数据完毕，等待其他线程写入完毕
线程Thread-1写入数据完毕，等待其他线程写入完毕
线程Thread-3写入数据完毕，等待其他线程写入完毕
线程Thread-0写入数据完毕，等待其他线程写入完毕
所有线程执行完毕，当前线程：Thread-1
并发处理任务...
并发处理任务...
并发处理任务...
并发处理任务...

进阶2 await指定时间


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 cyclicBarrier;
        public Writer(CyclicBarrier cyclicBarrier) {
            this.cyclicBarrier = cyclicBarrier;
        }

        @Override
        public void run() {
            System.out.println("线程"+Thread.currentThread().getName()+"正在写入数据...");
            try {
                Thread.sleep(5000);      //以睡眠来模拟写入数据操作
                System.out.println("线程"+Thread.currentThread().getName()+"写入数据完毕，等待其他线程写入完毕");
                try {
                    cyclicBarrier.await(2000, TimeUnit.MILLISECONDS);
                } catch (TimeoutException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
            } catch (InterruptedException e) {
                e.printStackTrace();
            }catch(BrokenBarrierException e){
                e.printStackTrace();
            }
            System.out.println(Thread.currentThread().getName()+"所有线程写入完毕，继续处理其他任务...");
        }
    }

执行结果：故意让最后一个线程启动延迟，因为在前面三个线程都达到barrier之后，等待了指定的时间发现第四个线程还没有达到barrier，就抛出异常并继续执行后面的任务。


线程Thread-0正在写入数据...
线程Thread-1正在写入数据...
线程Thread-2正在写入数据...
线程Thread-2写入数据完毕，等待其他线程写入完毕
线程Thread-0写入数据完毕，等待其他线程写入完毕
线程Thread-3正在写入数据...
线程Thread-1写入数据完毕，等待其他线程写入完毕
Thread-2所有线程写入完毕，继续处理其他任务...
Thread-1所有线程写入完毕，继续处理其他任务...
Thread-0所有线程写入完毕，继续处理其他任务...
java.util.concurrent.BrokenBarrierException
  at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:250)
  at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
  at Concurrent.CyclicBarrierTest$Writer.run(CyclicBarrierTest.java:88)
java.util.concurrent.TimeoutException
  at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:257)
  at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
  at Concurrent.CyclicBarrierTest$Writer.run(CyclicBarrierTest.java:88)
java.util.concurrent.BrokenBarrierException
  at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:250)
  at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
  at Concurrent.CyclicBarrierTest$Writer.run(CyclicBarrierTest.java:88)
Disconnected from the target VM, address: '127.0.0.1:56661', transport: 'socket'
java.util.concurrent.BrokenBarrierException
线程Thread-3写入数据完毕，等待其他线程写入完毕
  at java.util.concurrent.CyclicBarrier.dowait(CyclicBarrier.java:207)
Thread-3所有线程写入完毕，继续处理其他任务...
  at java.util.concurrent.CyclicBarrier.await(CyclicBarrier.java:435)
  at Concurrent.CyclicBarrierTest$Writer.run(CyclicBarrierTest.java:88)

进阶3 重用


public class Test {
    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(25000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println("CyclicBarrier重用");

        for(int i=0;i<N;i++) {
            new Writer(barrier).start();
        }
    }
    static class Writer extends Thread{
        private CyclicBarrier cyclicBarrier;
        public Writer(CyclicBarrier cyclicBarrier) {
            this.cyclicBarrier = cyclicBarrier;
        }

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

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

执行结果：

线程Thread-0正在写入数据...

线程Thread-1正在写入数据...

线程Thread-3正在写入数据...

线程Thread-2正在写入数据...

线程Thread-1写入数据完毕，等待其他线程写入完毕

线程Thread-3写入数据完毕，等待其他线程写入完毕

线程Thread-2写入数据完毕，等待其他线程写入完毕

线程Thread-0写入数据完毕，等待其他线程写入完毕

Thread-0所有线程写入完毕，继续处理其他任务...

Thread-3所有线程写入完毕，继续处理其他任务...

Thread-1所有线程写入完毕，继续处理其他任务...

Thread-2所有线程写入完毕，继续处理其他任务...

CyclicBarrier重用

线程Thread-4正在写入数据...

线程Thread-5正在写入数据...

线程Thread-6正在写入数据...

线程Thread-7正在写入数据...

线程Thread-7写入数据完毕，等待其他线程写入完毕

线程Thread-5写入数据完毕，等待其他线程写入完毕

线程Thread-6写入数据完毕，等待其他线程写入完毕

线程Thread-4写入数据完毕，等待其他线程写入完毕

Thread-4所有线程写入完毕，继续处理其他任务...

Thread-5所有线程写入完毕，继续处理其他任务...

Thread-6所有线程写入完毕，继续处理其他任务...

Thread-7所有线程写入完毕，继续处理其他任务...

从执行结果可以看出，在初次的4个线程越过barrier状态后，又可以用来进行新一轮的使用。而CountDownLatch无法进行重复使用
。