阿里P7架構師帶你深刻分析AQS實現原理

內容導航

• ReentrantLock重入鎖的使用引入AQS
• 什麼是AQS
• AQS的源碼分析

簡單解釋一下J.U.C，是JDK中提供的併發工具包, java.util.concurrent。裏面提供了不少併發編程中很經常使用的實用工具類，好比atomic原子操做、好比lock同步鎖、fork/join等。

從Lock做爲切入點

我想以lock做爲切入點來說解AQS，畢竟同步鎖是解決線程安全問題的通用手段，也是咱們工做中用得比較多的方式。

Lock API

Lock是一個接口，方法定義以下

Lock的實現

實現Lock接口的類有不少，如下爲幾個常見的鎖實現

• ReentrantLock：表示重入鎖，它是惟一一個實現了Lock接口的類。重入鎖指的是線程在得到鎖以後，再次獲取該鎖不須要阻塞，而是直接關聯一次計數器增長重入次數
• ReentrantReadWriteLock：重入讀寫鎖，它實現了ReadWriteLock接口，在這個類中維護了兩個鎖，一個是ReadLock，一個是WriteLock，他們都分別實現了Lock接口。讀寫鎖是一種適合讀多寫少的場景下解決線程安全問題的工具，基本原則是： 讀和讀不互斥、讀和寫互斥、寫和寫互斥。也就是說涉及到影響數據變化的操做都會存在互斥。
• StampedLock： stampedLock是JDK8引入的新的鎖機制，能夠簡單認爲是讀寫鎖的一個改進版本，讀寫鎖雖然經過分離讀和寫的功能使得讀和讀之間能夠徹底併發，可是讀和寫是有衝突的，若是大量的讀線程存在，可能會引發寫線程的飢餓。stampedLock是一種樂觀的讀策略，使得樂觀鎖徹底不會阻塞寫線程

ReentrantLock的簡單實用

如何在實際應用中使用ReentrantLock呢？咱們經過一個簡單的demo來演示一下

這段代碼主要作一件事，就是經過一個靜態的 incr()方法對共享變量 count作連續遞增，在沒有加同步鎖的狀況下多線程訪問這個方法必定會存在線程安全問題。因此用到了 ReentrantLock來實現同步鎖，而且在finally語句塊中釋放鎖。那麼我來引出一個問題，你們思考一下

多個線程經過lock競爭鎖時，當競爭失敗的鎖是如何實現等待以及被喚醒的呢?

什麼是AQS

aqs全稱爲AbstractQueuedSynchronizer，它提供了一個FIFO隊列，能夠當作是一個用來實現同步鎖以及其餘涉及到同步功能的核心組件，常見的有:ReentrantLock、CountDownLatch等。

AQS是一個抽象類，主要是經過繼承的方式來使用，它自己沒有實現任何的同步接口，僅僅是定義了同步狀態的獲取以及釋放的方法來提供自定義的同步組件。

能夠這麼說，只要搞懂了AQS，那麼J.U.C中絕大部分的api都能輕鬆掌握。

AQS的兩種功能

從使用層面來講，AQS的功能分爲兩種：獨佔和共享

• 獨佔鎖，每次只能有一個線程持有鎖，好比前面給你們演示的ReentrantLock就是以獨佔方式實現的互斥鎖
• 共享鎖，容許多個線程同時獲取鎖，併發訪問共享資源，好比ReentrantReadWriteLock

ReentrantLock的類圖

仍然以ReentrantLock爲例，來分析AQS在重入鎖中的使用。畢竟單純分析AQS沒有太多的含義。先理解這個類圖，能夠方便咱們理解AQS的原理

AQS的內部實現

AQS的實現依賴內部的同步隊列,也就是FIFO的雙向隊列，若是當前線程競爭鎖失敗，那麼AQS會把當前線程以及等待狀態信息構形成一個Node加入到同步隊列中，同時再阻塞該線程。當獲取鎖的線程釋放鎖之後，會從隊列中喚醒一個阻塞的節點(線程)。

AQS隊列內部維護的是一個FIFO的雙向鏈表，這種結構的特色是每一個數據結構都有兩個指針，分別指向直接的後繼節點和直接前驅節點。因此雙向鏈表能夠從任意一個節點開始很方便的訪問前驅和後繼。每一個Node實際上是由線程封裝，當線程爭搶鎖失敗後會封裝成Node加入到ASQ隊列中去

Node類的組成以下

釋放鎖以及添加線程對於隊列的變化

添加節點

當出現鎖競爭以及釋放鎖的時候，AQS同步隊列中的節點會發生變化，首先看一下添加節點的場景。

這裏會涉及到兩個變化

• 新的線程封裝成Node節點追加到同步隊列中，設置prev節點以及修改當前節點的前置節點的next節點指向本身
• 經過CAS講tail從新指向新的尾部節點

釋放鎖移除節點

head節點表示獲取鎖成功的節點，當頭結點在釋放同步狀態時，會喚醒後繼節點，若是後繼節點得到鎖成功，會把本身設置爲頭結點，節點的變化過程以下

