Java HashMap 新增方法(merge,compute)

Java8 Map里新增了几个方法,很多同学不知道具体功能是啥。


先从最简单的开始。

putIfAbsent()

HashMap<String, String> map = new HashMap<String, String>();
map.putIfAbsent("k", "v");

等价于:(功能等价,效率并不等价)

HashMap<String, String> map = new HashMap<String, String>();
if(!map.containsKey("k")) {
    map.put("k", "v");    
}

key不存在才put,存在就跳过。


源码:
    @Override
    public V putIfAbsent(K key, V value) {
        return putVal(hash(key), key, value, true, true);
    }
    final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
                   boolean evict) {
        ...// 省略一部分无关代码
            if (e != null) { 
                V oldValue = e.value;
                if (!onlyIfAbsent || oldValue == null) // 直接看这里
                    e.value = value;
                afterNodeAccess(e);
                return oldValue;
            }
        }
      ...// 继续省略一部分代码
    }

可以看到,putIfAbsent()调用了putVal(),第四个参数onlyIfAbsent传true,表示只有无此key时才put。如果putIfAbsent()的key不存在,和put()逻辑相同。如果已存在,onlyIfAbsent延迟到putVal()if (e != null)时才做判断。

merge()

        String k = "key";
        HashMap<String, Integer> map = new HashMap<String, Integer>() {{
            put(k, 1);
        }};
        map.merge(k, 2, (oldVal, newVal) -> oldVal + newVal);

等价于:

        String k = "key";
        HashMap<String, Integer> map = new HashMap<String, Integer>() {{
            put(k, 1);
        }};
        Integer newVal = 2;
        if(map.containsKey(k)) {
            map.put(k, map.get(k) + newVal);
        } else {
            map.put(k, newVal);
        }

如果key存在,则执行lambda表达式,表达式入参为oldValnewVal(neVal即merge()的第二个参数)。表达式返回最终put的val。如果key不存在,则直接putnewVal


源码:

    public V merge(K key, V value,
                   BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
      ···// 删除无关代码
        int hash = hash(key);
        Node<K,V>[] tab; Node<K,V> first; int n, i;
        int binCount = 0;
        TreeNode<K,V> t = null;
        Node<K,V> old = null; // 该key原来的节点对象
        if (size > threshold || (tab = table) == null ||
            (n = tab.length) == 0) //第一个if,判断是否需要扩容
            n = (tab = resize()).length;
        if ((first = tab[i = (n - 1) & hash]) != null) {
           // 第二个if,取出old Node对象
          ··· // 继续省略
        }
        if (old != null) {// 第三个if,如果 old Node 存在
            V v;
            if (old.value != null)
            // 如果old存在,执行lambda,算出新的val并写入old Node后返回。
                v = remappingFunction.apply(old.value, value);
            else
                v = value;
            if (v != null) {
          
                old.value = v;
                afterNodeAccess(old);
            }
            else
                removeNode(hash, key, null, false, true);
            return v;
        }
        if (value != null) {
        //如果old不存在且传入的newVal不为null,则put新的kv
            if (t != null)
                t.putTreeVal(this, tab, hash, key, value);
            else {
                tab[i] = newNode(hash, key, value, first);
                if (binCount >= TREEIFY_THRESHOLD - 1)
                    treeifyBin(tab, hash);
            }
            ...// 省略
        }
        return value;
    }

解释见注释。

compute()

        String k = "key";
        HashMap<String, Integer> map = new HashMap<String, Integer>() {{
            put(k, 1);
        }};
        map.compute(k, (key, oldVal) -> oldVal + 1);

等价于

    map.put(k, func(k, map.get(k)));

    public Integer func(String k, Integer oldVal) {
        return oldVal + 1;
    }

根据已知的 k v 算出新的v并put。
注意:
如果无此key,那么oldVal为null,lambda中涉及到oldVal的计算会报空指针。
源码和merge大同小异,就不放了。

computeIfAbsent()

由上可知,compute()有空指针的风险。所以用computeIfAbsent()来规避。

map.computeIfAbsent(k, key ->  1);
// 该方法等价于
map.putIfAbsent(k, 1);
// 所以computeIfAbsent在涉及到用key来计算val时才有使用价值。否则可以用putIfAbsent代替。

当key不存在时,才compute。其他行为见注释。

源码略。

computeIfPresent()

compute()的补充,key存在时才compute(),避免潜在的空指针情况。其他和compute()相同。
源码略。

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

推荐阅读更多精彩内容