1. 程式人生 > >JDK1.8 AbstractQueuedSynchronizer的實現分析(學習筆記)

JDK1.8 AbstractQueuedSynchronizer的實現分析(學習筆記)

lock方法會呼叫acquire方法,該方法在AQS中實現

 public final void acquire(int arg) {
        if (!tryAcquire(arg) &&
            acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            selfInterrupt();
    }

正常使用一個ReentrantLock 的lock() 方法時,在不能獲得所得情況下,該方法是阻塞的,對吧
以上acquire方法拆解一下方法呼叫
首先tryAcquire(arg),英語不好的我Acquire,百度給出的意思是獲得,獲取,那麼好這裡可以給成意思是嘗試獲得的意思?
如果申請資源失敗,則返回false,那麼會先呼叫addWaiter(Node.EXCLUSIVE),這裡的Node.EXCLUSIVE代表獨佔的意思,so這個節點是獨佔的?

/**
     * Creates and enqueues node for current thread and given mode.
     *
     * @param mode Node.EXCLUSIVE for exclusive, Node.SHARED for shared
     * @return the new node
     */
    private Node addWaiter(Node mode) {
        //封裝了當前執行緒??
        Node node = new Node(Thread.currentThread(), mode);
        // Try the fast path of enq; backup to full enq on failure
Node pred = tail; if (pred != null) { //如果當前連結串列有尾節點,就把封裝當前執行緒的節點追加到尾部?這裡的尾節點操作也是CAS的 node.prev = pred; if (compareAndSetTail(pred, node)) { pred.next = node; return node; } } // 如果沒有等待的執行緒(節點列表為null),則初始化頭尾節點,該方法是一個自旋的方法
enq(node); return node; }

這裡考慮多執行緒呼叫的情況,這裡的連結串列不存線上程安全問題,因為對每個頭尾節點的操作都是CAS的。詳見程式碼片段裡的
compareAndSetHead和compareAndSetTail方法。

/**
     * Inserts node into queue, initializing if necessary. See picture above.
     * @param node the node to insert
     * @return node's predecessor
     */
    private Node enq(final Node node) {
        for (;;) {
            Node t = tail;
            //第一次的時候尾節點肯定是一個 null,則初始話一個空內容節點為頭節點,第二次迴圈時尾節點已經不是null了
            if (t == null) { // Must initialize
                if (compareAndSetHead(new Node()))
                    tail = head;
            } else {
                node.prev = t; //將node節點的前驅節點設定成尾節點,即先嚐試追加到連結串列的尾部,如果尾節點是t,則將尾節點設定成node,退出迴圈
                if (compareAndSetTail(t, node)) { 
                    t.next = node;
                    return t;
                }
            }
        }
    }

所以第一個等待鎖的執行緒A在AQS中的資料結構是這樣的?

空節點 -> node(name:A)

看到這裡,AQS中對內部連結串列的頭尾操作都是CAS的,所以連結串列中的節點排序也是完全按照申請鎖的順序排列的。addWaiter方法的操作無非是將執行緒封裝為一個node,追加該節點到連結串列的尾節點,然後addWaiter 方法返回封裝好的node方法,繼續呼叫acquireQueued(node) ,
acquireQueued 方法是一個自旋的方法,假設當前只有一個執行緒A排隊申請鎖,目前連結串列形式為。

空節點 -> node(name:A)

/**
     * Acquires in exclusive uninterruptible mode for thread already in
     * queue. Used by condition wait methods as well as acquire.
     *
     * @param node the node
     * @param arg the acquire argument
     * @return {@code true} if interrupted while waiting
     */
    final boolean acquireQueued(final Node node, int arg) {
        boolean failed = true;
        try {
            boolean interrupted = false;
            for (;;) {//自旋方法
                final Node p = node.predecessor();
                //獲得p的前驅節點,並判斷是否為head節點,如果為head節點且成功獲取到資源,就將當前執行緒節點設定成佇列的頭節點,並返回
                if (p == head && tryAcquire(arg)) {
                    setHead(node);
                    p.next = null; // help GC
                    failed = false;
                    return interrupted;
                }
                if (shouldParkAfterFailedAcquire(p, node) &&
                    parkAndCheckInterrupt())
                    interrupted = true;
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }

這裡判斷p==head,如果node的前驅節點是head,說明該節點排成了老二,而老二如果成功獲取到資源,則變成老大,那麼這裡老大的起到的是什麼作用?
>

我的程式有一把鎖在外邊阻塞中(sleep),從程式碼上看一開始tryAcquire方法就成功申請到資源,就不會有他的node維護在這裡,就是說他的鎖不需要在這裡排隊?那麼後續的unlock是怎樣的一個機制?
因為鎖在別的執行緒中持有(state>0),所以這個acquireQueued方法中的tryAcquire會返回false
然後shouldParkAfterFailedAcquire,字面意思如果申請資源失敗則判斷是否可以掛起執行緒

/**
     * Checks and updates status for a node that failed to acquire.
     * Returns true if thread should block. This is the main signal
     * control in all acquire loops.  Requires that pred == node.prev.
     *
     * @param pred node's predecessor holding status
     * @param node the node
     * @return {@code true} if thread should block
     */
    private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
        //獲取節點的前驅節點等待狀態
        int ws = pred.waitStatus;
        //如果是SIGNAL狀態,返回true,第一次來這個肯定不會成立,因為是node的前驅節點是一個空節點。
        //空節點->node(A)
        if (ws == Node.SIGNAL)
            /*
             * This node has already set status asking a release
             * to signal it, so it can safely park.
             */
            return true;
        if (ws > 0) { //Node.CANCLED 如果節點是被放棄的(什麼情況下會放棄,中斷麼??)
            /*
             * Predecessor was cancelled. Skip over predecessors and
             * indicate retry.
             */
            do {
                //一直往前找,把當前節點插入到被取消的節點的前面
                node.prev = pred = pred.prev;
            } while (pred.waitStatus > 0);
            pred.next = node;
        } else {
            /*
             * waitStatus must be 0 or PROPAGATE.  Indicate that we
             * need a signal, but don't park yet.  Caller will need to
             * retry to make sure it cannot acquire before parking.
             */
            //將前驅節點的狀態設定成SIGNAL
            compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
        }
        return false;
    }

至此針對我的測試,這個方法會走compareAndSetWaitStatus(pred, ws, Node.SIGNAL); 分支,所以目前node(name:A)節點的前驅(head節點)的waitStatus狀態為 SIGNAL,該方法的第一次呼叫會返回false
目前針對執行緒A的連結串列表示

node(waitStatus:SIGNAL)->node(name:A)

因為shouldParkAfterFailedAcquire返回了false,所以不會呼叫parkAndCheckInterrupt,那麼會進入第二次迴圈。
node的前驅節點是head 這個是毫無疑問的,但是就是因為獲取鎖的執行緒還是一個熊孩子,沒有釋放資源(state>0)
所以依舊會呼叫shouldParkAfterFailedAcquire方法,但此時方法內部已經不一樣了,這裡已經開始滿足第一個條件分支了,所以此方法此次會返回true,因為因為shouldParkAfterFailedAcquire返回了true,所以會接著呼叫parkAndCheckInterrupt()方法
該方法實現如下:

 /**
     * Convenience method to park and then check if interrupted
     *
     * @return {@code true} if interrupted
     */
    private final boolean parkAndCheckInterrupt() {
        LockSupport.park(this);//呼叫park()使執行緒進入waiting狀態
        return Thread.interrupted();//如果被喚醒,檢視自己是不是被中斷的。
    }

LockSupport.park(this); 大神說park()會讓當前執行緒進入waiting狀態。在此狀態下,有兩種途徑可以喚醒該執行緒:1)被unpark();2)被interrupt()。
如果當前執行緒waiting了,那麼就不會有第三次迴圈了….因為當前執行緒已經waiting了……
這裡阻塞的執行緒應該是申請鎖的執行緒,LockSupport.park(this);引數的作用是對應的blocker會記錄在Thread的一個parkBlocker屬性中,通過jstack命令可以非常方便的監控具體的阻塞物件.

如果此時又有一個執行緒B,按照此邏輯繼續申請一把鎖,按AQS的處理邏輯會維護成如下形式

node(waitStatus:SIGNAL)->node(name:A)->node(name:B)

假設tryAcquire(arg)依舊失敗,state依舊不給釋放,熊孩子還不還,那麼對於node(name:B) 來說,在acquireQueued中他的前驅肯定不是head節點,所以他只能走shouldParkAfterFailedAcquire方法設定前驅節點的狀態,然後呼叫parkAndCheckInterrupt()方法,去阻塞當前當前申請鎖的執行緒。
同上經歷2次自旋後在AQS中連結串列的維護成如下形式

node(waitStatus:SIGNAL)->node(name:A,waitStatus:SIGNAL)->node(name:B)

這裡貼一個大神的總結和圖

再來總結下它的流程吧:
呼叫自定義同步器的tryAcquire()嘗試直接去獲取資源,如果成功則直接返回;
沒成功,則addWaiter()將該執行緒加入等待佇列的尾部,並標記為獨佔模式;
acquireQueued()使執行緒在等待佇列中休息,有機會時(輪到自己,會被unpark())會去嘗試獲取資源。獲取到資源後才返回。如果在整個等待過程中被中斷過,則返回true,否則返回false。
如果執行緒在等待過程中被中斷過,它是不響應的。只是獲取資源後才再進行自我中斷selfInterrupt(),將中斷補上。
由於此函式是重中之重,我再用流程圖總結一下:
圖

現在AQS的連結串列裡有三個節點。

node(waitStatus:SIGNAL)->node(name:A,waitStatus:SIGNAL)->node(name:B)

假設一開始長時間阻塞不還鎖的執行緒,時間到了玩夠了回家吃飯去,執行緒中呼叫了reentrantlock.unlock(); 方法,他玩夠了我這邊兩個兄弟還在那waiting呢,是不是會釋放他的執行緒標誌
釋放鎖的程式碼也是很簡單的

    public void unlock() {
        sync.release(1);
    }

這裡sync是實現AQS的一個內部類Sync的例項,所以本質上還是在呼叫AQS的功能,看看怎麼做的。

 /**
     * Releases in exclusive mode.  Implemented by unblocking one or
     * more threads if {@link #tryRelease} returns true.
     * This method can be used to implement method {@link Lock#unlock}.
     *
     * @param arg the release argument.  This value is conveyed to
     *        {@link #tryRelease} but is otherwise uninterpreted and
     *        can represent anything you like.
     * @return the value returned from {@link #tryRelease}
     */
    public final boolean release(int arg) {
        if (tryRelease(arg)) {//釋放資源
            Node h = head; 
            if (h != null && h.waitStatus != 0) //頭結點不為null,且頭結點是有狀態的?
                unparkSuccessor(h);//喚醒等待佇列裡的下一個執行緒
            return true;
        }
        return false;
    }

由上文得知,呼叫AQS的acquire(int args)方法,會呼叫子類的tryAcquire(int args)實現從而實現對執行緒狀態的重置,那麼這裡有一個tryRelease(arg) 是不是也是子類的一個實現呢?其實就是呼叫的子類實現,依舊不去管他的實現
暫且把tryRelease當作釋放資源,如果釋放成功則將頭節點做為引數,呼叫unparkSuccessor方法。

/**
     * Wakes up node's successor, if one exists.
     *
     * @param node the node
     */
    private void unparkSuccessor(Node node) { //這裡引數是頭結點。
        /*
         * If status is negative (i.e., possibly needing signal) try
         * to clear in anticipation of signalling.  It is OK if this
         * fails or if status is changed by waiting thread.
         */
        int ws = node.waitStatus;//判斷節點的等待狀態
        if (ws < 0)  //如果當前節點是有狀態的,清除狀態
            compareAndSetWaitStatus(node, ws, 0); //置零當前執行緒所在的結點狀態,允許失敗。

        /*
         * Thread to unpark is held in successor, which is normally
         * just the next node.  But if cancelled or apparently null,
         * traverse backwards from tail to find the actual
         * non-cancelled successor.
         */
        Node s = node.next; //拿到連結串列中的第二個節點,本例是node(name:A,waitstatus:SIGNAL) 
        if (s == null || s.waitStatus > 0) { //如果沒有第二個節點或者第二個節點的狀態為取消
            s = null;
            for (Node t = tail; t != null && t != node; t = t.prev)  //從尾節點開始向前找有狀態的節點並返回
                if (t.waitStatus <= 0)
                    s = t;
        }
        if (s != null)
            LockSupport.unpark(s.thread); //喚醒執行緒
    }

依據上面的邏輯本例中的node(name:A,waitstatus:SIGNAL) 節點所持有的執行緒將得到釋放。但是這個節點並沒有被移除。AQS中的連結串列結構變成了這樣

node(waitstatus:0)->node(name:A,waitstatus:SIGNAL)->node(name:B)

此時執行緒A將得到喚醒,然後他會繼續做final boolean acquireQueued(final Node node, int arg) 方法中的自旋。為方法繼續把acquireQueued的程式碼拿出來再貼一遍

/**
     * Acquires in exclusive uninterruptible mode for thread already in
     * queue. Used by condition wait methods as well as acquire.
     *
     * @param node the node
     * @param arg the acquire argument
     * @return {@code true} if interrupted while waiting
     */
    final boolean acquireQueued(final Node node, int arg) {
        boolean failed = true;
        try {
            boolean interrupted = false;
            for (;;) {//自旋方法
                final Node p = node.predecessor();
                //獲得p的前驅節點,並判斷是否為head節點,如果為head節點且成功獲取到資源,就將當前執行緒節點設定成佇列的頭節點,並返回
                if (p == head && tryAcquire(arg)) {
                    setHead(node);
                    p.next = null; // help GC
                    failed = false;
                    return interrupted;
                }
                if (shouldParkAfterFailedAcquire(p, node) &&
                    parkAndCheckInterrupt())
                    interrupted = true;
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }

此時執行緒A發現他的前驅節點為head,並且此時再申請資源tryAcquire(arg) ,因為那個熊孩子把資源釋放了,所以此時申請資源是成功的,然後會將當前節點設定為連結串列的頭結點,並釋放原來的頭結點佔用的記憶體。
這樣連結串列結構就成了這個樣子:

node(name:A,waitstatus:SIGNAL)->node(name:B)

針對執行緒A來說acquireQueued 方法已經執行完成,假設我們一直老老實實的等沒有通知中斷執行緒,則執行緒A中的程式碼行reentrantLock.lock() 方法將返回,一個程式碼行執行完成,會接著執行下一個程式碼行,lock() 加鎖成功了!那麼問題來了(問題和挖掘機沒有任何關係)tryAcquire和tryRelease 到底在爭搶和釋放什麼東西?
繼續看tryAcquire方法在ReentrantLock類中的實現(公平鎖)

   /**
           * Fair version of tryAcquire.  Don't grant access unless
           * recursive call or no waiters or is first.
           */
          protected final boolean tryAcquire(int acquires) {
              //獲得當前執行緒
              final Thread current = Thread.currentThread();
              //getState來自AQS,放方法取AQS中的全域性變數state 注意是全域性
              int c = getState();
              if (c == 0) { //AQS中預設state值為0,如果為0應該可以證明當前AQS佇列裡的執行緒沒有修改過state,也就是說沒有人持有鎖
                  if (!hasQueuedPredecessors() &&
                      compareAndSetState(0, acquires)) {
                      setExclusiveOwnerThread(current);
                      return true;
                  }
              }
              else if (current == getExclusiveOwnerThread()) {
                  int nextc = c + acquires;
                  if (nextc < 0)
                      throw new Error("Maximum lock count exceeded");
                  setState(nextc);
                  return true;
              }
              return false;
          }

可以發現tryAcquire 方法會先獲取AQS中的全域性變數,也可以說AQS維護的連結串列中的所有節點都會去檢查state這個變數,所以這個state是一個執行緒狀態,也可以看作是他們在搶的一把鎖,如果發現state為0,則代表競爭到資源,並通過CAS的方式設定該值為1,然後設定當前執行緒為exclusiveOwnerThread,另一個條件因為鎖是可重入的啊老鐵
這裡有一個hasQueuedPredecessors() 方法,參照某大神所說

正如hasQueuedPredecessors的註釋所說,該方法的作用是為了避免 “執行緒闖入”,即對於ReentrantLock來說,即便是state為0時,AQS佇列中也可能是有節點的(被取消的,打斷的節點)等等,這個時候為了保證AQS佇列的公平性,不再嘗試加鎖,而是返回false,到AQS的佇列中去排隊。所以,這也是該方法被用在公平鎖中的原因。

遺留問題:
1.為什麼用連結串列管理執行緒
2.節點的狀態SIGNAL 主要含義
3.在AQS中的執行緒如何響應中斷,中斷策略是什麼?

   /** waitStatus value to indicate thread has cancelled */
    static final int CANCELLED =  1;
    /** waitStatus value to indicate successor's thread needs unparking */
    static final int SIGNAL    = -1;
    /** waitStatus value to indicate thread is waiting on condition */
    static final int CONDITION = -2;
    /**
     * waitStatus value to indicate the next acquireShared should
     * unconditionally propagate
     */
    static final int PROPAGATE = -3;