【广撒网】ConcurrentHashMap提供的computeIfAbsent源码分析

JDK7的ConcurrentHashMap使用Segment分段锁机制,但是在JDK8采用CAS自旋》+synchronized锁机制。并且加锁的粒度更小,直接锁Node数组的第一个元素,即实现了对单个key加锁。

    private void analysisPo2Cache(Class < ?>clazz) {
       //维护缓存
        cache.computeIfAbsent(clazz.getName(), key - >{
        HashSet < Field > values = new HashSet < >();
        Field[] declaredFields = clazz.getDeclaredFields();
        Field.setAccessible(declaredFields, true);
        for (Field field: declaredFields) {
            Class < ?>fieldClass = field.getType();
            //判断是否含有List的属性
            if (List.class.isAssignableFrom(fieldClass)) {
                values.add(field);
            }
        }
        return values;
    });
}

源码分析:

JDK8的Map的数据结构是Node数组和链表+红黑树。
线程安全的核心思想:CAS+ synchronized锁对Node数组上第一个节点加锁。

  1. 多个线程均会创建ReservationNode节点,均执行执行synchronized(r)(因为锁的是对象,所以所以线程都会执行casTabAt方法)。CAS失败的线程释放锁,但是CAS成功的线程将执行lamda表达式(此时,持有synchronized(r)的锁)。
  2. 失败的线程将再次自旋,在(f = tabAt(tab, i = (n - 1) & h)) == null)方法中获取key所在位置的Node对象(lamda表达式没执行完毕,那么Node对象就是r,但是r已经被synchronized),此时synchronized(f)会被阻塞,等待synchronized(r)释放锁。当等到synchronized(r)释放锁后,key所在位置的Node对象会发生变化,会再次通过tabAt(tab, i) == f判断。开启下次自旋。
public V computeIfAbsent(K key, Function < ?super K, ?extends V > mappingFunction) {
    //key的lamda表达式都不能为空
    if (key == null || mappingFunction == null) throw new NullPointerException();
     //右移16位计算有效的hash值
    int h = spread(key.hashCode());
    V val = null;
    int binCount = 0;
    //开启自旋
    for (Node <K, V> [] tab = table;;) {
        Node <K,V> f;
        int n,i,fh;
        //初次put时,因为concurrenthashmap在构造时没有初始化table,所以先初始化table
        if (tab == null || (n = tab.length) == 0) tab = initTable();
        //n为数组长度,低位有效二次hash后计算出数组的下标,并获取该下标的node对象(线程不安全),且node所在的位置为null
        else if ((f = tabAt(tab, i = (n - 1) & h)) == null) {
            //创建空节点(预定节点)
            Node <K,V> r = new ReservationNode <K,V> ();
            //【神操作】这里加synchronized是为了CAS失败的线程在此自旋的时候在synchronized(f)处阻塞,等待CAS维护value释放锁。
            synchronized(r) {
                //CAS,使用预定节点去替换null(只有一个线程成功)
                if (casTabAt(tab, i, null, r)) {
                    //成功进入的线程维护value。
                    binCount = 1;
                    Node <K,V> node = null;
                    try {
                        //执行lamda表达式
                        if ((val = mappingFunction.apply(key)) != null) 
                            node = new Node <K,V> (h, key, val, null);
                    } finally {
                        //设置内存的值
                        setTabAt(tab, i, node);
                    }
                }
            }
            //CAS进入的线程结束自旋,其他线程继续向下执行。
            if (binCount != 0) break;
        } else if ((fh = f.hash) == MOVED) 
              //node的hash值为-1,表示正在扩容,本次操作会帮助扩容【多线程完成扩容】
              tab = helpTransfer(tab, f);
        else {
            //CAS加锁失败后的线程,自旋最终会到达此处
            boolean added = false;
            //【神操作2】即处理了lamda表达式执行慢,sync(r)方法持有锁,也解决了并发插入线程安全。
            synchronized(f) {
                //【神操作3】sync(r)释放锁后,线程才会获取到到此处的锁,此时f一定不是tabAt(tab, i) 。需要重新自旋。
                if (tabAt(tab, i) == f) {
                    //fh为该node的hash值,如果hash值为正数,说明该下标是按链表存储的
                    if (fh >= 0) {
                        binCount = 1;
                        //遍历链表(哈希桶)
                        for (Node < K, V > e = f;; ++binCount) {
                            K ek;V ev;
                            //若key相同,直接返回,结束自旋。
                            if (e.hash == h && ((ek = e.key) == key || (ek != null && key.equals(ek)))) {
                                val = e.val;
                                break;
                            }
                            Node < K,V > pred = e;
                           //如果链表中不存在这个key,则添加到链表末尾
                            if ((e = e.next) == null) {
                                if ((val = mappingFunction.apply(key)) != null) {
                                    added = true;
                                    pred.next = new Node < K,V > (h, key, val, null);
                                }
                                break;
                            }
                        }
                    } else if (f instanceof TreeBin) {
                        //红黑树的情况
                        binCount = 2;
                        TreeBin < K,V > t = (TreeBin < K, V > ) f;
                        TreeNode < K,V > r,
                        p;
                        //节点加到红黑树
                        if ((r = t.root) != null && (p = r.findTreeNode(h, key, null)) != null) val = p.val;
                        else if ((val = mappingFunction.apply(key)) != null) {
                            added = true;
                            t.putTreeVal(h, key, val);
                        }
                    }
                }
            }
            //加锁成功的线程binCount会变为1
            if (binCount != 0) {
                //bitCount>8,尝试将链表转换为红黑树
                if (binCount >= TREEIFY_THRESHOLD) 
                    treeifyBin(tab, i);
                if (!added)
                   return val;
                //结束自旋
                break;
            }
        }
    }
   //总node+1,该方法中满足条件会进行扩容。
    if (val != null) addCount(1L, binCount);
    return val;
}

总结:
【广撒网】若key对应的第一个Node节点为null,每个线程均会对新建的空节点对象加锁,但只有CAS成功的线程一直持有对象锁,其他线程释放锁,开启下次自旋。
【实时更新】若key对应的第一个Node节点不为null,线程获取最新的Node节点对象(但Node节点可能已经被加锁),尝试对Node加锁去put数据。

推荐阅读

Java8之ConcurrentHashMap实现原理

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

推荐阅读更多精彩内容