面試官:好了,你也休息了十分鍾了,咱們接着往下聊聊SynchronousQueue
的非公平模式吧。
Hydra:好的,有了前面公平模式的基礎,非公平模式理解起來就非常簡單了。公平模式下,SynchronousQueue
底層使用的是TransferQueue
,是一個先進先出的隊列,而非公平模式與它不同,底層采用了后進先出的TransferStack
棧來實現。
下面我們還是先寫一個例子來看看效果,首先創建3個線程使用put
方法向SynchronousQueue
中插入數據,結束后再使用3個線程調用take
方法:
SynchronousQueue<Integer> queue=new SynchronousQueue<>(false);
@AllArgsConstructor
class PutThread implements Runnable{
int i;
@SneakyThrows
@Override
public void run() {
queue.put(i);
System.out.println("putThread "+i+" end");
}
}
class TakeThread implements Runnable{
@SneakyThrows
@Override
public void run() {
System.out.println("takeThread take: "+queue.take());
}
}
for (int i = 1; i <=3; i++) {
new Thread(new PutThread(i)).start();
Thread.sleep(1000);
}
for (int i = 1; i <=3 ; i++) {
new Thread(new TakeThread()).start();
Thread.sleep(1000);
}
運行上面的代碼,查看結果:
takeThread take: 3
putThread 3 end
takeThread take: 2
putThread 2 end
takeThread take: 1
putThread 1 end
可以看到,生產者線程在執行完put
后會進行阻塞,直到有消費者線程調用take
方法取走了數據,才會喚醒被阻塞的線程。並且,數據的出隊與入隊順序是相反的,即非公平模式下采用的是后進先出的順序。
面試官:就是把結構從隊列換成了棧,真就這么簡單?
Hydra:並不是,包括底層節點以及出入棧的邏輯都做了相應的改變。我們先看節點,在之前的公平模式中隊列的節點是QNode
,非公平模式下棧中節點是SNode
,定義如下:
volatile SNode next; // 指向下一個節點的指針
volatile SNode match; // 存放和它進行匹配的節點
volatile Thread waiter; // 保存阻塞的線程
Object item;
int mode;
SNode(Object item) {
this.item = item;
}
和QNode
類似,如果是生產者構建的節點,那么item
非空,如果是消費者產生的節點,那么item
為null
。此外還有一個mode
屬性用來表示節點的狀態,它使用TransferStack
中定義的3個常量來表示不同狀態:
static final int REQUEST = 0; //消費者
static final int DATA = 1; //生產者
static final int FULFILLING = 2; //匹配中狀態
TransferStack
中沒有攜帶參數的構造函數,使用一個head
節點來標記棧頂節點:
volatile SNode head;
面試官:基本結構就講到這吧,還是老規矩,先從入隊操作開始分析吧。
Hydra:當棧為空、或棧頂元素的類型與自己相同時,會先創建一個SNode
節點,並將它的next
節點指向當前棧頂的head
,然后將head
指針指向自己。這個過程中通過使用CAS
保證線程安全,如果失敗則退出,在循環中采取自旋的方式不斷進行嘗試,直到節點入棧成功。用一張圖來表示兩個線程同時入棧的場景:
當節點完成入棧后,調用awaitFulfill
方法,等待匹配的操作的到來。在這一過程中,會使節點對應的線程進行自旋或掛起操作,直到匹配操作的節點將自己喚醒,或被其他線程中斷、等待超時。
當入棧后的節點是棧頂節點,或者節點的類型為FULFILLING
匹配狀態時,那么可能會馬上完成匹配,因此先進行自旋,當超過自旋次數上限后再掛起。而如果節點在自旋過程中,有新的節點壓入棧頂,會將非棧頂節點剩余的自旋次數直接清零,掛起線程避免浪費資源。
面試官:你上面也說了,掛起的線程有可能會超時或者被中斷,這時候應該怎么處理?
Hydra:當這兩種情況出現時,SNode
會將match
屬性設為自身,退出awaitFulfill
方法,然后調用clean
方法將對應的節點清理出棧。具體情形可分為兩種情況。先說簡單的情況,如果清理的是棧頂節點,那么直接將head
節點指向它的next
節點,即將當前棧頂結點彈出即可。
面試官:那么如果要刪除的節點不是棧頂的節點呢?
Hydra:如果清理的不是棧頂節點,會稍微有一些麻煩。因為棧的底層是一個單向的鏈表結構,所以需要從棧頂head
節點開始遍歷,遍歷到被刪除節點的后繼節點為止。所以在清除工作開始前,先使用了一個past
節點標記需要刪除節點的下一個節點,作為結束遍歷的標記。
然后創建一個標記節點p
,初始時指向head
結點,開始循環,如果p
的next
節點不是需要被刪除的節點,那么就將p
向后移一個位置,直到找到這個需要被刪除的中斷或超時的節點,然后將p
的next
指向這個刪除節點的next
節點,在邏輯上完成鏈表中節點的刪除。
面試官:單一類型節點的入棧應該說完了吧,接下來說說不同類型節點間是如何實現的匹配操作吧?
Hydra:好的,那我們先回顧一點上面的知識,前面說過每個節點有一個mode
屬性代表它的模式,REQUEST
表示它是消費者,DATA
表示是生產者,FULFILLING
表明正處於匹配中的狀態。
在一個新的線程調用方法時,先判斷它的類型mode
是什么,如果和當前棧頂head
節點類型不同,且head
節點的狀態不為匹配中時,將它的狀態設置為FULFILLING|mode
,壓入棧中。然后將嘗試匹配新的head
節點和它的next
節點,如果匹配成功,會將next
節點的match
屬性設置為head
節點,喚醒掛起的next
節點中的線程。
在完成匹配后,當前頭結點對應的線程會協助推進head
節點,將head
指向next
節點的下一個節點,即完成了棧頂兩節點的出棧。最終消費者線程會返回匹配的生產者節點中的item
數據值,而生產者線程也會結束運行退出。
我們以棧中當前節點為DATA
類型,新節點為REQUEST
類型畫一張圖,來直觀的感受一下上面的流程:
面試官:總算是講完了,能對SynchronousQueue
做一個簡單的總結嗎?
Hydra:SynchronousQueue
基於底層結構,實現了線程配對通信這一機制。在它的公平模式下使用的是先進先出(FIFO
)的隊列,非公平模式下使用的是后進先出(LIFO
)的棧,並且SynchronousQueue
沒有使用synchronized
或ReentrantLock
,而是使用了大量的CAS
操作來保證並發操作。可能我們在平常的工作中使用場景不是很多,但是在線程池的設計中使用了SynchronousQueue
,還是有很重要的應用場景的。
面試官:講的還行,不過剛才這些和公平模式聽起來感覺區別不大啊,沒有什么技術含量。這樣吧,你明天過來我們加試一場,我再給你打分。
Hydra:(溜了溜了,還是找家別的靠譜公司吧……)
最后
如果覺得對您有所幫助,小伙伴們可以點贊、轉發一下~非常感謝
微信搜索:碼農參上,來加個好友,點贊之交也好啊~
公眾號后台回復“面試”、“導圖”、“架構”、“實戰”,獲得免費資料哦~