读《Java并发编程实战》笔记。
import java.util.concurrent.atomic.AtomicReference;
/**
* 使用 Treiber 算法(Treiber,1986)构造的非阻塞栈
* <p>
*
* @author hyl
* @version v1.0: ConcurrentStack.java, v 0.1 2020/10/22 17:18 $
*/
public class ConcurrentStack<E> {
AtomicReference<Node<E>> top = new AtomicReference<>();
public void push(E item) {
Node<E> newHead = new Node<>(item);
Node<E> oldHead;
do {
oldHead = top.get();
newHead.next = oldHead;
} while (!top.compareAndSet(oldHead, newHead));
}
public E pop() {
Node<E> oldHead;
Node<E> newHead;
do {
oldHead = top.get();
if (oldHead == null)
return null;
newHead = oldHead.next;
} while (!top.compareAndSet(oldHead, newHead));
return oldHead.item;
}
private static class Node<E> {
public final E item;
public Node<E> next;
public Node(E item) {
this.item = item;
}
}
}
说明:
ConcurrentStack 中给出了如何通过原子引用来构建栈的示例。
- 栈是由Node元素构成的一个链表,其中栈顶作为根节点,并且在每个元素中都包含了一个值以及指向下一个元素的连接。
- push 方法创建一个新的节点,该节点的 next 域指向当前的栈顶,然后使用 CAS 把这个新节点放入栈顶。如果在开始插入节点时,位于栈顶的节点没有发生变化,那么 CAS 就会成功,如果栈顶节点发生了变化,那么 CAS 就会失败,而 push 方法会根据栈的当前状态来更新节点,并且再次尝试。
- 无论那种情况,在 CAS 执行完成后,栈仍会处于一致的状态。