多线程并发工具之cyclicBarrier

cyclicBarrier是一种控制线程并发工具，跟CountDownLatch类似，不同的是countDownLatch多线程下是一次性的。用过之后就不能在使用。

画个图来解释下，假设现在有3个线程编号1,2,3 三个线程同时开始，线程1,2到达第一个await这时就需要等到3到达await才可以接着往前走。这个时候1,2,3三个线程同时获取锁同时触发往下一个await走。这个就是cyclicBarrier跟countDownLatch不同的地方。countDownLatch到达第一个await等待最后一个线程到之后就不能再加锁了。

下面是代码实现

package concurrent.tools;

import java.util.concurrent.CyclicBarrier;

public class CyclicBarrierDemo2 {

    static class TaskThread extends Thread {

        CyclicBarrier barrier;

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

        @Override
        public void run() {
            try {
                Thread.sleep(1000);
                System.out.println(getName() + " 到达栅栏 A");
                barrier.await();
                System.out.println(getName() + " 冲破栅栏 A");
                Thread.sleep(2000);
                System.out.println(getName() + " 到达栅栏 B");
                barrier.await();//最后await的线程   执行 barrierAction操作
                System.out.println(getName() + " 冲破栅栏 B");
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

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

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

}

执行结果如下 

当然这时跑第一个await的同学不愿意了，第一次我跑了第一想要炫耀下怎么办呢？

CyclicBarrier给我们提供一个小功能就是barrierAction，等到最后一个线程到达await之后就会新创建一个小线程来执行一些动作。例如记录每个线程到达await的名次。

直接上代码，只需要在创建的时候，barrierAction参数传入新建的线程就好。

CyclicBarrier barrier = new CyclicBarrier(threadNum, new Runnable() {
            @Override
            public void run() {
                System.out.println(Thread.currentThread().getName() + " 完成最后任务");
            }
        });

修改之后的执行结果