ConcurrentHashMap1.8源码分析

最新推荐文章于 2022-04-08 13:46:38 发布

程序人生|

最新推荐文章于 2022-04-08 13:46:38 发布

阅读量349

点赞数

分类专栏： Java 文章标签： Java 程序员 HashMap 互联网架构师

本文链接：https://blog.csdn.net/xunjiushi9717/article/details/85310935

版权

本文深入探讨了JDK1.8中ConcurrentHashMap的源码实现，包括put方法的线程安全策略、扩容操作transfer()的细节、链表转红黑树的条件以及线程协助扩容的机制。分析了从put过程中的数组初始化、节点添加到扩容逻辑的各个关键步骤，旨在理解其高效且线程安全的设计。

摘要由CSDN通过智能技术生成

文章简介

想必大家对HashMap数据结构并不陌生，JDK1.7采用的是数组+链表的方式，JDK1.8采用的是数组+链表+红黑树的方式。虽然JDK1.8对于HashMap有了很大的改进，提高了存取效率，但是线程安全的问题不可忽视，所以就有了线程安全的解决方案，比如在方法上加synchronized同步锁的HashTable，或者并发包中的ConcurrentHashMap线程安全类，本文就来和大家一起探讨一下关于ConcurrentHashMap的源码，版本是JDK1.8，下面让我们正式开始吧。

备注：大家需要对HashMap1.8源码有一些了解，在原来HashMap1.8源码中比较常见的知识点本文不会具体展开。

内容导航

数组初始化线程安全实现
put(key,value)线程安全实现
transfer扩容及不同的扩容场景

01 put(key,value)方法

不妨先以一段大家熟悉的代码开始本文的旅程

ConcurrentHashMap<Integer,String> map=new ConcurrentHashMap<Integer, String>();
map.put(1,"Zhang");

当我们在put元素时，点开put方法的源码会发现，这里调用了一个putVal()的方法，同时将key和value作为参数传入

public V put(K key, V value) {
   
        return putVal(key, value, false);
}

继续点击putVal()方法，然后我们看看这里到底实现了什么

//key或者value都不能为空
if (key == null || value == null) throw new NullPointerException();
//计算hash值，实际上就是得到一个int类型的数，只是需要对这个数进行处理，目的是为了确定key，value组成的Node节点在数组下标中的位置
int hash = spread(key.hashCode());

不妨先看下spread(key.hashCode())的实现

key.hashCode()实际上调用的是native的方法，目的是得到一个整形数，为了使得这个整形数尽可能不一样，所以要对高16位和低16位进行异或运算，尽可能利用好每一位的值
static final int spread(int h) {
   
	//对key.hashCode的结果进行高16位和低16位的运算
	return (h ^ (h >>> 16)) & HASH_BITS;
}

接下来就是要初始化这个数组的大小，因为数组不初始化，代表key,value的每个Node类也不能放到对应的位置

if (tab == null || (n = tab.length) == 0)
    //初始化数组的大小
     tab = initTable();

private final Node<K,V>[] initTable() {
   
    Node<K,V>[] tab; int sc;
    //只有当数组为空或者大小为0的时候才对数组进行初始化
    while ((tab = table) == null || tab.length == 0) {
   
        //这里其实就是用一个sizeCtl记录是否已经有线程在进行初始化操作，如果有，则让出CPU的资源，也就是保证只有一个线程对数组进行初始化操作，从而保证线程安全。
        if ((sc = sizeCtl) < 0)
            Thread.yield(); // lost initialization race; just spin
        //使用CAS乐观锁机制比较SIZECTL和sc是否相等，只有当前值和内存中最新值相等的时候，才会将当前值赋值为-1,一旦被赋值为-1，上面有其他线程进来，就直接执行了Thread.yeild()方法了
        else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {
   
            try {
   
                if ((tab = table) == null || tab.length == 0) {
   
                    //三元运算符得到数组默认大小，点击DEFAULT_CAPACITY发现是16，这点和HashMap是一样的
                    int n = (sc > 0) ? sc : DEFAULT_CAPACITY;
                    @SuppressWarnings("unchecked")
                    //创建Node类型的数组，真正初始化的地方
                    Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n];
                    table = tab = nt;
                    //计算扩容的标准，采用的是位移运算，因为效率更高，sc最终结果为12
                    sc = n - (n >>> 2);
                }
            } finally {
   
                //不管无论最终将sc赋值为sizeCtl，这时候sizeCtl结果为12
                sizeCtl = sc;
            }
            break;
        }
    }
    return tab;
}

当数组初始化完成之后，就需要将key，value创建出来的Node节点放到数组中对应的位置了，分为几种情况，下面这种是原来某个位置就没有元素值，但是为了保证线程安全，放到多个线程同时添加，也使用CAS乐观锁的机制进行添加。

//根据(n-1)&hash的结果确认当前节点所在的位置是否有元素，效果和hash%n是一样的，只是&运算效率更高，这里hashmap也是这样做的，就不做更多赘述了
else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {
   
    //使用CAS乐观锁机制向对应的下标中添加对应的Node
    if (casTabAt(tab, i, null,
    	new Node<K,V>(hash, key, value, null)))
    	break;                   // no lock when adding to empty bin
}

//f实际上是当前数组下标的Node节点，这里判断它的hash值是否为MOVED，也就是-1，如果是-1，就调用helpTransfer(tab,f)方法帮助其他线程完成扩容操作，然后再添加元素。 
else if ((fh = f.hash) == MOVED)
     tab = helpTransfer