隊列(Queue)是一種先進先出的數據結構,Java中實現了Queue<E>接口的類都具有隊列的功能。我們常用的LinkedList就實現了Queue接口,具有了offer()、poll()等操作。但是,LinkedList是線程不安全的,只使用於單線程操作。如果要實現一個線程安全的隊列,一般有兩種方式:一是使用阻塞算法,給讀寫操作加鎖;二是使用非阻塞算法,通過循環CAS的方式來實現。本篇文章的ConcurrentLinkedQueue就是併發大師Doug Lea設計的一個非阻塞線程安全隊列。
1 ConcurrentLinkedQueue的結構
2 入隊操作
public boolean offer(E e) {
if (e == null) throw new NullPointerException();
//入隊前,創建一個入隊節點
Node<E> n = new Node</e><e>(e);
retry:
//死循環,入隊不成功反覆入隊。
for (;;) {
//創建一個指向tail節點的引用
Node<E> t = tail;
//p用來表示隊列的尾節點,默認情況下等於tail節點。
Node<E> p = t;
for (int hops = 0; ; hops++) {
//獲得p節點的下一個節點。
Node<E> next = succ(p);
//next節點不爲空,說明p不是尾節點,需要更新p後在將它指向next節點
if (next != null) {
//循環了兩次及其以上,並且當前節點還是不等於尾節點
if (hops > HOPS && t != tail)
continue retry;
p = next;
}
//如果p是尾節點,則設置p節點的next節點爲入隊節點。
else if (p.casNext(null, n)) {
//如果tail節點有大於等於1個next節點,則將入隊節點設置成tair節點,更新失敗了也沒關係,因爲失敗了表示有其他線程成功更新了tair節點。
if (hops >= HOPS)
casTail(t, n); // 更新tail節點,允許失敗
return true;
}
// p有next節點,表示p的next節點是尾節點,則重新設置p節點
else {
p = succ(p);
}
}
}
}
3 出隊操作
同tail節點,head節點爲了減少CAS操作,也採取了類似的規則:
1)head中有元素時,彈出元素,不更新head節點。
2)head中爲空時,彈出next元素,更新head節點爲真正的頭節點。
poll()函數的源碼如下:
public E poll() {
Node</e><e> h = head;
// p表示頭節點,需要出隊的節點
Node</e><e> p = h;
for (int hops = 0;; hops++) {
// 獲取p節點的元素
E item = p.getItem();
// 如果p節點的元素不爲空,使用CAS設置p節點引用的元素爲null,如果成功則返回p節點的元素。
if (item != null && p.casItem(item, null)) {
if (hops >= HOPS) {
//將p節點下一個節點設置成head節點
Node</e><e> q = p.getNext();
updateHead(h, (q != null) ? q : p);
}
return item;
}
// 如果頭節點的元素爲空或頭節點發生了變化,這說明頭節點已經被另外一個線程修改了。那麼獲取p節點的下一個節點
Node</e><e> next = succ(p);
// 如果p的下一個節點也爲空,說明這個隊列已經空了
if (next == null) {
// 更新頭節點。
updateHead(h, p);
break;
}
// 如果下一個元素不爲空,則將頭節點的下一個節點設置成頭節點
p = next;
}
return null;
}
先取出頭節點元素,如果不爲空,則通過CAS操作更新。更新失敗則說明已經被另外一個線程修改了,直接獲取p的下一個節點。如果爲空,則更新頭結點。