java面试题：ConcurrentHashMap的扩容机制

一. 扩容

• jdk8中，采用多线程扩容。整个扩容过程，通过CAS设置sizeCtl，transferIndex等变量协调多个线程进行并发扩容。
• 多线程无锁扩容的关键就是通过CAS设置sizeCtl与transferIndex变量，协调多个线程对table数组中的node进行迁移。

二. 何时扩容触发

当往hashMap中成功插入一个key/value节点时，有可能触发扩容动作：
1、如果新增节点之后，所在链表的元素个数达到了阈值 8，则会调用treeifyBin方法把链表转换成红黑树，不过在结构转换之前，会对数组长度进行判断，实现如下：

• 如果数组长度n小于阈值MIN_TREEIFY_CAPACITY，默认是64，则会调用tryPresize方法把数组长度扩大到原来的两倍，并触发transfer方法，重新调整节点的位置。

2、新增节点之后，会调用addCount方法记录元素个数，并检查是否需要进行扩容，当数组元素个数达到阈值时，会触发transfer方法，重新调整节点的位置。

3、当前线程发现此时map正在扩容，则协助扩容

什么时候线程不协助扩容：

如果准备加入扩容的线程，发现以下情况，放弃扩容，直接返回。

a、发现transferIndex=0,即所有node均已分配
b、发现扩容线程已经达到最大扩容线程数

如何保证线程安全

采用了 CAS + synchronized 来保证并发安全性。

put方法的逻辑

• 计算key的hash值
• 如果当前table还没有初始化先调用initTable方法将tab进行初始化
• tab中索引为i的位置的元素为null，则直接使用CAS将值插入即可。添加失败就进入下次循环。
• 判断当前table是否正在扩容，是的话协助扩容
• 当前为红黑树，将新的键值对插入到红黑树中（用synchronized锁住）
• 当前是链表，插入完键值对后再根据实际大小看是否需要转换成红黑树（用synchronized锁住）
• 对当前容量大小进行检查，如果超过了临界值（实际大小*加载因子）就需要扩容

put的过程很清晰，对当前的table进行无条件自循环直到put成功，可以分成以下六步流程来概述：

1、判断Node[]数组是否初始化，没有则进行初始化操作
2、通过hash定位数组的索引坐标，是否有Node节点，如果没有则使用CAS进行添加（链表的头节点），添加失败则进入下次循环。
3、检查到内部正在扩容，就帮助它一块扩容。
4、如果f！=null，则使用synchronized锁住f元素（链表/红黑树的头元素）。如果是Node（链表结构）则执行链表的添加操作；如果是TreeNode（树型结构）则执行树添加操作。
5、判断链表长度已经达到临界值8（默认值），当节点超过这个值就需要把链表转换为树结构。
6、如果添加成功就调用addCount（）方法统计size，并且检查是否需要扩容

相关变量介绍

nextTable

• 扩容期间，将table数组中的元素 迁移到 nextTable。即nextTable为创建的新的table，容量为原来的两倍

sizeCtl

• 多线程之间，以volatile的方式读取sizeCtl属性，来判断ConcurrentHashMap当前所处的状态。
• 通过cas设置sizeCtl属性，告知其他线程ConcurrentHashMap的状态变更。

不同状态，sizeCtl所代表的含义也有所不同

• 未初始化：sizeCtl=0：表示没有指定初始容量。
• sizeCtl>0：表示初始容量。
• 初始化中：sizeCtl=-1,标记作用，告知其他线程，正在初始化
• 正常状态：sizeCtl=0.75n,扩容阈值
• 扩容中：sizeCtl < 0 : 表示有其他线程正在执行扩容
• sizeCtl = (resizeStamp(n) << RESIZE_STAMP_SHIFT)+2 表示此时只有一个线程在执行扩容

transferIndex 扩容索引

扩容索引，表示已经分配给扩容线程的table数组索引位置。主要用来协调多个线程，并发安全地获取迁移任务（hash桶）。

private transient volatile int transferIndex;
private static final int MIN_TRANSFER_STRIDE = 16; //扩容线程每次最少要迁移16个hash桶

1、在扩容之前，transferIndex 在数组的最右边 。此时有一个线程发现已经到达扩容阈值，准备开始扩容。
2、扩容线程，在迁移数据之前，首先要将transferIndex右移（以CAS的方式修改 transferIndex=transferIndex-stride(要迁移hash桶的个数)），获取迁移任务。每个扩容线程都会通过for循环+CAS的方式设置transferIndex，因此可以确保多线程扩容的并发安全。

换个角度，我们可以将待迁移的table数组，看成一个任务队列，transferIndex看成任务队列的头指针。而扩容线程，就是这个队列的消费者。扩容线程通过CAS设置transferIndex索引的过程，就是消费者从任务队列中获取任务的过程。为了性能考虑，我们当然不会每次只获取一个任务（hash桶），因此ConcurrentHashMap规定，每次至少要获取16个迁移任务（迁移16个hash桶，MIN_TRANSFER_STRIDE = 16）

也就是说transferIndex其实代表当前剩下需要进行扩容的桶的个数。如果transferIndex == 0 那就代表当前没有桶需要进行扩容了。每次一个线程参加扩容transferIndex就需要减MIN_TRANSFER_STRIDE。

ForwardingNode节点

1、标记作用，表示其他线程正在扩容，并且此节点已经扩容完毕
2、关联了nextTable，扩容期间可以通过find方法，访问已经迁移到了nextTable中的数据

扩容过程分析

1、线程执行put操作，发现容量已经达到扩容阈值，需要进行扩容操作，例如此时transferindex=tab.length=32
2、扩容线程A 以CAS的方式修改transferindex=31-16=16 ,然后按照降序迁移table[31]至table[16]这个区间的hash桶(逆序遍历扩容)
3、迁移hash桶时，会将桶内的链表或者红黑树，按照一定算法，拆分成2份，将其插入nextTable[i]和nextTable[i+n]（n是table数组的长度）。 迁移完毕的hash桶,会被设置成ForwardingNode节点，以此告知访问此桶的其他线程，此节点已经迁移完毕。
4、此时线程2访问到了ForwardingNode节点，如果线程2执行的put或remove等写操作，那么就会先帮其扩容。如果线程2执行的是get等读方法，则会调用ForwardingNode的find方法，去nextTable里面查找相关元素