1. 程式人生 > >Java中對映Map的merge、compute、computeIfAbsent、computeIfPresent基本用法

Java中對映Map的merge、compute、computeIfAbsent、computeIfPresent基本用法

下面是Java8中Map的一些新方法merge、compute、computeIfAbsent、computeIfPresent介紹。

我們在專案開發中,經常使用map,key有時存在有時不存,我們需要是用containsKey去判斷,然後再決定如何修改value。 這樣比較麻煩,嫩不能在一個方法呼叫就完成這些工作呢(如果key存在value(還可以有其他邏輯判斷),就do a,如果不存在就do b)?答案是, 當然可以。

下面直接給出示例和執行結果,參看結果就明白各個方法的具體含義了。
 

import java.util.HashMap;
import java.util.Map;
import java.util.function.BiFunction;


public class MapMain {

    /**
     * @param args
     */
    public static void main(String[] args) {
        Map<String, String> myMap = new HashMap<>();
        String keyA = "A";
        String keyB = "B";
        String keyC = "C";
        String keyD = "D";
        String keyE = "E";
        String keyF = "F";
        String keyG = "G";
        String keyH = "H";
        myMap.put(keyA, "str01A");
        myMap.put(keyB, "str01B");
        myMap.put(keyC, "str01C");

        System.out.println("myMap initial content:"+ myMap);

        myMap.merge(keyA, "merge01", String::concat);
        myMap.merge(keyD, "merge01", String::concat);
        System.out.println("Map merge demo content:"+ myMap);

        BiFunction<String, String, String> biFunc = new BiFunction<String, String, String>(){
            @Override
            public String apply(String t, String u) {
                String result = t;
                if (t == null) {
                    result = u;
                }
                else {
                    result += "," + u;
                }
                return result;
            }
        };

        myMap.merge(keyA, "BiFuncMerge01", biFunc);
        myMap.merge(keyE, "BiFuncMerge01", biFunc);
        System.out.println("Map customized BiFunction merge demo content:"+ myMap);

        String msg = "msgCompute";
        myMap.compute(keyB, (k, v) -> (v == null) ? msg : v.concat(msg));
        myMap.compute(keyF, (k, v) -> (v == null) ? msg : v.concat(msg));
        System.out.println("Map customized BiFunction compute demo content:"+ myMap);

        myMap.computeIfAbsent(keyC, k -> genValue(k));
        myMap.computeIfAbsent(keyG, k -> genValue(k));
        System.out.println("Map customized Function computeIfAbsent demo content:"+ myMap);

        myMap.computeIfPresent(keyC, biFunc);
        myMap.computeIfPresent(keyH, biFunc);
        System.out.println("Map customized biFunc computeIfPresent demo content:"+ myMap);
    }

    static String genValue(String str) {  
        System.out.println("===");  
        return str + "2";  
    }  
}

執行結果

myMap initial content:{A=str01A, B=str01B, C=str01C}
Map merge demo content:{A=str01Amerge01, B=str01B, C=str01C, D=merge01}
Map customized BiFunction merge demo content:{A=str01Amerge01,BiFuncMerge01, B=str01B, C=str01C, D=merge01, E=BiFuncMerge01}
Map customized BiFunction compute demo content:{A=str01Amerge01,BiFuncMerge01, B=str01BmsgCompute, C=str01C, D=merge01, E=BiFuncMerge01, F=msgCompute}
===
Map customized Function computeIfAbsent demo content:{A=str01Amerge01,BiFuncMerge01, B=str01BmsgCompute, C=str01C, D=merge01, E=BiFuncMerge01, F=msgCompute, G=G2}
Map customized biFunc computeIfPresent demo content:{A=str01Amerge01,BiFuncMerge01, B=str01BmsgCompute, C=C,str01C, D=merge01, E=BiFuncMerge01, F=msgCompute, G=G2}