深度分析ConcurrentHashMap原理分析

ConcurrentHashMap的初步使用及场景

CHM的使用

ConcurrentHashMap是J.U.C包里面提供的一个线程安全并且高效的HashMap,所以ConcurrentHashMap在并发编程的场景中使用的频率比较高,那么这一节课我们就从ConcurrentHashMap的使用上以及源码层面来分析ConcurrentHashMap到底是如何实现安全性的

api使用

ConcurrentHashMap是Map的派生类,所以api基本和Hashmap是类似,主要就是put、get这些方法,接下来基于ConcurrentHashMap的put和get这两个方法作为切入点来分析ConcurrentHashMap的源码实现

ConcurrentHashMap的源码分析

先要做一个说明,这节课分析的ConcurrentHashMap是基于Jdk1.8的版本。 JDK1.7和Jdk1.8版本的变化 ConcurrentHashMap和HashMap的实现原理是差不多的,但是因为ConcurrentHashMap需要支持并发操作,所以在实现上要比hashmap稍微复杂一些。 在JDK1.7的实现上,ConrruentHashMap由一个个Segment组成,简单来说,ConcurrentHashMap是一个Segment数组,它通过继承ReentrantLock来进行加锁,通过每次锁住一个segment来保证每个segment内的操作的线程安全性从而实现全局线程安全。整个结构图如下

深度分析ConcurrentHashMap原理分析_第1张图片

当每个操作分布在不同的segment上的时候,默认情况下,理论上可以同时支持16个线程的并发写入。 相比于1.7版本,它做了两个改进

  1. 取消了segment分段设计,直接使用Node数组来保存数据,并且采用Node数组元素作为锁来实现每一行数据进行加锁来进一步减少并发冲突的概率

  2. 将原本数组+单向链表的数据结构变更为了数组+单向链表+红黑树的结构。为什么要引入红黑树呢?在正常情况下,key hash之后如果能够很均匀的分散在数组中,那么table数组中的每个队列的长度主要为0或者1.但是实际情况下,还是会存在一些队列长度过长的情况。如果还采用单向列表方式,那么查询某个节点的时间复杂度就变为O(n); 因此对于队列长度超过8的列表,JDK1.8采用了红黑树的结构,那么查询的时间复杂度就会降低到O(logN),可以提升查找的性能; 

    深度分析ConcurrentHashMap原理分析_第2张图片

这个结构和JDK1.8版本中的Hashmap的实现结构基本一致,但是为了保证线程安全性,ConcurrentHashMap的实现会稍微复杂一下。接下来我们从源码层面来了解一下它的原理. 我们基于put和get方法来分析它的实现即可

put方法第一阶段

public V put(K key, V value) {	
    return putVal(key, value, false);	
}
/** Implementation for put and putIfAbsent */	
final V putVal(K key, V value, boolean onlyIfAbsent) {	
    if (key == null || value == null) throw new NullPointerException();	
    int hash = spread(key.hashCode()); //计算hash值	
    int binCount = 0; //用来记录链表的长度	
    for (Node[] tab = table;;) { //这里其实就是自旋操作,当出现线程竞争时不断自旋	
        Node f; int n, i, fh;	
        if (tab == null || (n = tab.length) == 0)//如果数组为空,则进行数组初始化	
            tab = initTable(); //初始化数组	
//通过hash值对应的数组下标得到第一个节点; 以volatile读的方式来读取table数组中的元素,保证每次拿到的数据都是最新的	
        else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) { 	
//如果该下标返回的节点为空,则直接通过cas将新的值封装成node插入即可;如果cas失败,说明存在竞争,则进入下一次循环	
            if (casTabAt(tab, i, null,	
                         new Node(hash, key, value, null)))	
                break;                   // no lock when adding to empty bin	
        }	
//...	
}

假如在上面这段代码中存在两个线程,在不加锁的情况下:线程A成功执行casTabAt操作后,随后的线程B可以通过tabAt方法立刻看到table[i]的改变。原因如下:线程A的casTabAt操作,具有volatile读写相同的内存语义,根据volatile的happens-before规则:线程A的casTabAt操作,一定对线程B的tabAt操作可见

initTable

数组初始化方法,这个方法比较简单,就是初始化一个合适大小的数组 sizeCtl这个要单独说一下,如果没搞懂这个属性的意义,可能会被搞晕 这个标志是在Node数组初始化或者扩容的时候的一个控制位标识,负数代表正在进行初始化或者扩容操作 -1 代表正在初始化 -N 代表有N-1有二个线程正在进行扩容操作,这里不是简单的理解成n个线程,sizeCtl就是-N,这块后续在讲扩容的时候会说明 0标识Node数组还没有被初始化,正数代表初始化或者下一次扩容的大小

private final Node[] initTable() {	
    Node[] tab; int sc;	
    while ((tab = table) == null || tab.length == 0) {	
        if ((sc = sizeCtl) < 0)//被其他线程抢占了初始化的操作,则直接让出自己的CPU时间片	
            Thread.yield(); // lost initialization race; just spin	
//通过cas操作,将sizeCtl替换为-1,标识当前线程抢占到了初始化资格	
        else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {	
            try {	
                if ((tab = table) == null || tab.length == 0) {	
                    int n = (sc > 0) ? sc : DEFAULT_CAPACITY;//默认初始容量为16	
                    @SuppressWarnings("unchecked")	
//初始化数组,长度为16,或者初始化在构造ConcurrentHashMap的时候传入的长度	
                    Node[] nt = (Node[])new Node[n];	
                    table = tab = nt;//将这个数组赋值给table	
                    sc = n - (n >>> 2); //计算下次扩容的大小,实际就是当前容量的0.75倍,这里使用了右移来计算	
                }	
            } finally {	
                sizeCtl = sc; //设置sizeCtl为sc, 如果默认是16的话,那么这个时候sc=16*0.75=12	
            }	
            break;	
        }	
    }	
    return tab;	
}