這個過程也是涉及到兩個變化

• 修改head節點指向下一個得到鎖的節點
• 新的得到鎖的節點，將prev的指針指向null

這裏有一個小的變化，就是設置head節點不須要用CAS，緣由是設置head節點是由得到鎖的線程來完成的，而同步鎖只能由一個線程得到，因此不須要CAS保證，只須要把head節點設置爲原首節點的後繼節點，而且斷開原head節點的next引用便可

AQS的源碼分析

清楚了AQS的基本架構之後，咱們來分析一下AQS的源碼，仍然以ReentrantLock爲模型。

ReentrantLock的時序圖

調用ReentrantLock中的lock()方法，源碼的調用過程我使用了時序圖來展示

從圖上能夠看出來，當鎖獲取失敗時，會調用addWaiter()方法將當前線程封裝成Node節點加入到AQS隊列，基於這個思路，咱們來分析AQS的源碼實現

分析源碼

ReentrantLock.lock()

這個是獲取鎖的入口，調用sync這個類裏面的方法，sync是什麼呢？

sync是一個靜態內部類，它繼承了AQS這個抽象類，前面說過AQS是一個同步工具，主要用來實現同步控制。咱們在利用這個工具的時候，會繼承它來實現同步控制功能。

經過進一步分析，發現Sync這個類有兩個具體的實現，分別是 NofairSync(非公平鎖), FailSync(公平鎖).

• 公平鎖 表示全部線程嚴格按照FIFO來獲取鎖
• 非公平鎖 表示能夠存在搶佔鎖的功能，也就是說無論當前隊列上是否存在其餘線程等待，新線程都有機會搶佔鎖

公平鎖和非公平鎖的實現上的差別，我會在文章後面作一個解釋，接下來的分析仍然以 非公平鎖做爲主要分析邏輯。

NonfairSync.lock

這段代碼簡單解釋一下

• 因爲這裏是非公平鎖，因此調用lock方法時，先去經過cas去搶佔鎖
• 若是搶佔鎖成功，保存得到鎖成功的當前線程
• 搶佔鎖失敗，調用acquire來走鎖競爭邏輯

compareAndSetStatecompareAndSetState的代碼實現邏輯以下

這段代碼其實邏輯很簡單，就是經過cas樂觀鎖的方式來作比較並替換。上面這段代碼的意思是，若是當前內存中的state的值和預期值expect相等，則替換爲update。更新成功返回true，不然返回false.這個操做是原子的，不會出現線程安全問題，這裏面涉及到Unsafe這個類的操做，一級涉及到state這個屬性的意義。stateAQS中有一個這樣的屬性定義,這個對於重入鎖的實現來講，表示一個同步狀態。它有兩個含義的表示

• 當state=0時，表示無鎖狀態
• 當state>0時，表示已經有線程得到了鎖，也就是state=1，可是由於ReentrantLock容許重入，因此同一個線程屢次得到同步鎖的時候，state會遞增，好比重入5次，那麼state=5。 而在釋放鎖的時候，一樣須要釋放5次直到state=0其餘線程纔有資格得到鎖

private volatile int state;

須要注意的是：不一樣的AQS實現，state所表達的含義是不同的。UnsafeUnsafe類是在sun.misc包下，不屬於Java標準。可是不少Java的基礎類庫，包括一些被普遍使用的高性能開發庫都是基於Unsafe類開發的，好比Netty、Hadoop、Kafka等；Unsafe可認爲是Java中留下的後門，提供了一些低層次操做，如直接內存訪問、線程調度等

public final native boolean compareAndSwapInt(Object var1, long var2, int var4, int var5);

這個是一個native方法， 第一個參數爲須要改變的對象，第二個爲偏移量(即以前求出來的headOffset的值)，第三個參數爲期待的值，第四個爲更新後的值整個方法的做用是若是當前時刻的值等於預期值var4相等，則更新爲新的指望值 var5，若是更新成功，則返回true，不然返回false；

acquire

acquire是AQS中的方法，若是CAS操做未能成功，說明state已經不爲0，此時繼續acquire(1)操做,這裏你們思考一下，acquire方法中的1的參數是用來作什麼呢？若是沒猜中，往前面回顧一下state這個概念

這個方法的主要邏輯是

• 經過tryAcquire嘗試獲取獨佔鎖，若是成功返回true，失敗返回false
• 若是tryAcquire失敗，則會經過addWaiter方法將當前線程封裝成Node添加到AQS隊列尾部
• acquireQueued，將Node做爲參數，經過自旋去嘗試獲取鎖。

NonfairSync.tryAcquire

這個方法的做用是嘗試獲取鎖，若是成功返回true，不成功返回false

它是重寫AQS類中的tryAcquire方法，而且你們仔細看一下AQS中tryAcquire方法的定義，並無實現，而是拋出異常。按照通常的思惟模式，既然是一個不實現的模版方法，那應該定義成abstract，讓子類來實現呀？你們想一想爲何

nonfairTryAcquire

tryAcquire(1)在NonfairSync中的實現代碼以下

