Java中的CyclicBarrier同步屏障詳解

更新時間：2023年09月27日 11:24:05 作者：weixin_34234721

這篇文章主要介紹了Java中的CyclicBarrier同步屏障詳解,CyclicBarrier也叫同步屏障,在JDK1.5被引入,可以讓一組線程達到一個屏障時被阻塞,直到最后一個線程達到屏障時,屏障才會開門,所有被阻塞的線程才會繼續(xù)執(zhí)行,需要的朋友可以參考下

CyclicBarrier是什么？

CyclicBarrier也叫同步屏障，在JDK1.5被引入，可以讓一組線程達到一個屏障時被阻塞，直到最后一個線程達到屏障時，屏障才會開門，所有被阻塞的線程才會繼續(xù)執(zhí)行。他的主要用途是控制多個線程互相等待，只有當多個線程都到達時，這些線程才會繼續(xù)執(zhí)行。CyclicBarrier好比一扇門，默認情況下關閉狀態(tài)，堵住了線程執(zhí)行的道路，直到所有線程都就位，門才打開，讓所有線程一起通過。

CyclicBarrier如何使用和工作？

API CyclicBarrier有兩個構造函數(shù)

public CyclicBarrier(int parties)
public CyclicBarrier(int parties, Runnable barrierAction)

第一個參數(shù)，其參數(shù)表示屏障攔截的線程數(shù)量，每個線程調用await方法告訴CyclicBarrier已經到達屏障位置，線程被阻塞。

第二個參數(shù)，表示線程都處于barrier時，一起執(zhí)行之前，其中barrierAction任務會在所有線程到達屏障后執(zhí)行。讓線程處于barrier狀態(tài)的方法await()

public int await()
public int await(long timeout, TimeUnit unit)

第一個默認方法，表示要等到所有的線程都處于barrier狀態(tài)，才一起執(zhí)行

第二個方法，指定了等待的時間，當所有線程沒有都處于barrier狀態(tài)，又到了指定的時間，所在的線程就繼續(xù)執(zhí)行了。

其它的一些方法

獲取當前有多少個線程阻塞等待在臨界點上
int getNumberWaiting()
用于查詢阻塞等待的線程是否被中斷
boolean isBroken()

CyclicBarrier是通過維護計數(shù)器來實現(xiàn)的。線程執(zhí)行 await() 方法之后計數(shù)器會減 1，并進行等待，直到計數(shù)器為 0，所有調用 await() 方法而在等待的線程才能繼續(xù)執(zhí)行。

CyclicBarrier的底層原理

CyclicBarrier類是concurrent并發(fā)包下的一工具類。

CyclicBarrier實現(xiàn)主要基于ReentrantLock。

線程間同步阻塞是使用的是ReentrantLock，可重入鎖線程間通信使用的是Condition，Condition 將 Object 監(jiān)視器方法（wait、notify 和 notifyAll）分解成截然不同的對象，以便通過將這些對象與任意 Lock 實現(xiàn)組合使用。

public class CyclicBarrier {
    private static class Generation {
        boolean broken = false;
    }
    /** The lock for guarding barrier entry */
    private final ReentrantLock lock = new ReentrantLock();
    /** Condition to wait on until tripped */
    private final Condition trip = lock.newCondition();
    /** The number of parties */
    private final int parties;
    /* The command to run when tripped */
    private final Runnable barrierCommand;
    /** The current generation */
    private Generation generation = new Generation();
    ...省略后面代碼
}

其中Generation用來控制屏障的循環(huán)使用，如果generation.broken為true的話，說明這個屏障已經損壞，當某個線程await的時候，直接拋出異常

await實現(xiàn)