tabAt

该方法获取对象中offset偏移地址对应的对象field的值。实际上这段代码的含义等价于tab[i], 但是为什么不直接使用tab[i]来计算呢? getObjectVolatile,一旦看到volatile关键字,就表示可见性。因为对volatile写操作happen-before于volatile读操作,因此其他线程对table的修改均对get读取可见; 虽然table数组本身是增加了volatile属性,但是“volatile的数组只针对数组的引用具有volatile的语义,而不是它的元素”。 所以如果有其他线程对这个数组的元素进行写操作,那么当前线程来读的时候不一定能读到最新的值。 出于性能考虑,Doug Lea直接通过Unsafe类来对table进行操作。

static final  Node tabAt(Node[] tab, int i) {	
    return (Node)U.getObjectVolatile(tab, ((long)i << ASHIFT) + ABASE);	
}

图解分析

深度分析ConcurrentHashMap原理分析_第3张图片

put方法第二阶段

在putVal方法执行完成以后,会通过addCount来增加ConcurrentHashMap中的元素个数,并且还会可能触发扩容操作。这里会有两个非常经典的设计

  1. 高并发下的扩容

  2. 如何保证addCount的数据安全性以及性能

//将当前ConcurrentHashMap的元素数量加1,有可能触发transfer操作(扩容)	
    addCount(1L, binCount);	
    return null;	
}

addCount

在putVal最后调用addCount的时候,传递了两个参数,分别是1和binCount(链表长度),看看addCount方法里面做了什么操作 x表示这次需要在表中增加的元素个数,check参数表示是否需要进行扩容检查,大于等于0都需要进行检查

private final void addCount(long x, int check) {	
    CounterCell[] as; long b, s;	
判断counterCells是否为空,	
1.    如果为空,就通过cas操作尝试修改baseCount变量,对这个变量进行原子累加操作(做这个操作的意义是:如果在没有竞争的情况下,仍然采用baseCount来记录元素个数)	
2.    如果cas失败说明存在竞争,这个时候不能再采用baseCount来累加,而是通过CounterCell来记录 	
if ((as = counterCells) != null ||	
        !U.compareAndSwapLong(this, BASECOUNT, b = baseCount, s = b + x)) {	
        CounterCell a; long v; int m;	
        boolean uncontended = true;//是否冲突标识,默认为没有冲突	
这里有几个判断	
1. 计数表为空则直接调用fullAddCount	
2. 从计数表中随机取出一个数组的位置为空,直接调用fullAddCount	
3. 通过CAS修改CounterCell随机位置的值,如果修改失败说明出现并发情况(这里又用到了一种巧妙的方法),调用fullAndCount	
Random在线程并发的时候会有性能问题以及可能会产生相同的随机数,ThreadLocalRandom.getProbe可以解决这个问题,并且性能要比Random高	
        if (as == null || (m = as.length - 1) < 0 ||	
            (a = as[ThreadLocalRandom.getProbe() & m]) == null ||	
            !(uncontended =	
              U.compareAndSwapLong(a, CELLVALUE, v = a.value, v + x))) {	
            fullAddCount(x, uncontended);//执行fullAddCount方法	
            return;	
        }	
        if (check <= 1)//链表长度小于等于1,不需要考虑扩容	
            return;	
        s = sumCount();//统计ConcurrentHashMap元素个数	
    }	
    //….	
}

CounterCells解释

ConcurrentHashMap是采用CounterCell数组来记录元素个数的,像一般的集合记录集合大小,直接定义一个size的成员变量即可,当出现改变的时候只要更新这个变量就行。为什么ConcurrentHashMap要用这种形式来处理呢? 问题还是处在并发上,ConcurrentHashMap是并发集合,如果用一个成员变量来统计元素个数的话,为了保证并发情况下共享变量的的难全兴,势必会需要通过加锁或者自旋来实现,如果竞争比较激烈的情况下,size的设置上会出现比较大的冲突反而影响了性能,所以在ConcurrentHashMap采用了分片的方法来记录大小,具体什么意思,我们来分析下

private transient volatile int cellsBusy;// 标识当前cell数组是否在初始化或扩容中的CAS标志位	
/**	
 * Table of counter cells. When non-null, size is a power of 2.	
 */	
private transient volatile CounterCell[] counterCells;// counterCells数组,总数值的分值分别存在每个cell中	
@sun.misc.Contended static final class CounterCell {	
    volatile long value;	
    CounterCell(long x) { value = x; }	
}	
//看到这段代码就能够明白了,CounterCell数组的每个元素,都存储一个元素个数,而实际我们调用size方法就是通过这个循环累加来得到的	
//又是一个设计精华,大家可以借鉴; 有了这个前提,再会过去看addCount这个方法,就容易理解一些了	
final long sumCount() {	
    CounterCell[] as = counterCells; CounterCell a;	
    long sum = baseCount;	
    if (as != null) {	
        for (int i = 0; i < as.length; ++i) {	
            if ((a = as[i]) != null)	
                sum += a.value;	
        }	
    }	
    return sum;	
}

fullAddCount源码分析

fullAddCount主要是用来初始化CounterCell,来记录元素个数,里面包含扩容,初始化等操作

