在jdk8中,集合类 ConcurrentHashMap 经 Doug Lea 大师之手,借助volatile语义以及CAS操作进行优化,使得该集合类更好地发挥出了并发的优势。与jdk7中相比,在原有段锁(Segment)的基础上,引入了数组+链表+红黑树的存储模型,在查询效率上花费了不少心思。
ConcurrentHashMap内存存储结构图大致如下:
/**
* Initializes table, using the size recorded in sizeCtl.
*/
private final Node<K,V>[] initTable() {
Node<K,V>[] tab; int sc;
while ((tab = table) == null || tab.length == 0) {
if ((sc = sizeCtl) < 0)
Thread.yield(); // lost initialization race; just spin
else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {
try {
//此处为什么要再次判断
if ((tab = table) == null || tab.length == 0) {
int n = (sc > 0) ? sc : DEFAULT_CAPACITY;
@SuppressWarnings("unchecked")
Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n];
table = tab = nt;
sc = n - (n >>> 2);
}
} finally {
sizeCtl = sc;
}
break;
}
}
return tab;
}
Q: 为什么在else代码中要重复判断table是否为空?
A: 在多线程中,有可能多个线程进入到else代码块的。
举个例子:
假如线程a,b同时 执行 initTable方法:
此时sizeCtl为0,a,b同时执行U.compareAndSwapInt(this,SIZECTL, sc, -1 ),但只有一个线程判断为true进入到else中去。假如线程a成功,此时原子操作将sizeCtl设置为-1,线程b则重新执行while循环(此时table未初始化),此时sizeCtl为-1,则线程b让出cpu。
继续来看线程a ,在执行到finally{ sizeCtl = sc } 代码前,sizeCtl一直为-1,线程b只能永远执行Thread.yield() 让出cpu。
但在线程a执行sizeCtl = sc 之后,线程b可能会进入运行态 此时src = sizeCtl (等于 n - (n >>> 2));
所以线程b会重新执行U.compareAndSwapInt(this, SIZECTL, sc, -1 )判断, 线程b将 sizeCtl 设置为-1. 此时判断table != null。执行finally代码,将sizeCtl重新置为sc。(如果此时不判断table是否为null,则会再初始化一个table)
get方法思想:
- 计算出该节点的hash值,并算出该node在table中的位置 e = tabAt(tab, (n - 1) & h)。
- 判断首节点是不是要找的节点,是,直接返回。
- 如果该节点hash值小于0 则可能是forwadingNode也可能是Treebin节点,直接调用forwadingNode.find或者Treebin节点的find方法。
- 如果该节点hash值大于等于0,则是普通节点,通过遍历链表查到要找的节点。
public V get(Object key) {
Node<K,V>[] tab; Node<K,V> e, p; int n, eh; K ek;
//计算出该节点的hash值
int h = spread(key.hashCode());
if ((tab = table) != null && (n = tab.length) > 0 &&
(e = tabAt(tab, (n - 1) & h)) != null) {
//判断首节点是不是要找的节点
if ((eh = e.hash) == h) {
if ((ek = e.key) == key || (ek != null && key.equals(ek)))
return e.val;
}
//forwadingNode或者Treebin
else if (eh < 0)
return (p = e.find(h, key)) != null ? p.val : null;
//普通节点,通过遍历链表查到要找的节点
while ((e = e.next) != null) {
if (e.hash == h &&
((ek = e.key) == key || (ek != null && key.equals(ek))))
return e.val;
}
}
return null;
}
参考:https://bentang.me/tech/2016/12/01/jdk8-concurrenthashmap-1/