private int dowait(boolean timed, long nanos)
    throws InterruptedException, BrokenBarrierException,
           TimeoutException {
    final ReentrantLock lock = this.lock;
    lock.lock();
    try {
        final Generation g = generation;
        if (g.broken)
            throw new BrokenBarrierException();
        if (Thread.interrupted()) {
            breakBarrier();
            throw new InterruptedException();
        }
        int index = --count;
        if (index == 0) {  // tripped
            boolean ranAction = false;
            try {
                final Runnable command = barrierCommand;
                if (command != null)
                    command.run();
                ranAction = true;
                nextGeneration();
                return 0;
            } finally {
                if (!ranAction)
                    breakBarrier();
            }
        }
        // loop until tripped, broken, interrupted, or timed out
        for (;;) {
            try {
                if (!timed)
                    trip.await();
                else if (nanos > 0L)
                    nanos = trip.awaitNanos(nanos);
            } catch (InterruptedException ie) {
                if (g == generation && ! g.broken) {
                    breakBarrier();
                    throw ie;
                } else {
                    // We're about to finish waiting even if we had not
                    // been interrupted, so this interrupt is deemed to
                    // "belong" to subsequent execution.
                    Thread.currentThread().interrupt();
                }
            }
            if (g.broken)
                throw new BrokenBarrierException();
            if (g != generation)
                return index;
            if (timed && nanos <= 0L) {
                breakBarrier();
                throw new TimeoutException();
            }
        }
    } finally {
        lock.unlock();
    }
}

每當線程執(zhí)行await，內部變量count減1，如果count！= 0，說明有線程還未到屏障處，則在鎖條件變量trip上等待。
當count == 0時，說明所有線程都已經到屏障處，執(zhí)行條件變量的signalAll方法喚醒等待的線程。

其中 nextGeneration方法可以實現(xiàn)屏障的循環(huán)使用：

重新生成Generation對象
恢復count值

CountDownLatch與CyclicBarrier的比較

CountDownLatch與CyclicBarrier都是用于控制并發(fā)的工具類，都可以理解成維護的就是一個計數(shù)器，但是這兩者還是各有不同側重點的：

CountDownLatch一般用于某個線程A等待若干個其他線程執(zhí)行完任務之后，它才執(zhí)行；而CyclicBarrier一般用于一組線程互相等待至某個狀態(tài)，然后這一組線程再同時執(zhí)行；CountDownLatch強調一個線程等多個線程完成某件事情。CyclicBarrier是多個線程互等，等大家都完成，再攜手共進。
調用CountDownLatch的countDown方法后，當前線程并不會阻塞，會繼續(xù)往下執(zhí)行；而調用CyclicBarrier的await方法，會阻塞當前線程，直到CyclicBarrier指定的線程全部都到達了指定點的時候，才能繼續(xù)往下執(zhí)行；
CountDownLatch方法比較少，操作比較簡單，而CyclicBarrier提供的方法更多，比如能夠通過getNumberWaiting()，isBroken()這些方法獲取當前多個線程的狀態(tài)，并且CyclicBarrier的構造方法可以傳入barrierAction，指定當所有線程都到達時執(zhí)行的業(yè)務功能；
CountDownLatch是不能復用的，而CyclicLatch是可以復用的。

和 CountdownLatch 相似，都是通過維護計數(shù)器來實現(xiàn)的。線程執(zhí)行 await() 方法之后計數(shù)器會減 1，并進行等待，直到計數(shù)器為 0，所有調用 await() 方法而在等待的線程才能繼續(xù)執(zhí)行。

CyclicBarrier 和 CountdownLatch 的一個區(qū)別是，CyclicBarrier 的計數(shù)器通過調用 reset() 方法可以循環(huán)使用，所以它才叫做循環(huán)屏障。

CyclicBarrier的demo

public class CyclicBarrierExample {
    public static void main(String[] args) {
        final int totalThread = 3;
        CyclicBarrier cyclicBarrier = new CyclicBarrier(totalThread);
        ExecutorService executorService = Executors.newCachedThreadPool();
        for (int i = 0; i < totalThread; i++) {
            executorService.execute(() -> {
                System.out.println("到達屏障..");
                try {
                    cyclicBarrier.await();
                } catch (InterruptedException | BrokenBarrierException e) {
                    e.printStackTrace();
                }
                System.out.println("屏障之后開始執(zhí)行..");
            });
        }
        executorService.shutdown();
    }
}