private final void fullAddCount(long x, boolean wasUncontended) {	
    int h;	
//获取当前线程的probe的值,如果值为0,则初始化当前线程的probe的值,probe就是随机数	
    if ((h = ThreadLocalRandom.getProbe()) == 0) {	
        ThreadLocalRandom.localInit();      // force initialization	
        h = ThreadLocalRandom.getProbe();	
        wasUncontended = true; // 由于重新生成了probe,未冲突标志位设置为true	
    }	
    boolean collide = false;                // True if last slot nonempty	
    for (;;) {//自旋	
        CounterCell[] as; CounterCell a; int n; long v;	
//说明counterCells已经被初始化过了,我们先跳过这个代码,先看初始化部分	
        if ((as = counterCells) != null && (n = as.length) > 0) { 	
            if ((a = as[(n - 1) & h]) == null) {// 通过该值与当前线程probe求与,获得cells的下标元素,和hash 表获取索引是一样的	
                if (cellsBusy == 0) { //cellsBusy=0表示counterCells不在初始化或者扩容状态下     	
                    CounterCell r = new CounterCell(x); //构造一个CounterCell的值,传入元素个数	
                    if (cellsBusy == 0 &&//通过cas设置cellsBusy标识,防止其他线程来对counterCells并发处理	
                        U.compareAndSwapInt(this, CELLSBUSY, 0, 1)) {	
                        boolean created = false;	
                        try {               // Recheck under lock	
                            CounterCell[] rs; int m, j;	
//将初始化的r对象的元素个数放在对应下标的位置	
                            if ((rs = counterCells) != null &&	
                                (m = rs.length) > 0 &&	
                                rs[j = (m - 1) & h] == null) {	
                                rs[j] = r;	
                                created = true;	
                            }	
                        } finally {//恢复标志位	
                            cellsBusy = 0;	
                        }	
                        if (created)//创建成功,退出循环	
                            break;	
                        continue;//说明指定cells下标位置的数据不为空,则进行下一次循环	
                    }	
                }	
                collide = false;	
            }	
//说明在addCount方法中cas失败了,并且获取probe的值不为空	
            else if (!wasUncontended)       // CAS already known to fail	
                wasUncontended = true;      //设置为未冲突标识,进入下一次自旋	
//由于指定下标位置的cell值不为空,则直接通过cas进行原子累加,如果成功,则直接退出	
            else if (U.compareAndSwapLong(a, CELLVALUE, v = a.value, v + x))//	
                break;	
//如果已经有其他线程建立了新的counterCells或者CounterCells大于CPU核心数(很巧妙,线程的并发数不会超过cpu核心数)	
            else if (counterCells != as || n >= NCPU)	
                collide = false; //设置当前线程的循环失败不进行扩容	
            else if (!collide)//恢复collide状态,标识下次循环会进行扩容	
                collide = true;	
//进入这个步骤,说明CounterCell数组容量不够,线程竞争较大,所以先设置一个标识表示为正在扩容	
            else if (cellsBusy == 0 &&	
                     U.compareAndSwapInt(this, CELLSBUSY, 0, 1)) {	
                try {	
                    if (counterCells == as) {// Expand table unless stale	
//扩容一倍 2变成4,这个扩容比较简单	
                        CounterCell[] rs = new CounterCell[n << 1];	
                        for (int i = 0; i < n; ++i)	
                            rs[i] = as[i];	
                        counterCells = rs;	
                    }	
                } finally {	
                    cellsBusy = 0;//恢复标识	
                }	
                collide = false;	
                continue;//继续下一次自旋	
            }	
            h = ThreadLocalRandom.advanceProbe(h);//更新随机数的值	
        }

初始化CounterCells数组

//cellsBusy=0表示没有在做初始化,通过cas更新cellsbusy的值标注当前线程正在做初始化操作	
        else if (cellsBusy == 0 && counterCells == as &&	
                 U.compareAndSwapInt(this, CELLSBUSY, 0, 1)) {	
            boolean init = false;	
            try {                           // Initialize table	
                if (counterCells == as) {	
                    CounterCell[] rs = new CounterCell[2]; //初始化容量为2	
                    rs[h & 1] = new CounterCell(x);//将x也就是元素的个数放在指定的数组下标位置	
                    counterCells = rs;//赋值给counterCells	
                    init = true;//设置初始化完成标识	
                }	
            } finally {	
                cellsBusy = 0;//恢复标识	
            }	
            if (init)	
                break;	
        }	
//竞争激烈,其它线程占据cell 数组,直接累加在base变量中	
        else if (U.compareAndSwapLong(this, BASECOUNT, v = baseCount, v + x))	
            break;                          // Fall back on using base	
    }	
}

CounterCells初始化图解

初始化长度为2的数组,然后随机得到指定的一个数组下标,将需要新增的值加入到对应下标位置处

深度分析ConcurrentHashMap原理分析_第4张图片

transfer扩容阶段

判断是否需要扩容,也就是当更新后的键值对总数baseCount >= 阈值sizeCtl时,进行rehash,这里面会有两个逻辑。

  1. 如果当前正在处于扩容阶段,则当前线程会加入并且协助扩容

  2. 如果当前没有在扩容,则直接触发扩容操作