• 獲取當前線程，判斷當前的鎖的狀態
• 若是state=0表示當前是無鎖狀態，經過cas更新state狀態的值
• 若是當前線程是屬於重入，則增長重入次數

addWaiter

當tryAcquire方法獲取鎖失敗之後，則會先調用addWaiter將當前線程封裝成Node，而後添加到AQS隊列

• 將當前線程封裝成Node
• 判斷當前鏈表中的tail節點是否爲空，若是不爲空，則經過cas操做把當前線程的node添加到AQS隊列
• 若是爲空或者cas失敗，調用enq將節點添加到AQS隊列

enq

enq就是經過自旋操做把當前節點加入到隊列中

假若有兩個線程t1,t2同時進入enq方法，t==null表示隊列是首次使用，須要先初始化

另一個線程cas失敗，則進入下次循環，經過cas操做將node添加到隊尾

到目前爲止，經過addwaiter方法構造了一個AQS隊列，而且將線程添加到了隊列的節點中

acquireQueued

將添加到隊列中的Node做爲參數傳入acquireQueued方法，這裏面會作搶佔鎖的操做

• 獲取當前節點的prev節點
• 若是prev節點爲head節點，那麼它就有資格去爭搶鎖，調用tryAcquire搶佔鎖
• 搶佔鎖成功之後，把得到鎖的節點設置爲head，而且移除原來的初始化head節點
• 若是得到鎖失敗，則根據waitStatus決定是否須要掛起線程
• 最後，經過cancelAcquire取消得到鎖的操做

前面的邏輯都很好理解，主要看一下shouldParkAfterFailedAcquire這個方法和parkAndCheckInterrupt的做用

shouldParkAfterFailedAcquire

從上面的分析能夠看出，只有隊列的第二個節點能夠有機會爭用鎖，若是成功獲取鎖，則此節點晉升爲頭節點。對於第三個及之後的節點，if (p == head)條件不成立，首先進行shouldParkAfterFailedAcquire(p, node)操做

shouldParkAfterFailedAcquire方法是判斷一個爭用鎖的線程是否應該被阻塞。它首先判斷一個節點的前置節點的狀態是否爲Node.SIGNAL，若是是，是說明此節點已經將狀態設置-若是鎖釋放，則應當通知它，因此它能夠安全的阻塞了，返回true。

parkAndCheckInterrupt

若是shouldParkAfterFailedAcquire返回了true，則會執行： parkAndCheckInterrupt()方法，它是經過LockSupport.park(this)將當前線程掛起到WATING狀態，它須要等待一箇中斷、unpark方法來喚醒它，經過這樣一種FIFO的機制的等待，來實現了Lock的操做。

LockSupportLockSupport類是Java6引入的一個類，提供了基本的線程同步原語。LockSupport其實是調用了Unsafe類裏的函數，歸結到Unsafe裏，只有兩個函數：

unpark函數爲線程提供「許可(permit)」，線程調用park函數則等待「許可」。這個有點像信號量，可是這個「許可」是不能疊加的，「許可」是一次性的。permit至關於0/1的開關，默認是0，調用一次unpark就加1變成了1.調用一次park會消費permit，又會變成0。 若是再調用一次park會阻塞，由於permit已是0了。直到permit變成1.這時調用unpark會把permit設置爲1.每一個線程都有一個相關的permit，permit最多隻有一個，重複調用unpark不會累積

鎖的釋放

ReentrantLock.unlock

加鎖的過程分析完之後，再來分析一下釋放鎖的過程，調用release方法，這個方法裏面作兩件事，1，釋放鎖 ；2，喚醒park的線程

tryRelease

這個動做能夠認爲就是一個設置鎖狀態的操做，並且是將狀態減掉傳入的參數值（參數是1），若是結果狀態爲0，就將排它鎖的Owner設置爲null，以使得其它的線程有機會進行執行。

在排它鎖中，加鎖的時候狀態會增長1（固然能夠本身修改這個值），在解鎖的時候減掉1，同一個鎖，在能夠重入後，可能會被疊加爲二、三、4這些值，只有unlock()的次數與lock()的次數對應纔會將Owner線程設置爲空，並且也只有這種狀況下才會返回true。

unparkSuccessor

在方法unparkSuccessor(Node)中，就意味着真正要釋放鎖了，它傳入的是head節點（head節點是佔用鎖的節點），當前線程被釋放以後，須要喚醒下一個節點的線程

總結

經過這篇文章基本將AQS隊列的實現過程作了比較清晰的分析，主要是基於非公平鎖的獨佔鎖實現。在得到同步鎖時，同步器維護一個同步隊列，獲取狀態失敗的線程都會被加入到隊列中並在隊列中進行自旋；移出隊列（或中止自旋）的條件是前驅節點爲頭節點且成功獲取了同步狀態。在釋放同步狀態時，同步器調用tryRelease(int arg)方法釋放同步狀態，而後喚醒頭節點的後繼節點。

若是對java微服務、分佈式、高併發、高可用、大型互聯網架構技術、面試經驗交流。 能夠加我架構圈子羣:142019080 領取資料，羣內天天更新資料，免費領取。