Java集合 —— ConcurrentHashMap源码笔记

简介

HashMap是线程不安全的,所以 Java 还提供了 ConcurrentHashMap 类来解决高并发下的安全问题。

Java8 中,ConcurrentHashMap 相对于 Java7 来说, 它是通过 CAS 操作和 synchronized 来实现线程安全的, 而 Java7 是使用分段锁来实现线程安全的,并且Java7是对数组分段同步,而 Java8 是对数组元素同步。

这里大致看下的它的源码,ConcurrentHashMap 相对于 HashMap 来说,因为它处理的并发的情况,所以源码会复杂不少,这里做个大致了解与 HashMap 做个对比。

依然从 put 开始了解 ConcurrentHashMap 是如何实现线程安全的。

源码分析 Java8

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) {
        //这里可以直接发现一个和 HashMap 不同的地方, ConcurrentHashMap 不支持含 null 的键值对
        if (key == null || value == null) throw new NullPointerException();
        //计算哈希值
        int hash = spread(key.hashCode());
        //这个值将用来记录链表或者红黑树长度
        int binCount = 0;
        //此处开始自旋
        for (Node<K,V>[] tab = table;;) {
            Node<K,V> f; int n, i, fh;
            if (tab == null || (n = tab.length) == 0)
                //如果数组不存在或者长度为 0,则初始化数组
                tab = initTable();
            else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {
                //如果计算出来的位置在数组中还没有存放对象,那么通过 CAS 来放入
                if (casTabAt(tab, i, null,
                             new Node<K,V>(hash, key, value, null)))
                    break;                   // no lock when adding to empty bin
            }
            else if ((fh = f.hash) == MOVED)
                //由于可能在高并发的情况下, 所以正在扩容的时候也要考虑,这里会帮助扩容
                tab = helpTransfer(tab, f);
            else {
                //如果该位置不为空,则可能有链表或者红黑树
                V oldVal = null;
                //使用 synchronized 同步该位置下对应的数组里的对象
                synchronized (f) {
                    if (tabAt(tab, i) == f) {
                        if (fh >= 0) {
                        //如果存在链表
                            binCount = 1;
                            //遍历链表
                            for (Node<K,V> e = f;; ++binCount) {
                                //接来下的逻辑和 HashMap 一样,在链表尾部插入新对象
                                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;
                                }
                            }
                        }
                        else if (f instanceof TreeBin) {
                        //如果当前块里是红黑树
                            Node<K,V> p;
                            binCount = 2;
                            //此处遍历红黑树
                            if ((p = ((TreeBin<K,V>)f).putTreeVal(hash, key,
                                                           value)) != null) {
                                //如果 key 相同, 那么根据 onlyIfAbsent 判断是否需要替换                           
                                oldVal = p.val;
                                if (!onlyIfAbsent)
                                    p.val = value;
                            }
                        }
                        else if (f instanceof ReservationNode)
                            throw new IllegalStateException("Recursive update");
                    }
                }
                if (binCount != 0) {
                    if (binCount >= TREEIFY_THRESHOLD)
                        // 链表节点大于8 转成红黑树
                        treeifyBin(tab, i);
                    if (oldVal != null)
                        return oldVal;
                    break;
                }
            }
        }
        //扩容的逻辑也在这里面
        addCount(1L, binCount);
        return null;
    }

ConcurrentHashMap 的 put 方法,首先自旋,如果存放对象的块中为 null 时,将通过 CAS 来放入新键值对,如果已经存在对象的话,则使用 synchronized 给插入操作上锁。

如果发现正在扩容,那么将会利用并发的特性,来帮助扩容。

补上初始化数组的方法 initTable

initTable 方法

    /**
     * Initializes table, using the size recorded in sizeCtl.
     */
    private final Node<K,V>[] initTable() {
        Node<K,V>[] tab; int sc;
        //不断循环直到数组建立
        while ((tab = table) == null || tab.length == 0) {
            if ((sc = sizeCtl) < 0)
                //如果 sizeCtl 小于 0,则说明有其他地方先初始化数组,所以放弃执行权
                Thread.yield(); // lost initialization race; just spin
            else if (U.compareAndSwapInt(this, SIZECTL, sc, -1)) {
            //不然就通过 CAS 来修改 sizeCtl 值为 -1 ,告诉其它线程数组正在初始化
                try {
                    //接下来就是创建一个数组
                    if ((tab = table) == null || tab.length == 0) {
                        int n = (sc > 0) ? sc : DEFAULT_CAPACITY;
                        @SuppressWarnings("unchecked")
                        Node<K,V>[] nt = (Node<K,V>[])new Node<?,?>[n];
                        table = tab = nt;
                        sc = n - (n >>> 2);
                    }
                } finally {
                    //创建完数组后就将 sizeCtl 修改
                    sizeCtl = sc;
                }
                break;
            }
        }
        //返回创建好的数组
        return tab;
    }

可以看出 ConcurrentHashMap 初始化数组的核心是 sizeCtl 这个值,这个值是 volatile 修饰的,保证了它的可见性,通过判断这个值是不是小于 0,来决定是否需要执行数组的初始化。

最后

从这两段源码可以看出 ConcurrentHashMap 是怎么样实现一个线程安全的 HashMap 了。

这里引出一下 HashTable 为什么被弃用的问题。

HashTable 与 HashMap 的不同

  • 不支持 null 键值对,HashMap 可以。
  • 线程安全,是对修改过程同步实现的,这样效率会很低。而 HashMap 不是线程安全。
  • 初始容量是 11, 扩容是 2 * oldCap + 1, HashMap 为 16,2 * oldCap。
  • 初始容量即扩容大小不一样,所以计算 index 的值方法也不一样。

为什么弃用 HashTable

原因主要出在上面第二点,它的修改是对整个方法同步实现的,效率会低非常多,同时它与 HashMap 还是有许多不同的地方, ConcurrentHashMap 在容量以及扩容规则等都延续了 HashMap。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342

推荐阅读更多精彩内容