if (check >= 0) {//如果binCount>=0,标识需要检查扩容	
        Node[] tab, nt; int n, sc;	
//s标识集合大小,如果集合大小大于或等于扩容阈值(默认值的0.75)	
//并且table不为空并且table的长度小于最大容量	
        while (s >= (long)(sc = sizeCtl) && (tab = table) != null &&	
               (n = tab.length) < MAXIMUM_CAPACITY) {	
            int rs = resizeStamp(n);//这里是生成一个唯一的扩容戳,这个是干嘛用的呢?且听我慢慢分析	
            if (sc < 0) {//sc<0,也就是sizeCtl<0,说明已经有别的线程正在扩容了	
//这5个条件只要有一个条件为true,说明当前线程不能帮助进行此次的扩容,直接跳出循环	
//sc >>> RESIZE_STAMP_SHIFT!=rs 表示比较高RESIZE_STAMP_BITS位生成戳和rs是否相等,相同	
//sc=rs+1 表示扩容结束	
//sc==rs+MAX_RESIZERS 表示帮助线程线程已经达到最大值了	
//nt=nextTable -> 表示扩容已经结束	
//transferIndex<=0 表示所有的transfer任务都被领取完了,没有剩余的hash桶来给自己自己好这个线程来做transfer	
                if ((sc >>> RESIZE_STAMP_SHIFT) != rs || sc == rs + 1 ||	
                    sc == rs + MAX_RESIZERS || (nt = nextTable) == null ||	
                    transferIndex <= 0)	
                    break;	
                if (U.compareAndSwapInt(this, SIZECTL, sc, sc + 1))//当前线程尝试帮助此次扩容,如果成功,则调用transfer	
                    transfer(tab, nt);	
            }	
// 如果当前没有在扩容,那么rs肯定是一个正数,通过rs<

resizeStamp

这块逻辑要理解起来,也有一点复杂。 resizeStamp用来生成一个和扩容有关的扩容戳,具体有什么作用呢?我们基于它的实现来做一个分析

static final int resizeStamp(int n) {	
    return Integer.numberOfLeadingZeros(n) | (1 << (RESIZE_STAMP_BITS - 1));	
}

Integer.numberOfLeadingZeros 这个方法是返回无符号整数n最高位非0位前面的0的个数	
比如10的二进制是 0000 0000 0000 0000 0000 0000 0000 1010	
那么这个方法返回的值就是28	
根据resizeStamp的运算逻辑,我们来推演一下,假如n=16,那么resizeStamp(16)=32796转化为二进制是	
[0000 0000 0000 0000 1000 0000 0001 1100]	
接着再来看,当第一个线程尝试进行扩容的时候,会执行下面这段代码	
U.compareAndSwapInt(this, SIZECTL, sc, (rs << RESIZE_STAMP_SHIFT) + 2)	
rs左移16位,相当于原本的二进制低位变成了高位1000 0000 0001 1100 0000 0000 0000 0000	
然后再+2 =1000 0000 0001 1100 0000 0000 0000 0000+10=1000 0000 0001 1100 0000 0000 0000 0010	
高16位代表扩容的标记、低16位代表并行扩容的线程数	
	
    这样来存储有什么好处呢?	
1.    首先在CHM中是支持并发扩容的,也就是说如果当前的数组需要进行扩容操作,可以由多个线程来共同负责,这块后续会单独讲	
2.    可以保证每次扩容都生成唯一的生成戳,每次新的扩容,都有一个不同的n,这个生成戳就是根据n来计算出来的一个数字,n不同,这个数字也不同	
    第一个线程尝试扩容的时候,为什么是+2	
因为1表示初始化,2表示一个线程在执行扩容,而且对sizeCtl的操作都是基于位运算的,所以不会关心它本身的数值是多少,只关心它在二进制上的数值,而sc + 1会在	
低16位上加1。

transfer

扩容是ConcurrentHashMap的精华之一,扩容操作的核心在于数据的转移,在单线程环境下数据的转移很简单,无非就是把旧数组中的数据迁移到新的数组。但是这在多线程环境下,在扩容的时候其他线程也可能正在添加元素,这时又触发了扩容怎么办?可能大家想到的第一个解决方案是加互斥锁,把转移过程锁住,虽然是可行的解决方案,但是会带来较大的性能开销。因为互斥锁会导致所有访问临界区的线程陷入到阻塞状态,持有锁的线程耗时越长,其他竞争线程就会一直被阻塞,导致吞吐量较低。而且还可能导致死锁。 而ConcurrentHashMap并没有直接加锁,而是采用CAS实现无锁的并发同步策略,最精华的部分是它可以利用多线程来进行协同扩容 简单来说,它把Node数组当作多个线程之间共享的任务队列,然后通过维护一个指针来划分每个线程锁负责的区间,每个线程通过区间逆向遍历来实现扩容,一个已经迁移完的bucket会被替换为一个ForwardingNode节点,标记当前bucket已经被其他线程迁移完了。接下来分析一下它的源码实现 1、fwd:这个类是个标识类,用于指向新表用的,其他线程遇到这个类会主动跳过这个类,因为这个类要么就是扩容迁移正在进行,要么就是已经完成扩容迁移,也就是这个类要保证线程安全,再进行操作。 2、advance:这个变量是用于提示代码是否进行推进处理,也就是当前桶处理完,处理下一个桶的标识 3、finishing:这个变量用于提示扩容是否结束用的

private final void transfer(Node[] tab, Node[] nextTab) {	
    int n = tab.length, stride;	
//将 (n>>>3相当于 n/8) 然后除以 CPU核心数。如果得到的结果小于 16,那么就使用 16	
// 这里的目的是让每个 CPU 处理的桶一样多,避免出现转移任务不均匀的现象,如果桶较少的话,默认一个 CPU(一个线程)处理 16 个桶,也就是长度为16的时候,扩容的时候只会有一个线程来扩容	
    if ((stride = (NCPU > 1) ? (n >>> 3) / NCPU : n) < MIN_TRANSFER_STRIDE)	
        stride = MIN_TRANSFER_STRIDE; // subdivide range	
//nextTab未初始化,nextTab是用来扩容的node数组	
    if (nextTab == null) {            // initiating	
        try {	
            @SuppressWarnings("unchecked")	
//新建一个n<<1原始table大小的nextTab,也就是32	
            Node[] nt = (Node[])new Node[n << 1];	
            nextTab = nt;//赋值给nextTab	
        } catch (Throwable ex) {      // try to cope with OOME	
            sizeCtl = Integer.MAX_VALUE; //扩容失败,sizeCtl使用int的最大值	
            return;	
        }	
        nextTable = nextTab; //更新成员变量	
        transferIndex = n;//更新转移下标,表示转移时的下标	
    }	
    int nextn = nextTab.length;//新的tab的长度	
// 创建一个 fwd 节点,表示一个正在被迁移的Node,并且它的hash值为-1(MOVED),也就是前面我们在讲putval方法的时候,会有一个判断MOVED的逻辑。它的作用是用来占位,表示原数组中位置i处的节点完成迁移以后,就会在i位置设置一个fwd来告诉其他线程这个位置已经处理过了,具体后续还会在讲	
    ForwardingNode fwd = new ForwardingNode(nextTab);	
// 首次推进为 true,如果等于 true,说明需要再次推进一个下标(i--),反之,如果是 false,那么就不能推进下标,需要将当前的下标处理完毕才能继续推进	
    boolean advance = true;	
//判断是否已经扩容完成,完成就return,退出循环	
    boolean finishing = false; // to ensure sweep before committing nextTab	
通过for自循环处理每个槽位中的链表元素,默认advace为真,通过CAS设置transferIndex属性值,并初始化i和bound值,i指当前处理的槽位序号,bound指需要处理的槽位边界,先处理槽位15的节点;	
    for (int i = 0, bound = 0;;) {	
// 这个循环使用CAS不断尝试为当前线程分配任务	
    // 直到分配成功或任务队列已经被全部分配完毕	
    // 如果当前线程已经被分配过bucket区域	
    // 那么会通过--i指向下一个待处理bucket然后退出该循环	
        Node f; int fh;	
        while (advance) {	
            int nextIndex, nextBound;	
//--i表示下一个待处理的bucket,如果它>=bound,表示当前线程已经分配过bucket区域	
            if (--i >= bound || finishing)	
                advance = false;	
            else if ((nextIndex = transferIndex) <= 0) {//表示所有bucket已经被分配完毕	
                i = -1;	
                advance = false;	
            }	
//通过cas来修改TRANSFERINDEX,为当前线程分配任务,处理的节点区间为(nextBound,nextIndex)->(0,15)	
            else if (U.compareAndSwapInt	
                     (this, TRANSFERINDEX, nextIndex,	
                      nextBound = (nextIndex > stride ?	
                                   nextIndex - stride : 0))) {	
                bound = nextBound;//0	
                i = nextIndex - 1;//15	
                advance = false;	
            }	
        }	
//i<0说明已经遍历完旧的数组,也就是当前线程已经处理完所有负责的bucket	
        if (i < 0 || i >= n || i + n >= nextn) {	
            int sc;	
            if (finishing) {//如果完成了扩容	
                nextTable = null;//删除成员变量	
                table = nextTab;//更新table数组	
                sizeCtl = (n << 1) - (n >>> 1);//更新阈值(32*0.75=24)	
                return;	
            }	
// sizeCtl 在迁移前会设置为 (rs << RESIZE_STAMP_SHIFT) + 2   (详细介绍点击这里)	
// 然后,每增加一个线程参与迁移就会将 sizeCtl 加 1,	
// 这里使用 CAS 操作对 sizeCtl 的低16位进行减 1,代表做完了属于自己的任务	
            if (U.compareAndSwapInt(this, SIZECTL, sc = sizeCtl, sc - 1)) {	
第一个扩容的线程,执行transfer方法之前,会设置 sizeCtl = (resizeStamp(n) << RESIZE_STAMP_SHIFT) + 2)	
后续帮其扩容的线程,执行transfer方法之前,会设置 sizeCtl = sizeCtl+1	
每一个退出transfer的方法的线程,退出之前,会设置 sizeCtl = sizeCtl-1	
那么最后一个线程退出时:必然有	
sc == (resizeStamp(n) << RESIZE_STAMP_SHIFT) + 2),即 (sc - 2) == resizeStamp(n) << RESIZE_STAMP_SHIFT                   	
// 如果 sc - 2 不等于标识符左移 16 位。如果他们相等了,说明没有线程在帮助他们扩容了。也就是说,扩容结束了。	
                if ((sc - 2) != resizeStamp(n) << RESIZE_STAMP_SHIFT)	
                    return;	
// 如果相等,扩容结束了,更新 finising 变量	
                finishing = advance = true;	
// 再次循环检查一下整张表	
                i = n; // recheck before commit	
            }	
        }	
// 如果位置 i 处是空的,没有任何节点,那么放入刚刚初始化的 ForwardingNode ”空节点“	
        else if ((f = tabAt(tab, i)) == null)	
            advance = casTabAt(tab, i, null, fwd);	
//表示该位置已经完成了迁移,也就是如果线程A已经处理过这个节点,那么线程B处理这个节点时,hash值一定为MOVED	
        else if ((fh = f.hash) == MOVED) 	
            advance = true; // already processed 	
    }	
}

