并发(二)CyclicBarrier

CyclicBarrier

循环屏障，用于一组固定数目的线程互相等待。使用场景如下：

主任务有一组串行的执行节点，每个节点之间有一批任务，固定数量的线程执行这些任务，执行完成后，在节点完成集合后，再继续执行下一批任务。

如下图所示：

 

屏障可以在每个节点处循环使用。构造屏障时，提供了一个可选的Runnable参数，每次执行完成后，会触发此Runnable。

    @Test
    public void test1() throws InterruptedException {
        final AtomicBoolean flag = new AtomicBoolean(false);

        CyclicBarrier barrier = new CyclicBarrier(5, new Runnable() {
            @Override
            public void run() {
                System.out.println("所有子任务运行完成!");
                flag.set(true);
            }
        });

        ExecutorService executorService = Executors.newCachedThreadPool();

        for(int i = 0; i < 5; i++) {
            executorService.execute(new Task(barrier));
        }

        while (!flag.get()) {
            Thread.sleep(100);
        }
        flag.set(false);

        for(int i = 0; i < 5; i++) {
            executorService.execute(new Task(barrier));
        }

        while (true) {
            Thread.sleep(1000);
        }
    }

    public static class Task implements Runnable {

        private CyclicBarrier barrier;

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

        @Override
        public void run() {
            try {
                Thread.sleep((long)(Math.random()*10000));
                System.err.println("now await nums=" + barrier.getNumberWaiting());
                barrier.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
        }
    }