新聞中心
大家好,我是君哥。

為烏拉特后等地區(qū)用戶提供了全套網(wǎng)頁設(shè)計制作服務(wù),及烏拉特后網(wǎng)站建設(shè)行業(yè)解決方案。主營業(yè)務(wù)為成都網(wǎng)站建設(shè)、網(wǎng)站設(shè)計、烏拉特后網(wǎng)站設(shè)計,以傳統(tǒng)方式定制建設(shè)網(wǎng)站,并提供域名空間備案等一條龍服務(wù),秉承以專業(yè)、用心的態(tài)度為用戶提供真誠的服務(wù)。我們深信只要達到每一位用戶的要求,就會得到認可,從而選擇與我們長期合作。這樣,我們也可以走得更遠!
Java中 AQS 是 AbstractQueuedSynchronizer 類,AQS 依賴 FIFO 隊列來提供一個框架,這個框架用于實現(xiàn)鎖以及鎖相關(guān)的同步器,比如信號量、事件等。
在 AQS 中,主要有兩部分功能,一部分是操作 state 變量,第二部分是實現(xiàn)排隊和阻塞機制。
注意,AQS 并沒有實現(xiàn)任何同步接口,它只是提供了類似 acquireInterruptible 的方法,調(diào)用這些方法可以實現(xiàn)鎖和同步器。
1、管程模型
Java 使用 MESA 管程模型來管理類的成員變量和方法,讓這個類的成員變量和方法的操作是線程安全的。下圖是 MESA 管程模型,里面除了定義共享變量外,還定義了條件變量和條件變量等待隊列:
上圖中有三個知識點:
- MESA 管程模型封裝了共享變量和對共享變量的操作,線程要進入管程內(nèi)部,必須獲取到鎖,如果獲取鎖失敗就進入入口等待隊列阻塞等待。
- 如果線程獲取到鎖,就進入到管程內(nèi)部。但是進入到管程內(nèi)部,也不一定能立刻操作共享變量,而是要看條件變量是否滿足,如果不滿足,只能進入條件變量等待隊列阻塞等待。
- 在條件變量等待隊列中,如果被其他線程喚醒,也不一定能立刻操作共享變量,而是需要去入口等待隊列重新排隊等待獲取鎖。
Java 中的 MESA 管程模型有一點改進,就是管程內(nèi)部只有一個條件變量和一個等待隊列。下圖是 AQS 的管程模型:
AQS 的管程模型依賴 AQS 中的 FIFO 隊列實現(xiàn)入口等待隊列,要進入管程內(nèi)部,就由各種并發(fā)鎖的限制。而 ConditionObject 則實現(xiàn)了條件隊列,這個隊列可以創(chuàng)建多個。
下面就從入口等待隊列、并發(fā)鎖、條件等待隊列三個方面來帶你徹底理解 AQS。
2、入口等待隊列
2.1 獲取獨占鎖
獨占, 忽略 interrupts
public final void acquire(int arg){
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
selfInterrupt();
}這里的 tryAcquire 是抽象方法,由 AQS 的子類來實現(xiàn),因為每個子類實現(xiàn)的鎖是不一樣的。
2.1.1 入隊
上面的代碼可以看到,獲取鎖失敗后,會先執(zhí)行 addWaiter 方法加入隊列,然后執(zhí)行 acquireQueued 方法自旋地獲取鎖直到成功。
addWaiter 代碼邏輯如下圖,簡單說就是把 node 入隊,入隊后返回 node 參數(shù)給 acquireQueued 方法:
這里有一個點需要注意,如果隊列為空,則新建一個 Node 作為隊頭。
2.1.2 入隊后獲取鎖
acquireQueued 自旋獲取鎖邏輯如下圖:
這里有幾個細節(jié):
(1)waitStatus
- CANCELLED(1):當前節(jié)點取消獲取鎖。當?shù)却瑫r或被中斷(響應(yīng)中斷),會觸發(fā)變更為此狀態(tài),進入該狀態(tài)后節(jié)點狀態(tài)不再變化;
- SIGNAL(-1):后面節(jié)點等待當前節(jié)點喚醒;
- CONDITION(-2):Condition 中使用,當前線程阻塞在 Condition,如果其他線程調(diào)用了 Condition 的 signal 方法,這個結(jié)點將從等待隊列轉(zhuǎn)移到同步隊列隊尾,等待獲取同步鎖;
- PROPAGATE(-3):共享模式,前置節(jié)點喚醒后面節(jié)點后,喚醒操作無條件傳播下去;
- 0:中間狀態(tài),當前節(jié)點后面的節(jié)點已經(jīng)喚醒,但是當前節(jié)點線程還沒有執(zhí)行完成。
(2)獲取鎖失敗后掛起
如果前置節(jié)點不是頭節(jié)點,或者前置節(jié)點是頭節(jié)點但當前節(jié)點獲取鎖失敗,這時當前節(jié)點需要掛起,分三種情況:
- 前置節(jié)點 waitStatus=-1,如下圖:
- 前置節(jié)點 waitStatus > 0,如下圖:
- 前置節(jié)點 waitStatus < 0 但不等于 -1,如下圖:
(3)取消獲取鎖
如果獲取鎖拋出異常,則取消獲取鎖,如果當前節(jié)點是 tail 節(jié)點,分兩種情況如下圖:
如果當前節(jié)點不是 tail 節(jié)點,也分兩種情況,如下圖:
4.對中斷狀態(tài)忽略
5.如果前置節(jié)點的狀態(tài)是 0 或 PROPAGATE,會被當前節(jié)點自旋過程中更新成 -1,以便之后通知當前節(jié)點。
2.1.3 獨占 + 響應(yīng)中斷
對應(yīng)方法 acquireInterruptibly(int arg)。
跟忽略中斷(acquire方法)不同的是要響應(yīng)中斷,下面兩個地方響應(yīng)中斷:
- 獲取鎖之前會檢查當前線程是否中斷。
- 獲取鎖失敗入隊,在隊列中自旋獲取鎖的過程中也會檢查當前線程是否中斷。如果檢查到當前線程已經(jīng)中斷,則拋出 InterruptedException,當前線程退出。
2.1.4 獨占 + 響應(yīng)中斷 + 考慮超時
對應(yīng)方法 tryAcquireNanos(int arg, long nanosTimeout)。
這個方法具備了獨占 + 響應(yīng)中斷 + 超時的功能,下面2個地方要判斷是否超時:
- 自旋獲取鎖的過程中每次獲取鎖失敗都要判斷是否超時;
- 獲取鎖失敗 park 之前要判斷超時時間是否大于自旋的閾值時間 (spinForTimeoutThreshold = 1ns) 另外,park 線程的操作使用 parkNanos 傳入阻塞時間。
2.2 釋放獨占鎖
獨占鎖釋放分兩步:釋放鎖,喚醒后繼節(jié)點。
釋放鎖的方法 tryRelease 是抽象的,由子類去實現(xiàn)。
我們看一下喚醒后繼節(jié)點的邏輯,首先需要滿足兩個條件:
- head 節(jié)點不等于 null;
- head 節(jié)點 waitStatus 不等于 0。這里有兩種情況(在方法 unparkSuccessor):
- 情況一,后繼節(jié)點 waitStatus <= 0,直接喚醒后繼節(jié)點,如下圖:
- 情況二:后繼節(jié)點為空或者 waitStatus > 0,從后往前查找最接近當前節(jié)點的節(jié)點進行喚醒,如下圖:
2.3 獲取共享鎖
之前我們講了獨占鎖,這一小節(jié)我們談共享鎖,有什么不同呢?
2.3.1 共享,忽略 interrupts
對應(yīng)方法 acquireShared,代碼如下:
public final void acquireShared(int arg){
if (tryAcquireShared(arg) < 0)
doAcquireShared(arg);
}2.3.2 tryAcquireShared
這里獲取鎖使用的方法是 tryAcquireShared,獲取的是共享鎖。獲取共享鎖跟獲取獨占鎖不同的是,會返回一個整數(shù)值,說明如下:
- 返回負數(shù):獲取鎖失敗。
- 返回 0:獲取鎖成功但是之后再由線程來獲取共享鎖時就會失敗。
- 返回正數(shù):獲取鎖成功而且之后再有線程來獲取共享鎖時也可能會成功。所以需要把喚醒操作傳播下去。tryAcquireShared 獲取鎖失敗后(返回負數(shù)),就需要入隊后自旋獲取,也就是執(zhí)行方法 doAcquireShared。
2.3.3 doAcquireShared
怎么判斷隊列中等待節(jié)點是在等待共享鎖呢?nextWaiter == SHARED,這個參數(shù)值是入隊新建節(jié)點的時候構(gòu)造函數(shù)傳入的。
自旋過程中,如果獲取鎖成功(返回正數(shù)),首先把自己設(shè)置成新的 head 節(jié)點,然后把通知傳播下去。如下圖:
之后會喚醒后面節(jié)點并保證喚醒操作可以傳播下去。但是需要滿足四個條件中的一個:
- tryAcquireShared 返回值大于0,有多余的鎖,可以繼續(xù)喚醒后繼節(jié)點。
- 舊的 head 節(jié)點 waitStatus < 0,應(yīng)該是其他線程釋放共享鎖過程中把它的狀態(tài)更新成了 -3。
- 新的 hade 節(jié)點 waitStatus < 0,只要不是 tail 節(jié)點,就可能是 -1。這里會造成不必要的喚醒,因為喚醒后獲取不到鎖只能繼續(xù)入隊等待。
- 當前節(jié)點的后繼節(jié)點是空或者非空但正在等待共享鎖。
喚醒后面節(jié)點的操作,其實就是釋放共享鎖,對應(yīng)方法是 doReleaseShared,見釋放共享鎖一節(jié)。
2.3.4 共享 + 響應(yīng)中斷
對應(yīng)方法 acquireSharedInterruptibly(int arg)。
跟共享忽略中斷(acquireShared 方法)不同的是要響應(yīng)中斷,下面兩個地方響應(yīng)中斷:
- 獲取鎖之前會檢查當前線程是否中斷。
- 獲取鎖失敗入隊,在隊列中自旋獲取鎖的過程中也會檢查當前線程是否中斷。
如果檢查到當前線程已經(jīng)中斷,則拋出 InterruptedException,當前線程退出。
2.3.5 共享 + 響應(yīng)中斷 + 考慮超時
對應(yīng)方法 tryAcquireSharedNanos(int arg, long nanosTimeout)。
這個方法具備了共享 + 響應(yīng)中斷 + 超時的功能,下面兩個個地方要判斷是否超時:
- 自旋獲取鎖的過程中每次獲取鎖失敗都要判斷是否超時。
- 獲取鎖失敗 park 之前要判斷超時時間是否大于自旋的閾值時間(spinForTimeoutThreshold = 1ns)。
另外,park 線程的操作使用 parkNanos 傳入阻塞時間。
2.4 釋放共享鎖
釋放共享鎖代碼如下:
public final boolean releaseShared(int arg){
if (tryReleaseShared(arg)) {
doReleaseShared();
return true;
}
return false;
}首先嘗試釋放共享鎖,tryReleaseShared 代碼由子類來實現(xiàn)。釋放成功后執(zhí)行AQS中的 doReleaseShared 方法,是一個自旋操作。
自旋的條件是隊列中至少有兩個節(jié)點,這里分三種情況。
情況一:當前節(jié)點 waitStatus 是 -1,如下圖:
情況二:當前節(jié)點 waitStatus 是 0(被其他線程更新新成了中間狀態(tài)),如下圖:
情況三:當前節(jié)點 waitStatus 是 -3,為什么會這樣呢?需要解釋一下,head節(jié)點喚醒后繼節(jié)點之前 waitStatus 已經(jīng)被更新中間態(tài) 0 了,喚醒后繼節(jié)點動作還沒有執(zhí)行,又被其他線程更成了 -3,也就是其他線程釋放鎖執(zhí)行了上面情況二。這時需要先把 waitStatus 再更成 0 (在方法 unparkSuccessor),如下圖:
2.5 抽象方法
上面的講解可以看出,如果要基于 AQS 來實現(xiàn)并發(fā)鎖,可以根據(jù)需求重寫下面四個方法來實現(xiàn),這四個方法在 AQS 中沒有具體實現(xiàn):
- tryAcquire(int arg):獲取獨占鎖
- tryRelease(int arg):釋放獨占鎖
- tryAcquireShared(int arg):獲取共享鎖
- tryReleaseShared(int arg):釋放共享鎖
AQS 的子類需要重寫上面的方法來修改 state 值,并且定義獲取鎖或者釋放鎖時 state 值的變化。子類也可以定義自己的 state 變量,但是只有更新 AQS 中的 state變量才會對同步起作用。
還有一個判斷當前線程是否持有獨占鎖的方法 isHeldExclusively,也可以供子類重寫后使用。
獲取/釋放鎖的具體實現(xiàn)放到下篇文章講解。
2.6 總結(jié)
AQS 使用 FIFO 隊列實現(xiàn)了一個鎖相關(guān)的并發(fā)器模板,可以基于這個模板來實現(xiàn)各種鎖,包括獨占鎖、共享鎖、信號量等。
AQS 中,有一個核心狀態(tài)是 waitStatus,這個代表節(jié)點的狀態(tài),決定了當前節(jié)點的后續(xù)操作,比如是否等待喚醒,是否要喚醒后繼節(jié)點。
3 并發(fā)鎖
這一章節(jié)講解 Java AQS 中的并發(fā)鎖。其實 Java AQS 中的并發(fā)鎖主要是基于 state 這個變量值來實現(xiàn)的。
3.1 ReentrantLock
我們先來看一下 UML 類圖:
從圖中可以看到,ReentrantLock 使用抽象內(nèi)部類 Sync 來實現(xiàn)了 AQS 的方法,然后基于 Sync 這個同步器實現(xiàn)了公平鎖和非公平鎖。主要實現(xiàn)了下面 3 個方法:
- tryAcquire(int arg):獲取獨占鎖
- tryRelease(int arg):釋放獨占鎖
- isHeldExclusively:當前線程是否占有獨占鎖。ReentrantLock 默認實現(xiàn)的是非公平鎖,可以在構(gòu)造函數(shù)指定。
從實現(xiàn)的方法可以看到,ReentrantLock 中獲取的鎖是獨占鎖,我們再來看一下獲取和釋放獨占鎖的代碼:
public final void acquire(int arg){
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
selfInterrupt();
}獨占鎖的特點是調(diào)用上面 acquire 方法,傳入的參數(shù)是 1。
3.1.1 獲取公平鎖
獲取鎖首先判斷同步狀態(tài)(state)的值。
3.1.1.1 state 等于 0
這說明沒有線程占用鎖,當前線程如果符合下面兩個條件,就可以獲取到鎖:
沒有前任節(jié)點,如下圖:
CAS 的方式更新 state 值(把 0 更新成 1)成功。如果獲取獨占鎖成功,會更新 AQS 中 exclusiveOwnerThread 為當前線程,這個很容易理解。
3.1.1.2 state 不等于 0
這說明已經(jīng)有線程占有鎖,判斷占有鎖的線程是不是當前線程,如下圖:
state += 1 值如果小于 0,會拋出異常。
如果獲取鎖失敗,則進入 AQS 隊列等待喚醒。
3.1.2 獲取非公平鎖
跟公平鎖相比,非公平鎖的唯一不同是如果判斷到 state 等于 0,不用判斷有沒有前任節(jié)點,只要 CAS 設(shè)置 state 值(把 0 更新成 1)成功,就獲取到了鎖。
3.1.3 釋放鎖
公平鎖和非公平鎖,釋放邏輯完全一樣,都是在內(nèi)部類 Sync 中實現(xiàn)的。釋放鎖需要注意兩點,如下圖:
為什么 state 會大于 1,因為是可以重入的,占有鎖的線程可以多次獲取鎖。
3.1.4 總結(jié)
公平鎖的特點是每個線程都要進行排隊,不用擔心線程永遠獲取不到鎖,但有個缺點是每個線程入隊后都需要阻塞和被喚醒,這一定程度上影響了效率。非公平鎖的特點是每個線程入隊前都會先嘗試獲取鎖,如果獲取成功就不會入隊了,這比公平鎖效率高。但也有一個缺點,隊列中的線程有可能等待很長時間,高并發(fā)下甚至可能永遠獲取不到鎖。
3.2 ReentrantReadWriteLock
我們先來看一下 UML 類圖:
從圖中可以看到,ReentrantReadWriteLock 使用抽象內(nèi)部類Sync來實現(xiàn)了 AQS 的方法,然后基于 Sync 這個同步器實現(xiàn)了公平鎖和非公平鎖。主要實現(xiàn)了下面 3 個方法:
- tryAcquire(int arg):獲取獨占鎖
- tryRelease(int arg):釋放獨占鎖
- tryAcquireShared(int arg):獲取共享鎖
- tryReleaseShared(int arg):釋放共享鎖
- isHeldExclusively:當前線程是否占有獨占鎖 可見ReentrantReadWriteLock里面同時用到了共享鎖和獨占鎖。
下圖是定義的幾個常用變量:
下面這 2 個方法用戶獲取共享鎖和獨占鎖的數(shù)量:
static int sharedCount(int c){ return c >>> SHARED_SHIFT; }
static int exclusiveCount(int c){ return c & EXCLUSIVE_MASK; }從sharedCount 可以看到,共享鎖的數(shù)量要右移 16 位獲取,也就是說共享鎖占了高 16 位。從上圖 EXCLUSIVE_MASK 的定義看到,跟 EXCLUSIVE_MASK 進行與運算,得到的是低 16 位的值,所以獨占鎖占了低 16 位。如下圖:
這樣上面獲取鎖數(shù)量的方法就很好理解了。
3.2.1 讀鎖
讀鎖的實現(xiàn)對應(yīng)內(nèi)部類 ReadLock。
3.2.1.1 獲取讀鎖
獲取讀鎖實際上是 ReadLock 調(diào)用了 AQS 的下面方法,傳入?yún)?shù)是 1:
public final void acquireShared(int arg){
if (tryAcquireShared(arg) < 0)
doAcquireShared(arg);
}ReentrantReadWriteLock 內(nèi)部類 Sync 實現(xiàn)了 tryAcquireShared 方法,主要包括如下三種情況:
- 使用 exclusiveCount 方法查看 state 中是否有獨占鎖,如果有并且獨占線程不是當前線程,返回 -1,獲取失敗;
- 使用 sharedCount 查看 state 中共享鎖數(shù)量,如果讀鎖數(shù)量小于最大值(MAX_COUNT=65535),則再滿足下面 3 個條件就可以獲取成功并返回 1:
a.當前線程不需要阻塞(readerShouldBlock)。在公平鎖中,需要判斷是否有前置節(jié)點,如下圖就需要阻塞:
在非公平鎖中,則是判斷第一個節(jié)點是不是有獨占鎖,如下圖就需要阻塞:
b.使用 CAS 把 state 的值加 SHARED_UNIT(65536)。這里是不是就更理解讀鎖占高位的說法了,獲取一個讀鎖,state 的值就要加 SHARED_UNIT 這么多個。
c.給當前線程的 holdCount 加 1。
- 如果 2 失敗,自旋,重復(fù)上面的步驟直到獲取到鎖。tryAcquireShared (獲取共享鎖)會返回一個整數(shù),如下:
- 返回負數(shù):獲取鎖失敗。
- 返回 0:獲取鎖成功但是之后再由線程來獲取共享鎖時就會失敗。
- 返回正數(shù):獲取鎖成功而且之后再有線程來獲取共享鎖時也可能會成功。
3.2.1.2 釋放讀鎖
ReentrantReadWriteLock 釋放讀鎖是在 ReadLock 中調(diào)用了 AQS 下面方法,傳入的參數(shù)是1:
public final boolean releaseShared(int arg){
if (tryReleaseShared(arg)) {
doReleaseShared();
return true;
}
return false;
}ReentrantReadWriteLock 內(nèi)部類 Sync 實現(xiàn)了 releaseShared 方法,具體邏輯分為下面兩步:
- 當前線程 holdCounter 值減 1。
- CAS的方式將 state 的值減去 SHARED_UNIT。
3.2.2 寫鎖
寫鎖的實現(xiàn)對應(yīng)內(nèi)部類 WriteLock。
3.2.2.1 獲取寫鎖
ReentrantReadWriteLock 獲取寫鎖其實是在 WriteLock 中調(diào)用了 AQS 的下面方法,傳入?yún)?shù) 1:
public final void acquire(int arg){
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
selfInterrupt();
}在ReentrantReadWriteLock 內(nèi)部類 Sync 實現(xiàn)了 tryAcquire 方法,首先獲取 state 值和獨占鎖數(shù)量(exclusiveCount),之后分如下兩種情況,如下圖:
1.state 不等于 0:
- 獨占鎖數(shù)量等于 0,這時說明有線程占用了共享鎖,如果當前線程不是獨占線程,獲取鎖失敗。
- 獨占鎖數(shù)量不等于 0,獨占鎖數(shù)量加 1 后大于 MAX_COUNT,獲取鎖失敗。
- 上面 2 種情況不符合,獲取鎖成功,state 值加 1。2.state 等于 0,判斷當前線程是否需要阻塞(writerShouldBlock)。在公平鎖中,跟 readerShouldBlock 的邏輯完全一樣,就是判斷隊列中 head 節(jié)點的后繼節(jié)點是不是當前線程。在非公平鎖中,直接返回 false,即可以直接嘗試獲取鎖。
如果當前線程不需要阻塞,并且給 state 賦值成功,使用 CAS 方式把 state 值加 1,把獨占線程置為當前線程。
3.2.2.2 釋放寫鎖
ReentrantReadWriteLock 釋放寫鎖其實是在 WriteLock 中調(diào)用了 AQS 的下面方法,傳入?yún)?shù) 1:
public final boolean release(int arg){
if (tryRelease(arg)) {
Node h = head;
if (h != null && h.waitStatus != 0)
unparkSuccessor(h);
return true;
}
return false;
}ReentrantReadWriteLock 在 Sync 中實現(xiàn)了 tryRelease(arg) 方法,邏輯如下:
- 判斷當前線程是不是獨占線程,如果不是,拋出異常。
- state值減1后,用新state值判斷獨占鎖數(shù)量是否等于0
- 如果等于0,則把獨占線程置為空,返回true,這樣上面的代碼就可以喚醒隊列中的后置節(jié)點了
- 如果不等于0,返回false,不喚醒后繼節(jié)點。
3.3 CountDownLatch
我們先來看一下UML類圖:
從上面的圖中看出,CountDownLatch 的內(nèi)部類 Sync 實現(xiàn)了獲取共享鎖和釋放共享鎖的邏輯。
使用 CountDownLatch 時,構(gòu)造函數(shù)會傳入一個 int 類型的參數(shù) count,表示調(diào)動 count 次的 countDown 后主線程才可以被喚醒。
public CountDownLatch(int count){
if (count < 0) throw new IllegalArgumentException("count < 0");
this.sync = new Sync(count);
}上面的 Sync(count) 就是將 AQS 中的 state 賦值為 count。
3.3.1 await
CountDownLatch 的 await 方法調(diào)用了 AQS 中的 acquireSharedInterruptibly(int arg),傳入?yún)?shù) 1,不過這個參數(shù)并沒有用。代碼如下:
public final void acquireSharedInterruptibly(int arg)
throws InterruptedException {
if (Thread.interrupted())
throw new InterruptedException();
if (tryAcquireShared(arg) < 0)
doAcquireSharedInterruptibly(arg);
}
Sync 中實現(xiàn)了 tryAcquireShared 方法,await 邏輯如下圖:
上面的自旋過程就是等待 state 的值不斷減小,只有 state 值成為 0 的時候,主線程才會跳出自旋執(zhí)行之后的邏輯。
3.3.2 countDown
CountDownLatch 的 countDown 方法調(diào)用了 AQS 的 releaseShared(int arg),傳入?yún)?shù) 1,不過這個參數(shù)并沒有用。內(nèi)部類 Sync 實現(xiàn)了 tryReleaseShared 方法,邏輯如下圖:
3.3.3 總結(jié)
CountDownLatch 的構(gòu)造函數(shù)入?yún)⒅禃x值給 state 變量,入隊操作是主線程入隊,每個子線程調(diào)用了countDown 后 state 值減 1,當 state 值成為 0 后喚醒主線程。
3.4 Semaphore
Semaphore 是一個信號量,用來保護共享資源。如果線程要訪問共享資源,首先從 Semaphore 獲取鎖(信號量),如果信號量的計數(shù)器等于 0,則當前線程進入 AQS 隊列阻塞等待。否則,線程獲取鎖成功,信號量減 1。使用完共享資源后,釋放鎖(信號量加 1)。
Semaphore 跟管程模型不一樣的是,允許多個(構(gòu)造函數(shù)的 permits)線程進入管程內(nèi)部,因此也常用它來做限流。
UML 類圖如下:
Semaphore的構(gòu)造函數(shù)會傳入一個int類型參數(shù),用來初始化state的值。
3.4.1 acquire
獲取鎖的操作調(diào)用了 AQS 中的 acquireSharedInterruptibly 方法,傳入?yún)?shù) 1,代碼見 CountDownLatch 中 await 小節(jié)。Semaphore 在公平鎖和非公平鎖中分別實現(xiàn)了 tryAcquireShared 方法。
3.4.1.1 公平鎖
Semaphore 默認使用非公平鎖,如果使用公平鎖,需要在構(gòu)造函數(shù)指定。獲取公平鎖邏輯比較簡單,如下圖:
3.4.1.2 非公平鎖
acquire 在非公平的鎖唯一的區(qū)別就是不會判斷 AQS 隊列是否有前置節(jié)點(hasQueuedPredecessors),而是直接嘗試獲取鎖。
除了 acquire 方法外,還有其他幾個獲取鎖的方法,原理類似,只是調(diào)用了 AQS 中的不同方法。
3.4.2 release
釋放鎖的操作調(diào)用了 AQS 中的 releaseShared(int arg) 方法,傳入?yún)?shù) 1,在內(nèi)部類 Sync 中實現(xiàn)了 tryReleaseShared 方法,邏輯很簡單:使用 CAS 的方式將 state 的值加 1,之后喚醒隊列中的后繼節(jié)點。
3.5 ThreadPoolExecutor
ThreadPoolExecutor 中也用到了 AQS,看下面的 UML 類圖:
Worker 主要在 ThreadPoolExecutor 中斷線程的時候使用。Worker 自己實現(xiàn)了獨占鎖,在中斷線程時首先進行加鎖,中斷操作后釋放鎖。按照官方說法,這里不直接使用 ReentrantLock 的原因是防止調(diào)用控制線程池的方法(類似 setCorePoolSize)時能夠重新獲取到鎖,
3.5.1 tryAcquire
使用 CAS 的方式把 AQS 中 state 從 0 改為 1,把當前線程置為獨占線程。
3.5.2 tryRelease
把獨占線程置為空,把 AQS 中 state 改為 0。
Worker 初始化的時候會把 state 置為 -1,這樣是不能獲取鎖成功的。只有調(diào)用了 runWorker 方法,才會通過釋放鎖操作把 state 更為 0。這樣保證了只中斷運行中的線程,而不會中斷等待中的線程。
3.6 總結(jié)
AQS 基于雙向隊列實現(xiàn)了入口等待隊列,基于 state 變量實現(xiàn)了各種并發(fā)鎖,上篇文章講了入口等待隊列,而這篇文章主要講了基于 AQS 的并發(fā)鎖原理。
4、條件變量等待隊列
本章節(jié)主要講解管程模型中條件變量等待隊列。
4.1 官方示例
首先我們看一下官方給出的示例代碼:
public class BoundedBuffer {
final Lock lock = new ReentrantLock();
final Condition notFull = lock.newCondition();
final Condition notEmpty = lock.newCondition();
final Object[] items = new Object[100];
int putptr, takeptr, count;
public void put(Object x) throws InterruptedException {
lock.lock();
try {
while (count == items.length)
notFull.await();
items[putptr] = x;
if (++putptr == items.length) putptr = 0;
++count;
notEmpty.signal();
} finally {
lock.unlock();
}
}
public Object take() throws InterruptedException {
lock.lock();
try {
while (count == 0)
notEmpty.await();
Object x = items[takeptr];
if (++takeptr == items.length) takeptr = 0;
--count;
notFull.signal();
return x;
} finally {
lock.unlock();
}
}
}這個代碼定義了兩個條件變量,notFull 和 notEmpty,說明如下:
- 如果 items 數(shù)組已經(jīng)滿了,則 notFull 變量不滿足,線程需要進入 notFull 條件等待隊列進行等待。當 take 方法取走一個數(shù)組元素時,notFull 條件滿足了,喚醒 notFull 條件等待隊列中等待線程。
- 如果 items 數(shù)組為空,則 notEmpty 變量不滿足,線程需要進入 notEmpty 條件等待隊列進行等待。當 put 方法加入一個數(shù)組元素時,notEmpty 條件滿足了,喚醒 notEmpty 條件等待隊列中等待線程。
- 條件變量是綁定在 Lock 上的,示例代碼使用了 ReentrantLock。在執(zhí)行 await 和 signal 方法時首先要獲取到鎖。
4.2 原理簡介
Java AQS 的條件變量等待隊列是基于接口 Condition 和 ConditionObject 來實現(xiàn)的,URM 類圖如下:
Condition 接口主要定義了下面3個方法:
- await:進入條件等待隊列
- signal:喚醒條件等待隊列中的元素
- signalAll:喚醒條件等待隊列中的所有元素
4.3 await
條件等待隊列跟入口等待隊列有兩個不同:
- 雖然二者共用了 Node 類,但是條件等待隊列是單向隊列,入口等待隊列是雙向隊列,條件隊列中下一個節(jié)點的引用是 nextWaiter,入口等待隊列中下一個節(jié)點的引用是 next。
- 條件等待隊列中元素的 waitStatus 必須是 -2。await 方法的流程如下圖:
4.3.1 進入條件等待隊列
入隊方法對應(yīng)方法 addConditionWaiter,這里有三種情況:
- 隊列為空,則新建一個節(jié)點,如下圖:
- 隊列非空,最后一個元素的 waitStatus 是 -2,如下圖:
- 隊列非空,最后一個元素的 waitStatus 不是 -2,如下圖:
可以看到,這種情況會從隊列第一個元素開始檢查 waitStatus 不是 -2 的元素,并從隊列中移除。
4.3.2 釋放鎖
AQS 的并發(fā)鎖是基于 state 變量實現(xiàn)的,線程進入條件等待隊列后,要釋放鎖,即 state 會變?yōu)?0,釋放操作會喚醒入口等待隊列中的線程。對應(yīng)方法 fullyRelease,返回值是釋放鎖減掉的 state 值 savedState。
4.3.3 阻塞等待
釋放鎖后,線程阻塞,自旋等待被喚醒。
4.3.4 喚醒之后
喚醒之后,當前線程主要有四個動作:
- 轉(zhuǎn)入入口等待隊列,并把 waitStatus 改為 0。waitStatus 等于 0 表示中間狀態(tài),當前節(jié)點后面的節(jié)點已經(jīng)喚醒,但是當前節(jié)點線程還沒有執(zhí)行完成。
- 重新獲取鎖,如果獲取成功,則當前線程成為入口等待隊列頭結(jié)點,interruptMode 置為 1。
- 如果當前節(jié)點在條件等待隊列中有后繼節(jié)點,則剔除條件等待隊列中 waitStatus!=-2 的節(jié)點,即隊列中狀態(tài)為取消的節(jié)點。
- interruptMode 如果不等于 0,則處理中斷。
4.3.5 一個細節(jié)
上面提到了 interruptMode,這個屬性有三個值:
- 0:沒有被中斷
- -1:中斷后拋出 InterruptedException,這種情況是當前線程阻塞,沒有被 signal 之前發(fā)生了中斷
- 1:重新進入中斷狀態(tài),這種情況是指當前線程阻塞,被 signal 之后發(fā)生了中斷
4.3.6 擴展
AQS 還提供了其他幾個 await 方法,如下:
- awaitUninterruptibly:不用處理中斷。
- awaitNanos:自旋等待喚醒過程中有超時時間限制,超時則轉(zhuǎn)入入口等待隊列。
- awaitUntil:自旋等待喚醒過程中有截止時間,時間到則轉(zhuǎn)入入口等待隊列。
4.4 signal
喚醒條件等待隊列中的元素,首先判斷當前線程是否持有獨占鎖,如果沒有,拋出異常。
喚醒條件隊列中的元素,會從第一個元素也就是 firstWaiter 開始,根據(jù) firstWaiter 的 waitStatus 是不是 -2,分兩種情況。
4.4.1 waitStatus==-2
條件隊列第一個節(jié)點進入入口等待隊列,等待獲取鎖,如下圖:
這里有兩個注意點:
- 如果入口等待隊列中 tail 節(jié)點的 waitStatus 小于等于 0,則 firstWaiter 加入后需要把舊 tail 節(jié)點置為 -1 (表示后面節(jié)點等待當前節(jié)點喚醒),如下圖:
如果重置 waitStatus 狀態(tài)失敗,則 unpark 節(jié)點 firstWaiter。
- 如果入口等待隊列中 tail 節(jié)點的 waitStatus 大于 0,則 unpark 節(jié)點 firstWaiter。
4.4.2 waitStatus!=-2
如果 firstWaiter 的 waitStatus 不等于 -2,則查找 firstWaiter 的 nextWaiter,直到找到一個 waitStatus 等于 -2 的節(jié)點,然后將這個節(jié)點加入入口等待隊列隊尾,如下圖:
4.4.3 waitStatus 修改
上面的兩種情況無論哪種,進入入口等待隊列之前都要用 CAS 的方式把 waitStatus 改為 0。
4.5 signalAll
理解了 signal 的邏輯,signalAll 的邏輯就非常容易理解了。首先判斷當前線程是否持有獨占鎖,如果沒有,拋出異常。
將條件等待隊列中的所有節(jié)點依次加入入口等待隊列。如下圖:
4.6 使用案例
4.6.1 示例代碼
Java 并發(fā)包下有很多類使用到了 AQS 中的 Condition,如下圖:
這里我們以 CyclicBarrier 為例來講解。CyclicBarrier 是讓一組線程相互等待共同達到一個屏障點。從 Cyclic 可以看出 Barrier 可以循環(huán)利用,也就是當線程釋放之后可以繼續(xù)使用。
看下面這段示例代碼:
public static void main(String[] args){
CyclicBarrier cyclicBarrier = new CyclicBarrier(2, () -> {
System.out.println("柵欄中的線程執(zhí)行完成");
});
ExecutorService executorService = Executors.newFixedThreadPool(2);
executorService.submit(() -> {
try {
System.out.println("線程1:" + Thread.currentThread().getName());
cyclicBarrier.await();
} catch (Exception e) {
e.printStackTrace();
}
});
executorService.submit(() -> {
try {
System.out.println("線程2:" + Thread.currentThread().getName());
cyclicBarrier.await();
} catch (Exception e) {
e.printStackTrace();
}
});
executorService.shutdown();
}執(zhí)行結(jié)果:
線程1:pool-1-thread-1
線程2:pool-1-thread-2
柵欄中的線程執(zhí)行完成
4.6.2 原理講解
CyclicBarrier 初始化的時候,會指定線程的數(shù)量 count,每個線程執(zhí)行完邏輯后,調(diào)用 CyclicBarrier 的 await 方法,這個方法首先將 count 減 1,然后調(diào)用 Condition的 await,讓當前線程進入條件等待隊列。當最后一個線程將 count 減 1 后,count 數(shù)量等于 0,這時就會調(diào)用 Condition 的 signalAll 方法喚醒所有線程。
4.7 總結(jié)
Java 的管程模型使用了 MESA 模型,基于 AQS 實現(xiàn)的 MESA 模型中,使用雙向隊列實現(xiàn)了入口等待隊列,使用變量 state 實現(xiàn)了并發(fā)鎖,使用 Condition 實現(xiàn)了條件等待隊列。
在 AQS 的實現(xiàn)中,使用同步隊列這個術(shù)語來表示雙向隊列,本文中使用入口等待隊列來描述是為了更好的配合管程模型來講解。
AQS 的 Condition 中,使用 await 方法將當前線程放入條件變量等待隊列阻塞等待,使用 notify 來喚醒條件等待隊列中的線程,被喚醒之后,線程并不能立刻執(zhí)行,而是進入入口等待隊列等待獲取鎖。
網(wǎng)站名稱:為了帶你精通JavaAQS,我畫了40張圖,從管程模型講起!
新聞來源:http://www.5511xx.com/article/dhogpog.html


咨詢
建站咨詢