扩容过程图解

ConcurrentHashMap支持并发扩容,实现方式是,把Node数组进行拆分,让每个线程处理自己的区域,假设table数组总长度是64,默认情况下,那么每个线程可以分到16个bucket。 然后每个线程处理的范围,按照倒序来做迁移 通过for自循环处理每个槽位中的链表元素,默认advace为真,通过CAS设置transferIndex属性值,并初始化i和bound值,i指当前处理的槽位序号,bound指需要处理的槽位边界,先处理槽位31的节点; (bound,i) =(16,31) 从31的位置往前推动。

深度分析ConcurrentHashMap原理分析_第5张图片

假设这个时候ThreadA在进行transfer,那么逻辑图表示如下

深度分析ConcurrentHashMap原理分析_第6张图片

在当前假设条件下,槽位15中没有节点,则通过CAS插入在第二步中初始化的ForwardingNode节点,用于告诉其它线程该槽位已经处理过了;

深度分析ConcurrentHashMap原理分析_第7张图片

sizeCtl扩容退出机制

在扩容操作transfer的第2414行,代码如下

if (U.compareAndSwapInt(this, SIZECTL, sc = sizeCtl, sc - 1)) {	
    if ((sc - 2) != resizeStamp(n) << RESIZE_STAMP_SHIFT)	
每存在一个线程执行完扩容操作,就通过cas执行sc-1。	
接着判断(sc-2) !=resizeStamp(n) << RESIZE_STAMP_SHIFT ; 如果相等,表示当前为整个扩容操作的 最后一个线程,那么意味着整个扩容操作就结束了;如果不想等,说明还得继续	
这么做的目的,一方面是防止不同扩容之间出现相同的sizeCtl,另外一方面,还可以避免sizeCtl的ABA问题导致的扩容重叠的情况	
数据迁移阶段的实现分析	
通过分配好迁移的区间之后,开始对数据进行迁移。在看这段代码之前,先来了解一下原理	
            synchronized (f) {//对数组该节点位置加锁,开始处理数组该位置的迁移工作	
                if (tabAt(tab, i) == f) {//再做一次校验	
                    Node ln, hn;//ln表示低位, hn表示高位;接下来这段代码的作用是把链表拆分成两部分,0在低位,1在高位	
                    if (fh >= 0) {//下面部分代码原理点击这里	
                        int runBit = fh & n; 	
                        Node lastRun = f;	
//遍历当前bucket的链表,目的是尽量重用Node链表尾部的一部分	
                        for (Node p = f.next; p != null; p = p.next) { 	
                            int b = p.hash & n;	
                            if (b != runBit) {	
                                runBit = b;	
                                lastRun = p;	
                            }	
                        }	
                        if (runBit == 0) {如果最后更新的runBit是0,设置低位节点	
                            ln = lastRun;	
                            hn = null;	
                        }	
                        else {//否则,设置高位节点	
                            hn = lastRun;	
                            ln = null;	
                        }	
//构造高位以及低位的链表	
                        for (Node p = f; p != lastRun; p = p.next) {	
                            int ph = p.hash; K pk = p.key; V pv = p.val;	
                            if ((ph & n) == 0)	
                                ln = new Node(ph, pk, pv, ln);	
                            else	
                                hn = new Node(ph, pk, pv, hn);	
                        }	
                        setTabAt(nextTab, i, ln);//将低位的链表放在i位置也就是不动	
                        setTabAt(nextTab, i + n, hn);//将高位链表放在i+n位置	
                        setTabAt(tab, i, fwd); // 把旧table的hash桶中放置转发节点,表明此hash桶已经被处理	
                        advance = true;	
                    }	
                    //红黑树的扩容部分暂时忽略	
        }

高低位原理分析

ConcurrentHashMap在做链表迁移时,会用高低位来实现,这里有两个问题要分析一下

如何实现高低位链表的区分 假如我们有这样一个队列

深度分析ConcurrentHashMap原理分析_第8张图片

第14个槽位插入新节点之后,链表元素个数已经达到了8,且数组长度为16,优先通过扩容来缓解链表过长的问题,扩容这块的图解稍后再分析,先分析高低位扩容的原理 假如当前线程正在处理槽位为14的节点,它是一个链表结构,在代码中,首先定义两个变量节点ln和hn,实际就是lowNode和HighNode,分别保存hash值的第x位为0和不等于0的节点 通过fn&n可以把这个链表中的元素分为两类,A类是hash值的第X位为0,B类是hash值的第x位为不等于0(至于为什么要这么区分,稍后分析),并且通过lastRun记录最后要处理的节点。最终要达到的目的是,A类的链表保持位置不动,B类的链表为14+16(扩容增加的长度)=30 我们把14槽位的链表单独伶出来,我们用蓝色表示 fn&n=0的节点,假如链表的分类是这样

640?wx_fmt=png

for (Node p = f.next; p != null; p = p.next) {	
    int b = p.hash & n;	
    if (b != runBit) {	
        runBit = b;	
        lastRun = p;	
    }	
}	
通过上面这段代码遍历,会记录runBit以及lastRun,按照上面这个结构,那么runBit应该是蓝色节点,lastRun应该是第6个节点	
接着,再通过这段代码进行遍历,生成ln链以及hn链	
for (Node p = f; p != lastRun; p = p.next) {	
    int ph = p.hash; K pk = p.key; V pv = p.val;	
    if ((ph & n) == 0)	
        ln = new Node(ph, pk, pv, ln);	
    else	
        hn = new Node(ph, pk, pv, hn);	
}

深度分析ConcurrentHashMap原理分析_第9张图片

接着,通过CAS操作,把hn链放在i+n也就是14+16的位置,ln链保持原来的位置不动。并且设置当前节点为fwd,表示已经被当前线程迁移完了

setTabAt(nextTab, i, ln);	
setTabAt(nextTab, i + n, hn);	
setTabAt(tab, i, fwd);

迁移完成以后的数据分布如下

深度分析ConcurrentHashMap原理分析_第10张图片

为什么要做高低位的划分

要想了解这么设计的目的,我们需要从ConcurrentHashMap的根据下标获取对象的算法来看,在putVal方法中1018行

(f = tabAt(tab, i = (n - 1) & hash)) == null 

通过(n-1) & hash来获得在table中的数组下标来获取节点数据,【&运算是二进制运算符,1 & 1=1,其他都为0】	
假设我们的table长度是16, 二进制是【0001 0000】,减一以后的二进制是 【0000 1111】	
假如某个key的hash值=9,对应的二进制是【0000 1001】,那么按照(n-1) & hash的算法	
0000 1111 & 0000 1001 =0000 1001 , 运算结果是9	
当我们扩容以后,16变成了32,那么(n-1)的二进制是 【0001 1111】	
仍然以hash值=9的二进制计算为例	
0001 1111 & 0000 1001 =0000 1001 ,运算结果仍然是9	
我们换一个数字,假如某个key的hash值是20,对应的二进制是【0001 0100】,仍然按照(n-1) & hash算法,分别在16为长度和32位长度下的计算结果	
16位:  0000 1111 & 0001 0100=0000 0100	
32位: 0001 1111 & 0001 0100 =0001 0100	
从结果来看,同样一个hash值,在扩容前和扩容之后,得到的下标位置是不一样的,这种情况当然是不允许出现的,所以在扩容的时候就需要考虑,	
而使用高低位的迁移方式,就是解决这个问题.	
大家可以看到,16位的结果到32位的结果,正好增加了16.	
比如 20 & 15=4  、20 & 31=20 ;  4-20 =16	
比如 60 & 15=12 、60 & 31=28; 12-28=16	
所以对于高位,直接增加扩容的长度,当下次hash获取数组位置的时候,可以直接定位到对应的位置。	
这个地方又是一个很巧妙的设计,直接通过高低位分类以后,就使得不需要在每次扩容的时候来重新计算hash,极大提升了效率。

扩容结束以后的退出机制

如果线程扩容结束,那么需要退出,就会执行transfer方法的如下代码

//i<0说明已经遍历完旧的数组,也就是当前线程已经处理完所有负责的bucket	
if (i < 0 || i >= n || i + n >= nextn) {	
            int sc;	
            if (finishing) {//如果完成了扩容	
                nextTable = null;//删除成员变量	
                table = nextTab;//更新table数组	
                sizeCtl = (n << 1) - (n >>> 1);//更新阈值(32*0.75=24)	
                return;	
            }	
// sizeCtl 在迁移前会设置为 (rs << RESIZE_STAMP_SHIFT) + 2  	
// 然后,每增加一个线程参与迁移就会将 sizeCtl 加 1,	
// 这里使用 CAS 操作对 sizeCtl 的低16位进行减 1,代表做完了属于自己的任务	
            if (U.compareAndSwapInt(this, SIZECTL, sc = sizeCtl, sc - 1)) {	
第一个扩容的线程,执行transfer方法之前,会设置 sizeCtl = (resizeStamp(n) << RESIZE_STAMP_SHIFT) + 2)	
后续帮其扩容的线程,执行transfer方法之前,会设置 sizeCtl = sizeCtl+1	
每一个退出transfer的方法的线程,退出之前,会设置 sizeCtl = sizeCtl-1	
那么最后一个线程退出时:必然有	
sc == (resizeStamp(n) << RESIZE_STAMP_SHIFT) + 2),即 (sc - 2) == resizeStamp(n) << RESIZE_STAMP_SHIFT                   	
// 如果 sc - 2 不等于标识符左移 16 位。如果他们相等了,说明没有线程在帮助他们扩容了。也就是说,扩容结束了。	
                if ((sc - 2) != resizeStamp(n) << RESIZE_STAMP_SHIFT)	
                    return;	
// 如果相等,扩容结束了,更新 finising 变量	
                finishing = advance = true;	
// 再次循环检查一下整张表	
                i = n; // recheck before commit	
            }	
        }

