Java併發中的Condition是如何實現的

原文:http://youngforzy.top/2017/12/01/Java%E5%B9%B6%E5%8F%91%E4%B9%8BCondition%E7%9A%84%E5%AE%9E%E7%8E%B0%E5%88%86%E6%9E%90/

一、Condition 的概念

介紹

回憶 synchronized 關鍵字,它配合 Object 的 wait()、notify() 系列方法可以實現等待/通知模式。對於 Lock,通過 Condition 也可以實現等待/通知模式。Condition 是一個接口。Condition 接口的實現類是 Lock(AQS)中的 ConditionObject。Lock 接口中有個 newCondition() 方法,通過這個方法可以獲得 Condition 對象(其實就是 ConditionObject )。因此,通過 Lock 對象可以獲得 Condition 對象。

Lock lock  = new ReentrantLock();
Condition c1 = lock.newCondition();
Condition c2 = lock.newCondition();

二、Condition 的實現分析

實現

ConditionObject 類是 AQS 的內部類,實現了 Condition 接口。

public class ConditionObject implements Condition, java.io.Serializable {
        private transient Node firstWaiter;
        private transient Node lastWaiter;
        ...

可以看到,等待隊列和同步隊列一樣,使用的都是同步器 AQS 中的節點類 Node。同樣擁有首節點和尾節點,每個 Condition 對象都包含着一個 FIFO 隊列。結構圖如下:

等待

調用 Condition 的 await() 方法會使線程進入等待隊列,並釋放鎖,線程狀態變爲等待狀態。

public final void await() throws InterruptedException {
            if (Thread.interrupted())
                throw new InterruptedException();
            Node node = addConditionWaiter();
            //釋放同步狀態(鎖)
            int savedState = fullyRelease(node);
            int interruptMode = 0;
            //判斷節點是否放入同步對列
            while (!isOnSyncQueue(node)) {
                //阻塞
                LockSupport.park(this);
                //如果已經中斷了,則退出
                if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
                    break;
            }
            if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
                interruptMode = REINTERRUPT;
            if (node.nextWaiter != null) // clean up if cancelled
                unlinkCancelledWaiters();
            if (interruptMode != 0)
                reportInterruptAfterWait(interruptMode);
        }

分析上述方法的大概過程:

  1. 將當前線程創建爲節點,加入等待隊列。
  2. 釋放鎖,喚醒同步隊列中的後繼節點。
  3. while 循環判斷節點是否放入同步隊列;如果沒有放入則阻塞繼續 while 循環(如果已經中斷則退出);如果放入則退出 while 循環執行後面的判斷。
  4. 退出 while 說明節點已經在同步隊列中,調用 acquireQueued() 方法加入同步狀態競爭。
  5. 競爭到鎖後從 await() 方法返回,即退出該方法。

addConditionWaiter() 方法:

private Node addConditionWaiter() {
            Node t = lastWaiter;
            if (t != null && t.waitStatus != Node.CONDITION) {
                //清除條件隊列中所有狀態不爲Condition的節點
                unlinkCancelledWaiters();
                t = lastWaiter;
            }
            //將該線程創建節點,放入等待隊列
            Node node = new Node(Thread.currentThread(), Node.CONDITION);
            if (t == null)
                firstWaiter = node;
            else
                t.nextWaiter = node;
            lastWaiter = node;
            return node;
        }

過程分析:同步隊列的首節點移動到等待隊列。加入尾節點之前會清除所有狀態不爲 Condition 的節點。

通知

調用 Condition 的 signal() 方法可以喚醒等待隊列的首節點(等待時間最長),喚醒之前會將該節點移動到同步隊列中。

public final void signal() {
            //判斷是否獲取了鎖
            if (!isHeldExclusively())
                throw new IllegalMonitorStateException();
            Node first = firstWaiter;
            if (first != null)
                doSignal(first);
        }

過程:

  1. 先判斷當前線程是否獲取了鎖。
  2. 然後對首節點調用 doSignal() 方法。
private void doSignal(Node first) {
            do {
                if ( (firstWaiter = first.nextWaiter) == null)
                    lastWaiter = null;
                first.nextWaiter = null;
            } while (!transferForSignal(first) &&
                     (first = firstWaiter) != null);
        }

過程:

  1. 修改首節點。
  2. 調用 transferForSignal() 方法將節點移動到同步隊列。
final boolean transferForSignal(Node node) {
        //將節點狀態變爲0   
        if (!compareAndSetWaitStatus(node, Node.CONDITION, 0))
            return false;
        //將該節點加入同步隊列
        Node p = enq(node);
        int ws = p.waitStatus;
        //如果結點p的狀態爲cancel 或者修改waitStatus失敗,則直接喚醒
        if (ws > 0 || !compareAndSetWaitStatus(p, ws, Node.SIGNAL))
            LockSupport.unpark(node.thread);
        return true;
    }

調用同步器的 enq 方法,將節點移動到同步隊列,滿足條件後使用 LockSupport 喚醒該線程。

當 Condition 調用 signalAll() 方法:

public final void signalAll() {
    if (!isHeldExclusively())
        throw new IllegalMonitorStateException();
    Node first = firstWaiter;
    if (first != null)
        doSignalAll(first);
}

private void doSignalAll(Node first) {
    lastWaiter = firstWaiter = null;
    do {
        Node next = first.nextWaiter;
        first.nextWaiter = null;
        transferForSignal(first);
        first = next;
    } while (first != null);
}

可以看到 doSignalAll() 方法使用了 do-while 循環來喚醒每一個等待隊列中的節點,直到 first 爲 null 時停止循環。

一句話總結 signalAll() 的作用:將等待隊列中的全部節點移動到同步隊列中,並喚醒每個節點的線程。

總結

整個過程可以分爲三步:

第一步:一個線程獲取鎖後,通過調用 Condition 的 await() 方法,會將當前線程先加入到等待隊列中,並釋放鎖。然後就在 await() 中的一個 while 循環中判斷節點是否已經在同步隊列,是則嘗試獲取鎖,否則一直阻塞。

第二步:當線程調用 signal() 方法後,程序首先檢查當前線程是否獲取了鎖,然後通過 doSignal(Node first) 方法將節點移動到同步隊列,並喚醒節點中的線程。

第三步:被喚醒的線程,將從 await() 中的 while 循環中退出來,然後調用 acquireQueued() 方法競爭同步狀態。競爭成功則退出 await() 方法繼續執行。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章