深入理解AbstractQueuedSynchronizer(AQS)
一、AQS簡介
在同步組件的實現中,AQS是核心部分,同步組件的實現者通過使用AQS提供的模板方法實現同步組件語義,AQS則實現了對同步狀態的管理,以及對阻塞線程進行排隊,等待通知等等一些底層的實現處理。AQS的核心也包括了這些方面:同步隊列,獨占式鎖的獲取和釋放,共享鎖的獲取和釋放以及可中斷鎖,超時等待鎖獲取這些特性的實現,而這些實際上則是AQS提供出來的模板方法,歸納整理如下:
state是一個狀態值,不同實現AbstractQueuedSynchronizer的鎖所代表的意思都不同,如果我們使用過CountDownLatch,那么它的方法countDown() 就是執行的就是-1操作,操作的其實就是這個state值,就相當于一個計數器,對于Semaphore,CyclicBarrier,他們也是通過操作state來進行計數,ReentrantLock中,也是通過操作state來實現線程是否獲取到鎖,當state為0時,代表當前鎖時空閑的,沒有被線程持有,如果state為1,則當前鎖被線程持有,如果大于1,則證明線程重入了,state+1,而對state的操作又分為獨占和共享。
對于state的操作就是整個AQS的核心,操作包括:
- 獲取、更新:getState()、setState()、compareAndSet()
- 暴露同步狀態的值:acquire、acquireInterruptibly、tryAcquireNanos、release、acquireShared、acquireSharedInterruptibly、tryAcquireSharedNanos、releaseShared
細分如下:
- 獨占式鎖:獨占的意思就是同一時間只有一個線程能操作,其他線程過來都會被阻塞,只有當前線程完成任務后釋放了資源,其他線程才能繼續獲取資源,每一個線程都與資源進行綁定, 上面我們說的ReentrantLock就是獨占鎖。同一時間只有一個線程能拿到鎖執行,鎖的狀態只有0和1兩種情況。
- 共享式鎖:同一時間有多個線程可以拿到鎖協同工作,鎖的狀態大于或等于0。
- FIFO隊列(CLH):用于競爭鎖失敗時的排隊
整個框架的關鍵就是如何管理被阻塞的線程的隊列,該隊列是嚴格的FIFO隊列,因此,框架不支持基于優先級的同步。同步隊列的最佳選擇是自身沒有使用底層鎖來構造的非阻塞數據結構,一直以來,CLH鎖僅被用于自旋鎖。因為CLH鎖可以更容易地去實現“取消(cancellation)”和“超時”功能,因此選擇了CLH鎖作為實現的基礎。CLH隊列實際上并不那么像隊列,因為它的入隊和出隊操作都與它的用途(即用作鎖)緊密相關。它是一個鏈表隊列,通過兩個字段head和tail來存取,這兩個字段是可原子更新的,兩者在初始化時都指向了一個空節點。
- 條件隊列:給維護獨占同步的類以及實現Lock接口的類使用
AQS框架提供了一個ConditionObject類,給維護獨占同步的類以及實現Lock接口的類使用。一個鎖對象可以關聯任意數目的條件對象,可以提供典型的管程風格的await、signal和signalAll操作,包括帶有超時的,以及一些檢測、監控的方法。通過修正一些設計決策,ConditionObject類有效地將條件(conditions)與其它同步操作結合到了一起。該類只支持Java風格的管程訪問規則,這些規則中,僅當當前線程持有鎖且要操作的條件(condition)屬于該鎖時,條件操作才是合法的。這樣,一個ConditionObject關聯到一個ReentrantLock上就表現的跟內置的管程(通過Object.wait等)一樣了。兩者的不同僅僅在于方法的名稱、額外的功能以及用戶可以為每個鎖聲明多個條件。ConditionObject使用了與同步器一樣的內部隊列節點。但是,是在一個單獨的條件隊列中維護這些節點的。signal操作是通過將節點從條件隊列轉移到鎖隊列中來實現的,而沒有必要在需要喚醒的線程重新獲取到鎖之前將其喚醒。
- 這兩個隊列的最基本結構就是Node了
二、同步隊列(CLH)
- CLH鎖是一個自旋鎖。能確保無饑餓性。提供先來先服務的公平性。
當共享資源被某個線程占有,其他請求該資源的線程將會阻塞,從而進入同步隊列。就數據結構而言,隊列的實現方式無外乎兩者一是通過數組的形式,另外一種則是鏈表的形式。AQS中的同步隊列則是通過鏈式方式進行實現。接下來,很顯然我們至少會抱有這樣的疑問:
我們知道在AQS有一個靜態內部類Node,而且我們知道了節點的數據結構類型,每個節點擁有其前驅和后繼節點,很顯然這是一個雙向隊列。同樣的我們可以用一段demo看一下。
public class LockDemo {private static ReentrantLock lock = new ReentrantLock();public static void main(String[] args) {for (int i = 0; i < 5; i++) {Thread thread = new Thread(() -> {lock.lock();try {Thread.sleep(10000);} catch (InterruptedException e) {e.printStackTrace();} finally {lock.unlock();}});thread.start();}} }實例代碼中開啟了5個線程,先獲取鎖之后再睡眠10S中,實際上這里讓線程睡眠是想模擬出當線程無法獲取鎖時進入同步隊列的情況。通過debug,當Thread-4(在本例中最后一個線程)獲取鎖失敗后進入同步時,AQS時現在的同步隊列如圖所示:
Thread-0先獲得鎖后進行睡眠,其他線程(Thread-1,Thread-2,Thread-3,Thread-4)獲取鎖失敗進入同步隊列,同時也可以很清楚的看出來每個節點有兩個域:prev(前驅)和next(后繼),并且每個節點用來保存獲取同步狀態失敗的線程引用以及等待狀態等信息。另外AQS中有兩個重要的成員變量:
也就是說AQS實際上通過頭尾指針來管理同步隊列,同時實現包括獲取鎖失敗的線程進行入隊,釋放鎖時對同步隊列中的線程進行通知等核心方法。其示意圖如下:
通過對源碼的理解以及做實驗的方式,現在我們可以清楚的知道2點:
1.節點的數據結構,即AQS的靜態內部類Node,節點的等待狀態等信息;
2.同步隊列是一個雙向隊列,AQS通過持有頭尾指針管理同步隊列;
那么,節點如何進行入隊和出隊是怎樣做的了?實際上這對應著鎖的獲取和釋放兩個操作:獲取鎖失敗進行入隊操作,獲取鎖成功進行出隊操作。
三、ConditionObject條件變量及條件隊列
ConditionObject是AbstractQueuedSynchronizer的內部類,他是實現線程間的同步的基礎設施,它是與鎖結合使用的(如ReentrantLock),ConditionObject實現了Condition接口,Condition接口提供了await(),signal()等方法,實現線程的掛起和喚醒,ConditionObject是一個條件變量,每個條件變量對應一個條件隊列,當調用Condition的await()被掛起的線程將會存放在條件隊列中,調用signal()時將從條件隊列中移除并放入AQS隊列中。
同步隊列和條件隊列的關系:
四、獨占鎖
獨占鎖的獲取(acquire方法)
我們繼續通過看源碼和debug的方式來看,還是以上面的demo為例,調用lock()方法是獲取獨占式鎖,獲取失敗就將當前線程加入同步隊列,成功則線程執行。而lock()方法實際上會調用AQS的acquire()方法,源碼如下
關鍵信息請看注釋,acquire根據當前獲得同步狀態成功與否做了兩件事情:
獲取同步狀態失敗,進行入隊操作。當線程獲取獨占式鎖失敗后就會將當前線程加入同步隊列,那么加入隊列的方式是怎樣的?我們接下來就應該去研究一下addWaiter()和acquireQueued()。addWaiter()源碼如下:
private Node addWaiter(Node mode) {// 1. 將當前線程構建成Node類型Node node = new Node(Thread.currentThread(), mode);// Try the fast path of enq; backup to full enq on failure// 2. 當前尾節點是否為null?Node pred = tail;if (pred != null) {// 2.2 將當前節點尾插入的方式插入同步隊列中node.prev = pred;if (compareAndSetTail(pred, node)) {pred.next = node;return node;}}// 2.1. 當前同步隊列尾節點為null,說明當前線程是第一個加入同步隊列進行等待的線程enq(node);return node; }分析可以看上面的注釋。程序的邏輯主要分為兩個部分:
另外還會有另外一個問題:如果 if (compareAndSetTail(pred, node))為false怎么辦?會繼續執行到enq()方法,同時很明顯compareAndSetTail是一個CAS操作,通常來說如果CAS操作失敗會繼續自旋(死循環)進行重試。因此,經過我們這樣的分析,enq()方法可能承擔兩個任務:
那么是不是真的就像我們分析的一樣了?只有源碼會告訴我們答案,enq()源碼如下:
private Node enq(final Node node) {for (;;) {Node t = tail;if (t == null) { // Must initialize//1. 構造頭結點if (compareAndSetHead(new Node()))tail = head;} else {// 2. 尾插入,CAS操作失敗自旋嘗試node.prev = t;if (compareAndSetTail(t, node)) {t.next = node;return t;}}} }在上面的分析中我們可以看出在第1步中會先創建頭結點,說明同步隊列是帶頭結點的鏈式存儲結構。帶頭結點與不帶頭結點相比,會在入隊和出隊的操作中獲得更大的便捷性,因此同步隊列選擇了帶頭結點的鏈式存儲結構。那么帶頭節點的隊列初始化時機是什么?自然而然是在tail為null時,即當前線程是第一次插入同步隊列。compareAndSetTail(t, node)方法會利用CAS操作設置尾節點,如果CAS操作失敗會在for (;;)死循環中不斷嘗試,直至成功return返回為止。因此,對enq()方法可以做這樣的總結:
在當前線程是第一個加入同步隊列時,調用compareAndSetHead(new Node())方法,完成鏈式隊列的頭結點的初始化;自旋不斷嘗試CAS尾插入節點直至成功為止。
現在我們已經很清楚獲取獨占式鎖失敗的線程包裝成Node然后插入同步隊列的過程了?那么緊接著會有下一個問題?在同步隊列中的節點(線程)會做什么事情了來保證自己能夠有機會獲得獨占式鎖了?帶著這樣的問題我們就來看看acquireQueued()方法,從方法名就可以很清楚,這個方法的作用就是排隊獲取鎖的過程,源碼如下:
final boolean acquireQueued(final Node node, int arg) {boolean failed = true;try {boolean interrupted = false;for (;;) {// 1. 獲得當前節點的先驅節點final Node p = node.predecessor();// 2. 當前節點能否獲取獨占式鎖 // 2.1 如果當前節點的先驅節點是頭結點并且成功獲取同步狀態,即可以獲得獨占式鎖if (p == head && tryAcquire(arg)) {//隊列頭指針用指向當前節點setHead(node);//釋放前驅節點p.next = null; // help GCfailed = false;return interrupted;}// 2.2 獲取鎖失敗,線程進入等待狀態等待獲取獨占式鎖if (shouldParkAfterFailedAcquire(p, node) &&parkAndCheckInterrupt())interrupted = true;}} finally {if (failed)cancelAcquire(node);} }程序邏輯通過注釋已經標出,整體來看這是一個這又是一個自旋的過程(for (;😉),代碼首先獲取當前節點的先驅節點,如果先驅節點是頭結點的并且成功獲得同步狀態的時候(if (p == head && tryAcquire(arg))),當前節點所指向的線程能夠獲取鎖。反之,獲取鎖失敗進入等待狀態。整體示意圖為下圖:
獲取鎖成功,出隊操作
獲取鎖的節點出隊的邏輯是:
//隊列頭結點引用指向當前節點 setHead(node); //釋放前驅節點 p.next = null; // help GC failed = false; return interrupted;setHead()方法為:
private void setHead(Node node) {head = node;node.thread = null;node.prev = null; }將當前節點通過setHead()方法設置為隊列的頭結點,然后將之前的頭結點的next域設置為null并且pre域也為null,即與隊列斷開,無任何引用方便GC時能夠將內存進行回收。示意圖如下:
經過上面的分析,獨占式鎖的獲取過程也就是acquire()方法的執行流程如下圖所示:
獨占鎖的釋放就相對來說比較容易理解了,廢話不多說先來看下源碼:
public final boolean release(int arg) {if (tryRelease(arg)) {Node h = head;if (h != null && h.waitStatus != 0)unparkSuccessor(h);return true;}return false; }這段代碼邏輯就比較容易理解了,如果同步狀態釋放成功(tryRelease返回true)則會執行if塊中的代碼,當head指向的頭結點不為null,并且該節點的狀態值不為0的話才會執行unparkSuccessor()方法。unparkSuccessor方法源碼:
private void unparkSuccessor(Node node) {int ws = node.waitStatus;if (ws < 0)compareAndSetWaitStatus(node, ws, 0);//頭節點的后繼節點Node s = node.next;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)//后繼節點不為null時喚醒該線程LockSupport.unpark(s.thread); }源碼的關鍵信息請看注釋,首先獲取頭節點的后繼節點,當后繼節點的時候會調用LookSupport.unpark()方法,該方法會喚醒該節點的后繼節點所包裝的線程。因此,每一次鎖釋放后就會喚醒隊列中該節點的后繼節點所引用的線程,從而進一步可以佐證獲得鎖的過程是一個FIFO(先進先出)的過程。
到現在我們終于啃下了一塊硬骨頭了,通過學習源碼的方式非常深刻的學習到了獨占式鎖的獲取和釋放的過程以及同步隊列。可以做一下總結:
總體來說:在獲取同步狀態時,AQS維護一個同步隊列,獲取同步狀態失敗的線程會加入到隊列中進行自旋;移除隊列(或停止自旋)的條件是前驅節點是頭結點并且成功獲得了同步狀態。在釋放同步狀態時,同步器會調用unparkSuccessor()方法喚醒后繼節點。
我們知道lock相較于synchronized有一些更方便的特性,比如能響應中斷以及超時等待等特性,現在我們依舊采用通過學習源碼的方式來看看能夠響應中斷是怎么實現的。可響應中斷式鎖可調用方法lock.lockInterruptibly();而該方法其底層會調用AQS的acquireInterruptibly方法,源碼為:
public final void acquireInterruptibly(int arg)throws InterruptedException {if (Thread.interrupted())throw new InterruptedException();if (!tryAcquire(arg))//線程獲取鎖失敗doAcquireInterruptibly(arg); }在獲取同步狀態失敗后就會調用doAcquireInterruptibly方法:
private void doAcquireInterruptibly(int arg)throws InterruptedException {//將節點插入到同步隊列中final Node node = addWaiter(Node.EXCLUSIVE);boolean failed = true;try {for (;;) {final Node p = node.predecessor();//獲取鎖出隊if (p == head && tryAcquire(arg)) {setHead(node);p.next = null; // help GCfailed = false;return;}if (shouldParkAfterFailedAcquire(p, node) &&parkAndCheckInterrupt())//線程中斷拋異常throw new InterruptedException();}} finally {if (failed)cancelAcquire(node);} }關鍵信息請看注釋,現在看這段代碼就很輕松了吧,與acquire方法邏輯幾乎一致,唯一的區別是當parkAndCheckInterrupt返回true時即線程阻塞時該線程被中斷,代碼拋出被中斷異常。
通過調用lock.tryLock(timeout,TimeUnit)方式達到超時等待獲取鎖的效果,該方法會在三種情況下才會返回:
1.在超時時間內,當前線程成功獲取了鎖;
2.當前線程在超時時間內被中斷;
3.超時時間結束,仍未獲得鎖返回false。
我們仍然通過采取閱讀源碼的方式來學習底層具體是怎么實現的,該方法會調用AQS的方法tryAcquireNanos(),源碼為:
public final boolean tryAcquireNanos(int arg, long nanosTimeout)throws InterruptedException {if (Thread.interrupted())throw new InterruptedException();return tryAcquire(arg) ||//實現超時等待的效果doAcquireNanos(arg, nanosTimeout); }很顯然這段源碼最終是靠doAcquireNanos方法實現超時等待的效果,該方法源碼如下:
private boolean doAcquireNanos(int arg, long nanosTimeout)throws InterruptedException {if (nanosTimeout <= 0L)return false;//1. 根據超時時間和當前時間計算出截止時間final long deadline = System.nanoTime() + nanosTimeout;final Node node = addWaiter(Node.EXCLUSIVE);boolean failed = true;try {for (;;) {final Node p = node.predecessor();//2. 當前線程獲得鎖出隊列if (p == head && tryAcquire(arg)) {setHead(node);p.next = null; // help GCfailed = false;return true;}// 3.1 重新計算超時時間nanosTimeout = deadline - System.nanoTime();// 3.2 已經超時返回falseif (nanosTimeout <= 0L)return false;// 3.3 線程阻塞等待 if (shouldParkAfterFailedAcquire(p, node) &&nanosTimeout > spinForTimeoutThreshold)LockSupport.parkNanos(this, nanosTimeout);// 3.4 線程被中斷拋出被中斷異常if (Thread.interrupted())throw new InterruptedException();}} finally {if (failed)cancelAcquire(node);} }程序邏輯如圖所示:
程序邏輯同獨占鎖可響應中斷式獲取基本一致,唯一的不同在于獲取鎖失敗后,對超時時間的處理上,在第1步會先計算出按照現在時間和超時時間計算出理論上的截止時間,比如當前時間是8h10min,超時時間是10min,那么根據deadline = System.nanoTime() + nanosTimeout計算出剛好達到超時時間時的系統時間就是8h 10min+10min = 8h 20min。然后根據deadline - System.nanoTime()就可以判斷是否已經超時了,比如,當前系統時間是8h 30min很明顯已經超過了理論上的系統時間8h 20min,deadline - System.nanoTime()計算出來就是一個負數,自然而然會在3.2步中的If判斷之間返回false。如果還沒有超時即3.2步中的if判斷為true時就會繼續執行3.3步通過LockSupport.parkNanos使得當前線程阻塞,同時在3.4步增加了對中斷的檢測,若檢測出被中斷直接拋出被中斷異常。
五、共享鎖
在聊完AQS對獨占鎖的實現后,我們繼續一鼓作氣的來看看共享鎖是怎樣實現的?共享鎖的獲取方法為acquireShared,源碼為:
public final void acquireShared(int arg) {if (tryAcquireShared(arg) < 0)doAcquireShared(arg); }這段源碼的邏輯很容易理解,在該方法中會首先調用tryAcquireShared方法,tryAcquireShared返回值是一個int類型,當返回值為大于等于0的時候方法結束說明獲得成功獲取鎖,否則,表明獲取同步狀態失敗即所引用的線程獲取鎖失敗,會執行doAcquireShared方法,該方法的源碼為:
private void doAcquireShared(int arg) {final Node node = addWaiter(Node.SHARED);boolean failed = true;try {boolean interrupted = false;for (;;) {final Node p = node.predecessor();if (p == head) {int r = tryAcquireShared(arg);if (r >= 0) {// 當該節點的前驅節點是頭結點且成功獲取同步狀態setHeadAndPropagate(node, r);p.next = null; // help GCif (interrupted)selfInterrupt();failed = false;return;}}if (shouldParkAfterFailedAcquire(p, node) &&parkAndCheckInterrupt())interrupted = true;}} finally {if (failed)cancelAcquire(node);} }現在來看這段代碼會不會很容易了?邏輯幾乎和獨占式鎖的獲取一模一樣,這里的自旋過程中能夠退出的條件是當前節點的前驅節點是頭結點并且tryAcquireShared(arg)返回值大于等于0即能成功獲得同步狀態。
共享鎖的釋放在AQS中會調用方法
releaseShared: public final boolean releaseShared(int arg) {if (tryReleaseShared(arg)) {doReleaseShared();return true;}return false; }當成功釋放同步狀態之后即tryReleaseShared會繼續執行doReleaseShared方法:
private void doReleaseShared() {for (;;) {Node h = head;if (h != null && h != tail) {int ws = h.waitStatus;if (ws == Node.SIGNAL) {if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))continue; // loop to recheck casesunparkSuccessor(h);}else if (ws == 0 &&!compareAndSetWaitStatus(h, 0, Node.PROPAGATE))continue; // loop on failed CAS}if (h == head) // loop if head changedbreak;} }這段方法跟獨占式鎖釋放過程有點點不同,在共享式鎖的釋放過程中,對于能夠支持多個線程同時訪問的并發組件,必須保證多個線程能夠安全的釋放同步狀態,這里采用的CAS保證,當CAS操作失敗continue,在下一次循環中進行重試。
關于可中斷鎖以及超時等待的特性其實現和獨占式鎖可中斷獲取鎖以及超時等待的實現幾乎一致,具體的就不再說了,如果理解了上面的內容對這部分的理解也是水到渠成的。
六、線程獲取獨占資源分析(以ReentrantLock為例)
tryAcquire(arg)判斷資源是否被占用,在tryAcquire()中主要通過獲取state的值判斷資源是否被占用,getState()獲取state的值,如果為0,則證明資源沒被 占用,就使用CAS操作更新值,然后state的值為1,然后使用setExclusiveOwnerThread()設置獨占資源的線程(獨占資源和每一個線程進行綁定),然后返回,下面有一個else if判斷,如果當前線程等于獨占資源的線程,證明線程重入了,那么更新state的值為2(state原本為1,重入后+1),最后返回。
protected final boolean tryAcquire(int acquires) {//獲取當前線程final Thread current = Thread.currentThread();//獲取stateint c = getState();//資源沒被獨占if (c == 0) {//使用CAS對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");//state+1setState(nextc);return true;}return false; }如果資源已經被獨占了,那么就需要將當前線程插入AQS隊列中,然后將當前線程掛起,我們看一下addWaiter(Node.EXCLUSIVE),可以看出將Node節點插入了AQS的隊尾。
private Node addWaiter(Node mode) {Node node = new Node(Thread.currentThread(), mode);// Try the fast path of enq; backup to full enq on failureNode pred = tail;if (pred != null) {node.prev = pred;if (compareAndSetTail(pred, node)) {pred.next = node;return node;}}enq(node);return node;}acquireQueued就是將當前線程掛起,我們看里面的一個方法parkAndCheckInterrupt(), 可以看出使用 LockSupport.park(this)來將當前線程掛起。
private final boolean parkAndCheckInterrupt() {LockSupport.park(this);return Thread.interrupted(); }當一個線程調用了release(arg)方法釋放資源時,它會調用tryRelease(arg)嘗試釋放資源。
public final boolean release(int arg) {if (tryRelease(arg)) {Node h = head;if (h != null && h.waitStatus != 0)unparkSuccessor(h);return true;}return false; }嘗試釋放資源的邏輯很簡單,就是用AQS中的state-1,如果為0,則能證明可以釋放,就會通過setState(arg)方法設置state的值,注意,bash 如果c=0,則證明此線程不是重入的線程,那么就直接釋放資源,并且解除與資源綁定的線程(setExclusiveOwnerThread(null)),如果c != 0 ,則證明資源被重入, 則將state - 1 ,不解除與獨占資源綁定的線程。
protected final boolean tryRelease(int releases) {int c = getState() - releases;if (Thread.currentThread() != getExclusiveOwnerThread())throw new IllegalMonitorStateException();boolean free = false;if (c == 0) {free = true;setExclusiveOwnerThread(null);}setState(c);return free釋放完資源后,就會喚醒使用LockSupport.unpark(s.thread)喚醒AQS中的線程,然后又重復上面的過程。
private void unparkSuccessor(Node node) {int ws = node.waitStatus;if (ws < 0)compareAndSetWaitStatus(node, ws, 0);Node s = node.next;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);}七、共享操作(以CountDownLatch為例)
當線程進入countDownLatch.await()方式時,會調用AQS的acquireSharedInterruptibly方法,然后進入tryAcquireShared(arg)方法。
public final void acquireSharedInterruptibly(int arg) throws InterruptedException {if (Thread.interrupted())throw new InterruptedException();if (tryAcquireShared(arg) < 0)doAcquireSharedInterruptibly(arg); }tryAcquireShared(arg)的作用就是判斷AQS狀態值state,如果state = 0 , 返回1,否則返回-1,如果返回1,則證明線程對資源的操作已經結束,如果為-1,則證明對資源的操作
尚未結束,此時就會調用doAcquireSharedInterruptibly(arg)方法。
調用doAcquireSharedInterruptibly(arg)會通過addWaiter(Node.SHARED)將當前線程封裝成SHARED類型的Node,然后插入AQS隊列的尾部,然后調用
parkAndCheckInterrupt()使用LockSupport.park(this)將當前線程掛起。
當使用countDownLatch.countDown()時,其實就是對資源的釋放,最終會調用releaseShared(int arg)方法,然后調用tryReleaseShared(int releases)嘗試釋放資源
public final boolean releaseShared(int arg) {if (tryReleaseShared(arg)) {doReleaseShared();return true;}return false; }調用tryReleaseShared(int releases)嘗試釋放資源,釋放資源的邏輯時根據state來判斷,如果state = 0,證明當前資源沒有任何線程持有,所以就直接返回,如果state != 0 , 證明資源還被線程持有,所以當前線程就將state - 1,然后繼續向下執行doReleaseShared()。
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;}} }doReleaseShared()會調用unparkSuccessor(h)來喚醒線程(LockSupport.unpark(s.thread)),因為傳遞過去的參數h = head , 所以可知喚醒的線程是隊頭節點的線程。
private void doReleaseShared() {for (;;) {Node h = head;if (h != null && h != tail) {int ws = h.waitStatus;if (ws == Node.SIGNAL) {if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))continue; // loop to recheck casesunparkSuccessor(h);}else if (ws == 0 && !compareAndSetWaitStatus(h, 0, Node.PROPAGATE))continue; // loop on failed CAS}if (h == head) // loop if head changedbreak;} }八、關于公平鎖非公平鎖的疑問
公平鎖與非公平鎖的含義都很明白,公平鎖必須排隊獲取鎖,鎖的獲取順序完全根據排隊順序而來,而非公平就是誰搶到是誰的。我們都知道AQS中的鎖獲取,如果首次獲取失敗會進入到內部的同步隊列中阻塞等待,只有前面的節點喚醒當前節點才能去嘗試獲取鎖。而鎖獲取的核心,實際上是在tryAcquire方法中定義的,在節點從阻塞中醒來時,都會用tryAcquire方法來獲取鎖,而公平與否在tryAcquire方法中體現,即在公平鎖中,tryAcquire獲取的前提是需要沒有存在前驅節點的,而非公平鎖就沒有這個限制。
可以看到,對沒公平和公平兩種方式都有實現,直接看源碼:
總結
以上是生活随笔為你收集整理的深入理解AbstractQueuedSynchronizer(AQS)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 初识Lock与AbstractQueue
- 下一篇: 还没做2022年计划?这个超赞工具送给你