面試官:好了域帐,聊完了ArrayBlockingQueue
,我們接著說說LinkedBlockingQueue
吧
Hydra:還真是不給人喘口氣的機會是整,LinkedBlockingQueue
是一個基于鏈表的阻塞隊列肖揣,內(nèi)部是由節(jié)點Node
構(gòu)成,每個被加入隊列的元素都會被封裝成下面的Node
節(jié)點浮入,并且節(jié)點中有指向下一個元素的指針:
static class Node<E> {
E item;
Node<E> next;
Node(E x) { item = x; }
}
LinkedBlockingQueue
中的關(guān)鍵屬性有下面這些:
private final int capacity;//隊列容量
private final AtomicInteger count = new AtomicInteger();//隊列中元素數(shù)量
transient Node<E> head;//頭節(jié)點
private transient Node<E> last;//尾節(jié)點
//出隊鎖
private final ReentrantLock takeLock = new ReentrantLock();
//出隊的等待條件對象
private final Condition notEmpty = takeLock.newCondition();
//入隊鎖
private final ReentrantLock putLock = new ReentrantLock();
//入隊的等待條件對象
private final Condition notFull = putLock.newCondition();
構(gòu)造函數(shù)分為指定隊列長度和不指定隊列長度兩種龙优,不指定時隊列最大長度是int
的最大值。當(dāng)然了,你要是真存這么多的元素,很有可能會引起內(nèi)存溢出:
public LinkedBlockingQueue() {
this(Integer.MAX_VALUE);
}
public LinkedBlockingQueue(int capacity) {
if (capacity <= 0) throw new IllegalArgumentException();
this.capacity = capacity;
last = head = new Node<E>(null);
}
還有另一種在初始化時就可以將集合作為參數(shù)傳入的構(gòu)造方法,實現(xiàn)非常好理解,只是循環(huán)調(diào)用了后面會講到的enqueue
入隊方法轧苫,這里暫且略過岔乔。
在LinkedBlockingQueue
中茁影,隊列的頭節(jié)點head
是不存元素的蝇更,它的item
是null
访圃,next
指向的元素才是真正的第一個元素,它也有兩個用于阻塞等待的Condition
條件對象徽鼎。與之前的ArrayBlockingQueue
不同棠隐,這里出隊和入隊使用了不同的鎖takeLock
和putLock
嗡贺。隊列的結(jié)構(gòu)是這樣的:
面試官:為什么要使用兩把鎖,之前ArrayBlockingQueue
使用一把鎖,不是也可以保證線程的安全么土全?
Hydra:使用兩把鎖概页,可以保證元素的插入和刪除并不互斥项鬼,從而能夠同時進行,達到提高吞吐量的的效果
面試官:嗯,那還是老規(guī)矩,先說插入方法是怎么實現(xiàn)的吧
Hydra:這次就不提父類AbstractQueue
的add
方法了程奠,反正它調(diào)用的也是子類的插入方法offer
距境,我們就直接來看offer
方法的源碼:
public boolean offer(E e) {
if (e == null) throw new NullPointerException();
final AtomicInteger count = this.count;//隊列中元素個數(shù)
if (count.get() == capacity)//已滿
return false;
int c = -1;
Node<E> node = new Node<E>(e);
final ReentrantLock putLock = this.putLock;
putLock.lock();
try {
//并發(fā)情況,再次判斷隊列是否已滿
if (count.get() < capacity) {
enqueue(node);
//注意這里獲取的是未添加元素前的對列長度
c = count.getAndIncrement();
if (c + 1 < capacity)//未滿
notFull.signal();
}
} finally {
putLock.unlock();
}
if (c == 0)
signalNotEmpty();
return c >= 0;
}
offer
方法中,首先判斷隊列是否已滿,未滿情況下將元素封裝成Node
對象欣鳖,嘗試獲取插入鎖怀酷,在獲取鎖后會再進行一次隊列已滿判斷檐束,如果已滿則直接釋放鎖格嘁。在持有鎖且隊列未滿的情況下苗膝,調(diào)用enqueue
入隊方法嵌戈。
enqueue
方法的實現(xiàn)也非常的簡單,將當(dāng)前尾節(jié)點的next
指針指向新節(jié)點降传,再把last
指向新節(jié)點:
private void enqueue(Node<E> node) {
last = last.next = node;
}
畫一張圖,方便你理解:
在完成入隊后依沮,判斷隊列是否已滿审磁,如果未滿則調(diào)用notFull.signal()
瘩蚪,喚醒等待將元素插入隊列的線程汹押。
面試官:我記得在ArrayBlockingQueue
里插入元素后细诸,是調(diào)用的notEmpty.signal()
塘偎,怎么這里還不一樣了?
Hydra:說到這赔退,就不得不再提一下使用兩把鎖來分別控制插入和獲取元素的好處了离钝。在ArrayBlockingQueue
中,使用了同一把鎖對入隊和出隊進行控制吱肌,那么如果在插入元素后再喚醒插入線程规揪,那么很有可能等待獲取元素的線程就一直得不到喚醒揖庄,造成等待時間過長。
而在LinkedBlockingQueue
中狂打,分別使用了入隊鎖putLock
和出隊鎖takeLock
对省,插入線程和獲取線程是不會互斥的。所以插入線程可以在這里不斷的喚醒其他的插入線程晾捏,而無需擔(dān)心是否會使獲取線程等待時間過長蒿涎,從而在一定程度上提高了吞吐量。當(dāng)然了惦辛,因為offer
方法是非阻塞的劳秋,并不會掛起阻塞線程,所以這里喚醒的是阻塞插入的put
方法的線程胖齐。
面試官:那接著往下看玻淑,為什么要在c
等于0的情況下才去喚醒notEmpty
中的等待獲取元素的線程?
Hydra:其實獲取元素的方法和上面插入元素的方法是一個模式的市怎,只要有一個獲取線程在執(zhí)行方法岁忘,那么就會不斷的通過notEmpty.signal()
喚醒其他的獲取線程辛慰。只有當(dāng)c
等于0時区匠,才證明之前隊列中已經(jīng)沒有元素,這時候獲取線程才可能會被阻塞,在這個時候才需要被喚醒驰弄。上面的這些可以用一張圖來說明:
由于我們之前說過麻汰,隊列中的head
節(jié)點可以認(rèn)為是不存儲數(shù)據(jù)的標(biāo)志性節(jié)點,所以可以簡單的認(rèn)為出隊時直接取出第二個節(jié)點戚篙,當(dāng)然這個過程不是非常的嚴(yán)謹(jǐn)五鲫,我會在后面講解出隊的過程中再進行補充說明。
面試官:那么阻塞方法put
和它有什么區(qū)別岔擂?
Hydra:put
和offer
方法整體思路一致位喂,不同的是加鎖是使用的是可被中斷的方式,并且當(dāng)隊列中元素已滿時乱灵,將線程加入notFull
等待隊列中進行等待塑崖,代碼中體現(xiàn)在:
while (count.get() == capacity) {
notFull.await();
}
這個過程體現(xiàn)在上面那張圖的notFull
等待隊列中的元素上,就不重復(fù)說明了痛倚。另外规婆,和put
方法比較類似的,還有一個攜帶等待時間參數(shù)的offer
方法蝉稳,可以進行有限時間內(nèi)的阻塞添加抒蚜,當(dāng)超時后放棄插入元素,我們只看和offer
方法不同部分的代碼:
public boolean offer(E e, long timeout, TimeUnit unit){
...
long nanos = unit.toNanos(timeout);//轉(zhuǎn)換為納秒
...
while (count.get() == capacity) {
if (nanos <= 0)
return false;
nanos = notFull.awaitNanos(nanos);
}
enqueue(new Node<E>(e));
...
}
awaitNanos
方法在await
方法的基礎(chǔ)上耘戚,增加了超時跳出的機制嗡髓,會在循環(huán)中計算是否到達預(yù)設(shè)的超時時間。如果在到達超時時間前被喚醒收津,那么會返回超時時間減去已經(jīng)消耗的時間器贩。無論是被其他線程喚醒返回,還是到達指定的超時時間返回朋截,只要方法返回值小于等于0蛹稍,那么就認(rèn)為它已經(jīng)超時,最終直接返回false
結(jié)束部服。
面試官:費這么大頓功夫才把插入講明白唆姐,我先喝口水,你接著說獲取元素方法
Hydra:……那先看非阻塞的poll
方法
public E poll() {
final AtomicInteger count = this.count;
if (count.get() == 0)//隊列為空
return null;
E x = null;
int c = -1;
final ReentrantLock takeLock = this.takeLock;
takeLock.lock();
try {
if (count.get() > 0) {//隊列非空
x = dequeue();
//出隊前隊列長隊
c = count.getAndDecrement();
if (c > 1)
notEmpty.signal();
}
} finally {
takeLock.unlock();
}
if (c == capacity)
signalNotFull();
return x;
}
出隊的邏輯和入隊的非常相似廓八,當(dāng)隊列非空時就執(zhí)行dequeue
進行出隊操作奉芦,完成出隊后如果隊列仍然非空,那么喚醒等待隊列中掛起的獲取元素的線程剧蹂。并且當(dāng)出隊前的元素數(shù)量等于隊列長度時声功,在出隊后喚醒等待隊列上的添加線程。
出隊方法dequeue
的源碼如下:
private E dequeue() {
Node<E> h = head;
Node<E> first = h.next;
h.next = h; // help GC
head = first;
E x = first.item;
first.item = null;
return x;
}
之前提到過宠叼,頭節(jié)點head
并不存儲數(shù)據(jù)先巴,它的下一個節(jié)點才是真正意義上的第一個節(jié)點其爵。在出隊操作中,先得到頭節(jié)點的下一個節(jié)點first
節(jié)點伸蚯,將當(dāng)前頭節(jié)點的next
指針指向自己摩渺,代碼中有一個簡單的注釋是help gc
,個人理解這里是為了降低gc
中的引用計數(shù)剂邮,方便它更早被回收摇幻。之后再將新的頭節(jié)點指向first
,并返回清空為null
前的內(nèi)容挥萌。使用圖來表示是這樣的:
面試官:(看看手表)take
方法的整體邏輯也差不多绰姻,能簡單概括一下嗎
Hydra:阻塞方法take
方法和poll
的思路基本一致,是一個可以被中斷的阻塞獲取方法引瀑,在隊列為空時龙宏,會掛起當(dāng)前線程,將它添加到條件對象notEmpty
的等待隊列中伤疙,等待其他線程喚醒银酗。
面試官:再給你一句話的時間,總結(jié)一下它和ArrayBlockingQueue
的異同徒像,我要下班回家了
Hydra:好吧黍特,我總結(jié)一下,有下面幾點:
- 隊列長度不同锯蛀,
ArrayBlockingQueue
創(chuàng)建時需指定長度并且不可修改灭衷,而LinkedBlockingQueue
可以指定也可以不指定長度 - 存儲方式不同,
ArrayBlockingQueue
使用數(shù)組旁涤,而LinkedBlockingQueue
使用Node
節(jié)點的鏈表 -
ArrayBlockingQueue
使用一把鎖來控制元素的插入和移除翔曲,而LinkedBlockingQueue
將入隊鎖和出隊鎖分離,提高了并發(fā)性能 -
ArrayBlockingQueue
采用數(shù)組存儲元素劈愚,因此在插入和移除過程中不需要生成額外對象瞳遍,LinkedBlockingQueue
會生成新的Node
節(jié)點,對gc
會有影響
面試官:明天上午9點菌羽,老地方掠械,我們再聊聊別的
Hydra:……
如果文章對您有所幫助,歡迎關(guān)注公眾號
碼農(nóng)參上