put方法第三阶段

如果对应的节点存在,判断这个节点的hash是不是等于MOVED(-1),说明当前节点是ForwardingNode节点, 意味着有其他线程正在进行扩容,那么当前现在直接帮助它进行扩容,因此调用helpTransfer方法

else if ((fh = f.hash) == MOVED)	
    tab = helpTransfer(tab, f);	
helpTransfer	
从名字上来看,代表当前是去协助扩容	
final Node[] helpTransfer(Node[] tab, Node f) {	
    Node[] nextTab; int sc;	
// 判断此时是否仍然在执行扩容,nextTab=null的时候说明扩容已经结束了	
    if (tab != null && (f instanceof ForwardingNode) &&	
        (nextTab = ((ForwardingNode)f).nextTable) != null) {	
        int rs = resizeStamp(tab.length);//生成扩容戳	
        while (nextTab == nextTable && table == tab &&	
               (sc = sizeCtl) < 0) {//说明扩容还未完成的情况下不断循环来尝试将当前线程加入到扩容操作中	
            //下面部分的整个代码表示扩容结束,直接退出循环	
//transferIndex<=0表示所有的Node都已经分配了线程	
//sc=rs+MAX_RESIZERS 表示扩容线程数达到最大扩容线程数	
//sc >>> RESIZE_STAMP_SHIFT !=rs, 如果在同一轮扩容中,那么sc无符号右移比较高位和rs的值,那么应该是相等的。如果不相等,说明扩容结束了	
//sc==rs+1 表示扩容结束	
            if ((sc >>> RESIZE_STAMP_SHIFT) != rs || sc == rs + 1 ||	
                sc == rs + MAX_RESIZERS || transferIndex <= 0)	
                break;//跳出循环	
            if (U.compareAndSwapInt(this, SIZECTL, sc, sc + 1)) {//在低16位上增加扩容线程数	
                transfer(tab, nextTab);//帮助扩容	
                break;	
            }	
        }	
        return nextTab;	
    }	
    return table;//返回新的数组	
}

