针对扩容和put的原理,这个问题,我在学习的时候,一直有一个疑问点:假如当前A线程已经对15这个位置扩完容了,那如果这时候B线程又对15这个位置插入了元素,要怎么处理?
通过对代码的仔细查看学习,找到了这个问题的答案
在扩容的时候,假如线程A对i这个位置扩容完成了,那线程A会把原数组中i位置,放入一个ForwardingNode对象,放入的这个对象很有讲究,也起到了很大的作用,这个和put时的一些逻辑处理有关系
扩容的思想
先说扩容的思想
在concurrentHashMap中,如果当前线程A触发了扩容的机制,线程A会开始进行扩容,在扩容的时候,涉及到一个步伐的概念,简单来说,就是加入了辅助扩容的机制,所以,此时一个线程会负责一部分数据的扩容,举个简单例子来说
1.假如当前数组中有32个位置,那线程A负责31到16这区间的元素的扩容
2.如果此时线程B来put元素,发现要put到30这个位置,发现正在被扩容,那此时就会去辅助扩容,所谓的辅助扩容,就是线程B负责15到0这个区间的扩容
在源码中,会计算一个线程所负责扩容的大小,是根据CPU核数来计算的,最小是16个,也就是说,一个线程最少负责16个位置元素的扩容
java.util.concurrent.ConcurrentHashMap#transfer
/**
* 1.这里的stride是步长,这个步长最小值是16
*/
if ((stride = (NCPU > 1) ? (n >>> 3) / NCPU : n) < MIN_TRANSFER_STRIDE)
stride = MIN_TRANSFER_STRIDE; // subdivide range
这里是计算步长的公式,NCPU是当前CPU的核数,n是当前老数组的长度,会根据这个公式去计算
1.如果cpu核数大于1,那就(n >>> 3) / NCPU 以这个值作为步长
2.如果CPU核数小于等于1,那就以数组长度作为步长
3.如果这样计算得到的步长依旧小于16,那就把16作为步长
也就是说,如果CPU核数只有1个,并且当前数组长度大于16,那就以数组长度作为步长;如果CPU核数有多个,就会根据一个公式,计算一个阈值,如果这个值小于16,就以16作为步长
/**
* 2.对新的table进行初始化和赋值
* 默认扩容之后的table长度是原table的一倍
* nextTab就是扩容之后的数组
*/
if (nextTab == null) {
// initiating
try {
@SuppressWarnings("unchecked")
Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n << 1];
nextTab = nt;
} catch (Throwable ex) {
// try to cope with OOME
sizeCtl = Integer.MAX_VALUE;
return;
}
nextTable = nextTab;
/**