ConcurrentLinkedQueue
ConcurrentLinkedQueue 的实现原理和AQS内部的阻塞队列类似,AQS是基于双向链表,通过对head/tail进行CAS操作,实现入队和出队,队列里面放的是线程,而ConcurrentLinkedQueue是一个单向链表,里面放的是实际的元素
public class ConcurrentLinkedQueue<E> extends AbstractQueue<E> implements
Queue<E>, java.io.Serializable {
private static class Node<E> {
volatile E item;
volatile Node<E> next;
}
private transient volatile Node<E> head;
private transient volatile Node<E> tail;
}
在AQS的阻塞队列中,每次入队后,tail一定后移一个位置,指向新的最后一个元素;每次出队,head一定后移一个位置,以保证head指向队列头部,tail指向链表尾部。
但在ConcurrentLinkedQueue中,head/tail的更新可能落后于节点的入队和出队,因为它不是直接对head/tail指针进行CAS操作的,而是对Node中的item进行操作。
ConcurrentLinkedQueue代码实现
public ConcurrentLinkedQueue() {
head = tail = new Node<E>(null);
}
public boolean offer(E e) {
checkNotNull(e);
final Node<E> newNode = new Node<E>(e);
for (Node<E> t = tail, p = t;;) {
Node<E> q = p.next;
if (q == null) {
if (p.casNext(null, newNode)) {
if (p != t)
casTail(t, newNode);
return true;
}
}
else if (p == q)
p = (t != (t = tail)) ? t : head;
else
p = (p != t && t != (t = tail)) ? t : q;
}
}
假设队列中有1个节点item1,tail指向该节点,假设有一个线程要入队item2节点
- p=tail,t=tail, p.next=NULL, q=p.next
- p.casNext对p.next进行cas操作,设置p.next = item2,执行后p依然等于tail,即p等于t所以casTail不会执行,return true
然后有一个线程要入队item3
- p=tail,q=p.next
- q!=NULL,因此不会入队新节点。p=q 即p=p.next,for循环回到第一排
- q=p.next即q=null,对p的next执行CAS操作,入队item3节点
- p!=t, 执行casTail操作, tail后移2个位置,到达队列尾部,每追加2个节点,才后移1次tail指针
public E poll() {
restartFromHead:
for (;;) {
for (Node<E> h = head, p = h, q;;) {
E item = p.item;
if (item != null && p.casItem(item, null)) {
if (p != h)
updateHead(h, ((q = p.next) != null) ? q : p);
return item;
}
else if ((q = p.next) == null) {
updateHead(h, p);
return null;
}
else if (p == q)
continue restartFromHead;
else
p = q;
}
}
}
- p=head, p.item = null
- q=p.next,p!=q执行p=q(p指针后移)进入下次for循环
- p.item!=null,p.casItem设置item为null
- p!=head,此时队列中有了2个item为 NULL 的节点(head节点和p节点),移动head指针,对其执行updateHead 操作。
因为head/tail 并不是精确地指向队列头部和尾部,所以不能简单地通过比较 head/tail 指针来判断 队列是否为空,而是需要从head指针开始遍历,找第1个不为NULL的节点。如果找到,则队列不为空; 如果找不到,则队列为空。
public boolean isEmpty() {
return first() == null;
}
Node<E> first() {
restartFromHead:
for (;;) {
for (Node<E> h = head, p = h, q;;) {
boolean hasItem = (p.item != null);
if (hasItem || (q = p.next) == null) {
updateHead(h, p);
return hasItem ? p : null;
}
else if (p == q)
continue restartFromHead;
else
p = q;
}
}
}
|