put方法第四阶段

这个方法的主要作用是,如果被添加的节点的位置已经存在节点的时候,需要以链表的方式加入到节点中 如果当前节点已经是一颗红黑树,那么就会按照红黑树的规则将当前节点加入到红黑树中

else { //进入到这个分支,说明f是当前nodes数组对应位置节点的头节点,并且不为空	
            V oldVal = null;	
            synchronized (f) { //给对应的头结点加锁	
                if (tabAt(tab, i) == f) {//再次判断对应下标位置是否为f节点	
                    if (fh >= 0) { //头结点的hash值大于0,说明是链表	
                        binCount = 1; //用来记录链表的长度	
                        for (Node e = f;; ++binCount) {//遍历链表	
                            K ek;	
//如果发现相同的key,则判断是否需要进行值的覆盖	
                            if (e.hash == hash &&	
                                ((ek = e.key) == key ||	
                                 (ek != null && key.equals(ek)))) {	
                                oldVal = e.val;	
                                if (!onlyIfAbsent) //默认情况下,直接覆盖旧的值	
                                    e.val = value;	
                                break;	
                            }	
//一直遍历到链表的最末端,直接把新的值加入到链表的最后面	
                            Node pred = e;	
                            if ((e = e.next) == null) {	
                                pred.next = new Node(hash, key,	
                                                          value, null);	
                                break;	
                            }	
                        }	
                    }	
//如果当前的f节点是一颗红黑树	
                    else if (f instanceof TreeBin) {	
                        Node p;	
                        binCount = 2;	
//则调用红黑树的插入方法插入新的值	
                        if ((p = ((TreeBin)f).putTreeVal(hash, key,	
                                                       value)) != null) {	
                            oldVal = p.val; //同样,如果值已经存在,则直接替换	
                            if (!onlyIfAbsent)	
                                p.val = value;	
                        }	
                    }	
                }	
            }

put方法第四个阶段

判断链表的长度是否已经达到临界值8. 如果达到了临界值,这个时候会根据当前数组的长度来决定是扩容还是将链表转化为红黑树。也就是说如果当前数组的长度小于64,就会先扩容。否则,会把当前链表转化为红黑树

            if (binCount != 0) {//说明上面在做链表操作	
                //如果链表长度已经达到临界值8 就需要把链表转换为树结构	
if (binCount >= TREEIFY_THRESHOLD)	
                    treeifyBin(tab, i); 	
                if (oldVal != null)//如果val是被替换的,则返回替换之前的值	
                    return oldVal;	
                break;	
            }

treeifyBin

在putVal的最后部分,有一个判断,如果链表长度大于8,那么就会触发扩容或者红黑树的转化操作。

private final void treeifyBin(Node[] tab, int index) {	
    Node b; int n, sc;	
    if (tab != null) {	
        if ((n = tab.length) < MIN_TREEIFY_CAPACITY) //tab的长度是不是小于64,如果是,则执行扩容	
            tryPresize(n << 1);  	
        else if ((b = tabAt(tab, index)) != null && b.hash >= 0) {//否则,将当前链表转化为红黑树结构存储	
            synchronized (b) {// 将链表转换成红黑树	
                if (tabAt(tab, index) == b) {	
                    TreeNode hd = null, tl = null;	
                    for (Node e = b; e != null; e = e.next) {	
                        TreeNode p =	
                            new TreeNode(e.hash, e.key, e.val,	
                                              null, null);	
                        if ((p.prev = tl) == null)	
                            hd = p;	
                        else	
                            tl.next = p;	
                        tl = p;	
                    }	
                    setTabAt(tab, index, new TreeBin(hd));	
                }	
            }	
        }	
    }	
}

tryPresize

tryPresize里面部分代码和addCount的部分代码类似,看起来会稍微简单一些

private final void tryPresize(int size) {	
//对size进行修复,主要目的是防止传入的值不是一个2次幂的整数,然后通过tableSizeFor来讲入参转化为离该整数最近的2次幂	
    int c = (size >= (MAXIMUM_CAPACITY >>> 1)) ? MAXIMUM_CAPACITY :	
        tableSizeFor(size + (size >>> 1) + 1);	
    int sc;	
    while ((sc = sizeCtl) >= 0) {	
        Node[] tab = table; int n;	
//下面这段代码和initTable是一样的,如果table没有初始化,则开始初始化	
        if (tab == null || (n = tab.length) == 0) {	
            n = (sc > c) ? sc : c;	
            if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {	
                try {	
                    if (table == tab) {	
                        @SuppressWarnings("unchecked")	
                        Node[] nt = (Node[])new Node[n];	
                        table = nt;	
                        sc = n - (n >>> 2);//0.75	
                    }	
                } finally {	
                    sizeCtl = sc;	
                }	
            }	
        }	
//	
        else if (c <= sc || n >= MAXIMUM_CAPACITY)	
            break;	
        else if (tab == table) {//这段代码和addCount后部分代码是一样的,做辅助扩容操作	
            int rs = resizeStamp(n);	
            if (sc < 0) {	
                Node[] nt;	
                if ((sc >>> RESIZE_STAMP_SHIFT) != rs || sc == rs + 1 ||	
                    sc == rs + MAX_RESIZERS || (nt = nextTable) == null ||	
                    transferIndex <= 0)	
                    break;	
                if (U.compareAndSwapInt(this, SIZECTL, sc, sc + 1))	
                    transfer(tab, nt);	
            }	
            else if (U.compareAndSwapInt(this, SIZECTL, sc,	
                                         (rs << RESIZE_STAMP_SHIFT) + 2))	
                transfer(tab, null);	
        }	
    }	
}

你可能感兴趣的:(深度分析ConcurrentHashMap原理分析)