Java8 中使用Stream 讓List 轉(zhuǎn) Map使用問(wèn)題小結(jié)

更新時(shí)間：2021年06月16日 14:59:32 作者：Kevin.ZhangCG

這篇文章主要介紹了Java8 中使用Stream 讓List 轉(zhuǎn) Map使用總結(jié),本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

在使用 Java 的新特性 Collectors.toMap() 將 List 轉(zhuǎn)換為 Map 時(shí)存在一些不容易發(fā)現(xiàn)的問(wèn)題，這里總結(jié)一下備查。

空指針風(fēng)險(xiǎn)

java.lang.NullPointerException

當(dāng) List 中有 null 值的時(shí)候，使用 Collectors.toMap() 轉(zhuǎn)為 Map 時(shí)，會(huì)報(bào) java.lang.NullPointerException，如下：

List<SdsTest> sdsTests = new ArrayList<>();
    SdsTest sds1 = new SdsTest("aaa","aaa");
    SdsTest sds2 = new SdsTest("bbb",null);

    sdsTests.add(sds1);
    sdsTests.add(sds2);

    Map<String, String> map = sdsTests.stream().collect(Collectors.toMap(SdsTest::getName, SdsTest::getAge));    
    System.out.println(map.toString());

---------
運(yùn)行錯(cuò)誤：
Exception in thread "main" java.lang.NullPointerException
    at java.util.HashMap.merge(HashMap.java:1216)
    at java.util.stream.Collectors.lambda$toMap$150(Collectors.java:1320)
    .....

　　原因是toMap()方法中使用Map.merge()方法合并時(shí)，merge 不允許 value 為 null 導(dǎo)致的，源碼如下：

default V merge(K key, V value, BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
    Objects.requireNonNull(remappingFunction);
    // 在這里判斷了value不可為null
    Objects.requireNonNull(value);
    V oldValue = get(key);
    V newValue = (oldValue == null) ? value : remappingFunction.apply(oldValue, value);
    ...

解決方法

業(yè)務(wù)控制不要出現(xiàn) Null 值【有 Null 的地方，可以賦值默認(rèn)值】在轉(zhuǎn)換時(shí)加判斷，如果為 null，則給一個(gè)默認(rèn)值

Map<String, String> map = sdsTests.stream().collect(Collectors.toMap(SdsTest::getName, sdsTest -> sdsTest.getAge() == null ? "0" : sdsTest.getAge()));

使用 collect(..) 構(gòu)建，允許空值

Map<String, String> nmap = sdsTests.stream().collect(HashMap::new,(k, v) -> k.put(v.getName(), v.getAge()), HashMap::putAll);
// TODO 下游業(yè)務(wù)從Map取值要做NPE判斷

使用 Optional 對(duì)值進(jìn)行包裝

Map<String, Optional<String>> opmap = sdsTests.stream().collect(Collectors.toMap(SdsTest::getName, sdsTest -> Optional.ofNullable(sdsTest.getAge())));
System.out.println("bbb.age=" + opmap.get("bbb").orElse("0"));
------------
輸出：
bbb.age=0

建議

優(yōu)先業(yè)務(wù)控制，盡量避免 List 中存在 Null
其次推薦第 4 種方法【使用 Optional 對(duì)值進(jìn)行包裝】，能很好的避免 NPE 問(wèn)題

key重復(fù)風(fēng)險(xiǎn)

java.lang.IllegalStateException: Duplicate key xx

當(dāng) List 中有重復(fù)值的時(shí)候，使用 Collectors.toMap() 轉(zhuǎn)為 Map 時(shí)，會(huì)報(bào)：java.lang.IllegalStateException: Duplicate key xx，例如

List<SdsTest> sdsTests = new ArrayList<>();
    SdsTest sds1 = new SdsTest("aaa","aaa");
    SdsTest sds2 = new SdsTest("aaa","ccc");

    sdsTests.add(sds1);
    sdsTests.add(sds2);

    Map<String, String> map = sdsTests.stream().collect(Collectors.toMap(SdsTest::getName, SdsTest::getAge));    
    System.out.println(map.toString());

---------
運(yùn)行錯(cuò)誤：
Exception in thread "main" java.lang.IllegalStateException: Duplicate key aaa
        at java.util.stream.Collectors.lambda$throwingMerger$92(Collectors.java:133)
        at java.util.stream.Collectors$$Lambda$6/1177096266.apply(Unknown Source)
        at java.util.HashMap.merge(HashMap.java:1245)
            .....

原因是兩個(gè)參數(shù)的toMap(xx, xx)方法, 當(dāng)出現(xiàn)重復(fù)key觸發(fā)merge時(shí)，直接拋出異常。源碼如下：

public static <T, K, U>
Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
                            Function<? super T, ? extends U> valueMapper) {
     // 注意這里的throwingMerger()
     return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new);
}

接下來(lái)我們看throwingMerger() 方法：【注意方法注釋】

/**
 * Returns a merge function, suitable for use in
 * {@link Map#merge(Object, Object, BiFunction) Map.merge()} or
 * {@link #toMap(Function, Function, BinaryOperator) toMap()}, which always
 * throws {@code IllegalStateException}.  This can be used to enforce the
 * assumption that the elements being collected are distinct.
 *
 * @param <T> the type of input arguments to the merge function
 * @return a merge function which always throw {@code IllegalStateException}
 */
 private static <T> BinaryOperator<T> throwingMerger() {
      return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); };
 }
...

解決方法

業(yè)務(wù)控制盡量不要出現(xiàn)重復(fù)值
出現(xiàn)重復(fù) key 時(shí)，使用后面的 value 覆蓋前面的 value

SdsTest sds1 = new SdsTest("aaa","aaa");
SdsTest sds2 = new SdsTest("bbb","bbb");
SdsTest sds3 = new SdsTest("aaa","ccc");

sdsTests.add(sds1);
sdsTests.add(sds2);
sdsTests.add(sds3);

// 寫(xiě)法一
Map<String, String> nmap = sdsTests.stream().collect(HashMap::new,(k, v) -> k.put(v.getName(), v.getAge()), HashMap::putAll);
System.out.println("nmap->:" + nmap.toString());

// 寫(xiě)法二
Map<String, String> nmap1 = sdsTests.stream().collect(Collectors.toMap(SdsTest::getName, SdsTest::getAge, (k1, k2) -> k2));
System.out.println("nmap1->:" + nmap1.toString());
...
----------------------
輸出：
nmap->:{aaa=ccc, bbb=bbb}
nmap1->:{aaa=ccc, bbb=bbb}

出現(xiàn)重復(fù) key 時(shí)，把對(duì)應(yīng)的 value 拼接起來(lái)

...
Map<String, String> nmap1 = sdsTests.stream().collect(Collectors.toMap(SdsTest::getName, SdsTest::getAge, (k1, k2) -> k1 + "," + k2));
System.out.println("nmap1->:" + nmap1.toString());
...
----------------
輸出：
nmap1->:{aaa=aaa,ccc, bbb=bbb}

把重復(fù) key 的值拼成一個(gè)集合

......
Map<String, List<String>> map = sdsTests.stream().collect(Collectors.toMap(SdsTest::getName,
    s -> {
        List<String> ages = new ArrayList<>();
        ages.add(s.getAge());
        return ages;
    },
    (List<String> v1, List<String> v2) -> {
        v1.addAll(v2);
        return v1;
    }));
System.out.println("map->"+map.toString());
------------
輸出：
map->{aaa=\[aaa, ccc\], bbb=\[bbb\]}

建議：