Java集合源码解析-ConcurrentHashMap（JDK8），java从入门到精通pdf脚本之家

}

只允许一个线程对表进行初始化，若不巧有其他线程进来了，则会让其他线程交出 CPU 等待下次系统调度。这保证了表同时只会被一个线程初始化。

//检测到桶结点是 ForwardingNode 类型，协助扩容

else if ((fh = f.hash) == MOVED)

tab = helpTransfer(tab, f);

//桶结点是普通的结点，锁住该桶头结点并试图在该链表的尾部添加一个节点

else {

V oldVal = null;

synchronized (f) {

if (tabAt(tab, i) == f) {

//向普通的链表中添加元素

if (fh >= 0) {

binCount = 1;

for (Node<K,V> e = f;; ++binCount) {

K ek;

if (e.hash == hash &&((ek = e.key) == key ||(ek != null && key.equals(ek)))) {

oldVal = e.val;

if (!onlyIfAbsent)

e.val = value;

break;

}

Node<K,V> pred = e;

if ((e = e.next) == null) {

pred.next = new Node<K,V>(hash, key,value, null);

break;

}

}

}

//向红黑树中添加元素，TreeBin 结点的hash值为TREEBIN（-2）

else if (f instanceof TreeBin) {

Node<K,V> p;

binCount = 2;

if ((p = ((TreeBin<K,V>)f).putTreeVal(hash, key, value)) != null) {

oldVal = p.val;

if (!onlyIfAbsent)

p.val = value;

}

}

}

}

//binCount != 0 说明向链表或者红黑树中添加或修改一个节点成功

//binCount == 0 说明 put 操作将一个新节点添加成为某个桶的首节点

if (binCount != 0) {

//链表深度超过 8 转换为红黑树

if (binCount >= TREEIFY_THRESHOLD)

treeifyBin(tab, i);

//oldVal != null 说明此次操作是修改操作

//直接返回旧值即可，无需做下面的扩容边界检查

if (oldVal != null)

return oldVal;

break;

}

}

}

//CAS 式更新baseCount，并判断是否需要扩容

addCount(1L, binCount);

//程序走到这一步说明此次 put 操作是一个添加操作，否则早就 return 返回了

return null;

注意ForwardingNode 节点类型

这个节点内部保存了一个 nextTable 引用，它指向一张 hash 表;

在扩容操作中，我们需要对每个桶中的结点进行分离和转移;

如果某个桶结点中所有节点都已经迁移完成了(已经被转移到新表 nextTable);

那么会在原 table 表的该位置挂上一个 ForwardingNode 结点，说明此桶已经完成迁移.

ForwardingNode继承自 Node 结点,并且它唯一的构造函数将构建一个k/v/next 都为 null 的结点,反正它就是个标识，无需那些属性;

但是 hash 值却为 MOVED.

所以，我们在 putVal 方法中遍历整个 hash 表的桶结点,如果遇到 hash 值等于 MOVED,说明已经有线程正在扩容 rehash 操作,整体上还未完成，,过我们要插入的桶的位置已经完成了所有节点的迁移

由于检测到当前哈希表正在扩容,于是让当前线程去协助扩容.

final Node<K,V>[] helpTransfer(Node<K,V>[] tab, Node<K,V> f) {

Node<K,V>[] nextTab; int sc;

if (tab != null && (f instanceof ForwardingNode) &&

(nextTab = ((ForwardingNode<K,V>)f).nextTable) != null) {

//返回一个 16 位长度的扩容校验标识

int rs = resizeStamp(tab.length);

while (nextTab