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;
            }
        }
      ...// 繼續(xù)省略一部分代碼
    }

可以看到崇众,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存在,則執(zhí)行l(wèi)ambda表達式赦抖,表達式入參為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原來的節(jié)點對象
        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對象
          ··· // 繼續(xù)省略
        }
        if (old != null) {// 第三個if舍杜,如果 old Node 存在
            V v;
            if (old.value != null)
            // 如果old存在,執(zhí)行l(wèi)ambda赵辕,算出新的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()來規(guī)避嫉晶。

map.computeIfAbsent(k, key ->  1);
// 該方法等價于
map.putIfAbsent(k, 1);
// 所以computeIfAbsent在涉及到用key來計算val時才有使用價值。否則可以用putIfAbsent代替田篇。

當key不存在時替废,才compute。其他行為見注釋泊柬。

源碼略椎镣。

computeIfPresent()

compute()的補充,key存在時才compute()兽赁,避免潛在的空指針情況擂达。其他和compute()相同。
源碼略冈绊。

?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末瞧柔,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子述吸,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 206,968評論 6 482
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件充活,死亡現場離奇詭異,居然都是意外死亡蜡娶,警方通過查閱死者的電腦和手機混卵,發(fā)現死者居然都...
    沈念sama閱讀 88,601評論 2 382
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來窖张,“玉大人淮菠,你說我怎么就攤上這事』缈埃” “怎么了合陵?”我有些...
    開封第一講書人閱讀 153,220評論 0 344
  • 文/不壞的土叔 我叫張陵,是天一觀的道長澄阳。 經常有香客問我拥知,道長,這世上最難降的妖魔是什么碎赢? 我笑而不...
    開封第一講書人閱讀 55,416評論 1 279
  • 正文 為了忘掉前任低剔,我火速辦了婚禮,結果婚禮上肮塞,老公的妹妹穿的比我還像新娘襟齿。我一直安慰自己,他們只是感情好枕赵,可當我...
    茶點故事閱讀 64,425評論 5 374
  • 文/花漫 我一把揭開白布猜欺。 她就那樣靜靜地躺著,像睡著了一般拷窜。 火紅的嫁衣襯著肌膚如雪开皿。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,144評論 1 285
  • 那天篮昧,我揣著相機與錄音赋荆,去河邊找鬼。 笑死懊昨,一個胖子當著我的面吹牛窄潭,可吹牛的內容都是我干的。 我是一名探鬼主播酵颁,決...
    沈念sama閱讀 38,432評論 3 401
  • 文/蒼蘭香墨 我猛地睜開眼嫉你,長吁一口氣:“原來是場噩夢啊……” “哼信认!你這毒婦竟也來了?” 一聲冷哼從身側響起均抽,我...
    開封第一講書人閱讀 37,088評論 0 261
  • 序言:老撾萬榮一對情侶失蹤嫁赏,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后油挥,有當地人在樹林里發(fā)現了一具尸體潦蝇,經...
    沈念sama閱讀 43,586評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,028評論 2 325
  • 正文 我和宋清朗相戀三年深寥,在試婚紗的時候發(fā)現自己被綠了攘乒。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,137評論 1 334
  • 序言:一個原本活蹦亂跳的男人離奇死亡惋鹅,死狀恐怖则酝,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情闰集,我是刑警寧澤沽讹,帶...
    沈念sama閱讀 33,783評論 4 324
  • 正文 年R本政府宣布,位于F島的核電站武鲁,受9級特大地震影響爽雄,放射性物質發(fā)生泄漏。R本人自食惡果不足惜沐鼠,卻給世界環(huán)境...
    茶點故事閱讀 39,343評論 3 307
  • 文/蒙蒙 一挚瘟、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧饲梭,春花似錦乘盖、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,333評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至监氢,卻和暖如春布蔗,著一層夾襖步出監(jiān)牢的瞬間藤违,已是汗流浹背浪腐。 一陣腳步聲響...
    開封第一講書人閱讀 31,559評論 1 262
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留顿乒,地道東北人议街。 一個月前我還...
    沈念sama閱讀 45,595評論 2 355
  • 正文 我出身青樓,卻偏偏與公主長得像璧榄,于是被迫代替她去往敵國和親特漩。 傳聞我的和親對象是個殘疾皇子吧雹,可洞房花燭夜當晚...
    茶點故事閱讀 42,901評論 2 345

推薦閱讀更多精彩內容