Java並發包工具類CountDownLatch的應用詳解

Posted on by

CountDownLatch是Java並發包中非常實用的一個工具類，它可以幫助我們實現線程之間的同步和協作。CountDownLatch的核心思想是通過計數器來控制線程的執行順序。當計數器的值降為0時，所有等待的線程都會被喚醒，然後開始執行下一步操作。

1.CountDownLatch的源碼解讀

在Java中，CountDownLatch的實現是基於AbstractQueuedSynchronizer類的。AbstractQueuedSynchronizer是一個非常重要的同步器，Java中的許多並發類都是基於它來實現的，例如Semaphore、ReentrantLock、ReadWriteLock等。

CountDownLatch的核心實現類是Sync，它是一個繼承自AbstractQueuedSynchronizer的內部類。下面是Sync類的源碼：

private static final class Sync extends AbstractQueuedSynchronizer {
    Sync(int count) {
        setState(count);
    }

    int getCount() {
        return getState();
    }

    protected int tryAcquireShared(int acquires) {
        return (getState() == 0) ? 1 : -1;
    }

    protected boolean tryReleaseShared(int releases) {
        for (;;) {
            int c = getState();
            if (c == 0)
                return false;
            int nextc = c-1;
            if (compareAndSetState(c, nextc))
                return nextc == 0;
        }
    }
}

Sync類中有三個重要的方法：

  • tryAcquireShared(int acquires)：嘗試獲取鎖，如果計數器的值等於0，表示所有線程都已經執行完畢，返回1，否則返回-1，表示獲取鎖失敗。
  • tryReleaseShared(int releases)：釋放鎖，將計數器的值減1，並返回減1後的計數器的值。如果計數器的值減為0，表示所有線程都已經執行完畢，返回true，否則返回false。
  • getCount()：返回當前計數器的值。

tryAcquireShared()方法是CountDownLatch的關鍵所在，它會嘗試獲取鎖。如果計數器的值等於0，說明所有線程都已經執行完畢，可以返回1，表示獲取鎖成功；否則返回-1，表示獲取鎖失敗。這裡使用瞭AbstractQueuedSynchronizer類的基礎方法，即getState()方法，該方法用於獲取同步器的狀態。

tryReleaseShared()方法用於釋放鎖，將計數器的值減1，並返回減1後的計數器的值。如果計數器的值減為0，表示所有線程都已經執行完畢，返回true，否則返回false。這裡使用瞭AtomicInteger類的基礎方法，即compareAndSetState()方法，該方法用於比較並設置同步器的狀態。

2.CountDownLatch的原理解析

CountDownLatch的工作原理非常簡單，它通過計數器來控制線程的執行順序。當計數器的值降為0時，所有等待的線程都會被喚醒，然後開始執行下一步操作。

CountDownLatch是一個多線程協作的工具類，它允許一個或多個線程等待其他線程完成某個操作後再繼續執行。CountDownLatch有一個計數器，當計數器的值變為0時，等待的線程就會被喚醒。CountDownLatch的使用方式非常簡單，主要包括兩個方法：await()和countDown()。

  • await()方法：該方法會阻塞當前線程，直到計數器的值變為0。
  • countDown()方法：該方法會將計數器的值減1。

下面是一個簡單的示例代碼：

public class CountDownLatchDemo {
    public static void main(String[] args) throws InterruptedException {
        final int count = 3;
        final CountDownLatch latch = new CountDownLatch(count);

        for (int i = 0; i < count; i++) {
            new Thread(() -> {
                // 線程執行任務
                System.out.println(Thread.currentThread().getName() + " 執行任務...");
                // 任務執行完畢，計數器減1
                latch.countDown();
            }).start();
        }

        // 等待所有任務執行完畢
        latch.await();
        System.out.println("所有任務執行完畢...");
    }
}

在該示例代碼中，我們創建瞭一個CountDownLatch對象，並將計數器初始化為3。然後創建瞭3個線程，每個線程執行一個任務，任務執行完畢後，將計數器減1。最後，在主線程中調用latch.await()方法等待所有任務執行完畢。

CountDownLatch的實現原理是基於AbstractQueuedSynchronizer類的。當我們調用await()方法時，線程會嘗試獲取鎖，如果計數器的值不為0，則獲取鎖失敗，線程會被加入到同步隊列中阻塞。當我們調用countDown()方法時，計數器的值會減1，如果計數器的值減為0，表示所有線程都已經執行完畢，此時同步隊列中的線程會被喚醒，繼續執行下一步操作。

具體來說，在Sync類中，tryAcquireShared(int acquires)方法會嘗試獲取鎖，如果計數器的值等於0，表示所有線程都已經執行完畢，返回1，否則返回-1，表示獲取鎖失敗。tryReleaseShared(int releases)方法用於釋放鎖，將計數器的值減1，並返回減1後的計數器的值。如果計數器的值減為0，表示所有線程都已經執行完畢，返回true，否則返回false。

3.CountDownLatch的應用場景

CountDownLatch是一個非常實用的工具類，它可以幫助我們實現線程之間的同步和協作。下面介紹一些CountDownLatch的常見應用場景：

  • 等待多個線程執行完畢：如果有多個線程需要執行，但是必須等待所有線程都執行完畢才能進行下一步操作，可以使用CountDownLatch來實現。我們可以創建一個CountDownLatch對象，並將計數器的值初始化為線程數，每個線程執行完畢後，調用countDown()方法將計數器減1。最後，在主線程中調用await()方法等待所有線程執行完畢。
  • 控制線程的執行順序：如果有多個線程需要按照特定的順序執行，可以使用CountDownLatch來實現。我們可以創建多個CountDownLatch對象，每個對象的計數器的值都為1，表示隻有一個線程可以執行。線程執行完畢後，調用下一個CountDownLatch對象的countDown()方法，喚醒下一個線程。
  • 等待外部事件的發生：如果我們需要等待一個外部事件的發生，例如某個網絡連接的建立或某個文件的讀取完成，可以使用CountDownLatch來實現。我們可以在主線程中創建一個CountDownLatch對象，並將計數器的值初始化為1，然後在另一個線程中等待外部事件的發生。當外部事件發生時，調用CountDownLatch對象的countDown()方法，喚醒主線程繼續執行。
  • 控制並發線程數：如果我們需要控制並發線程的數量，可以使用CountDownLatch來實現。我們可以創建一個CountDownLatch對象，並將計數器的值初始化為線程數量，每個線程執行完畢後，調用countDown()方法將計數器減1。如果某個線程需要等待其他線程執行完畢，可以調用await()方法等待計數器的值變為0。

4.總結

CountDownLatch是一個非常實用的工具類，它可以幫助我們實現線程之間的同步和協作。CountDownLatch的使用非常簡單，隻需要調用兩個方法：await()和countDown()。CountDownLatch的實現原理是基於AbstractQueuedSynchronizer類的，它通過同步隊列來實現線程之間的協作。

CountDownLatch的應用場景非常廣泛，包括等待多個線程執行完畢、控制線程的執行順序、等待外部事件的發生、控制並發線程數等。在實際開發中，我們可以根據具體的需求來選擇合適的應用場景，使用CountDownLatch來實現線程之間的同步和協作，從而提高程序的性能和可靠性。

到此這篇關於Java並發包工具類CountDownLatch的應用詳解的文章就介紹到這瞭,更多相關Java CountDownLatch內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：