Java8 Collectors.toMap的坑

按照常規思維，往一個map里put一個已經存在的key，會把原有的key對應的value值覆蓋，然而通過一次線上問題，發現Java8中的Collectors.toMap反其道而行之，它默認給拋異常，拋異常...

 

線上業務代碼出現Duplicate Key的異常，影響了業務邏輯，查看拋出異常部分的代碼，類似以下寫法：

Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName));

 

然后list里面有id相同的對象，結果轉map的時候居然直接拋異常了。。查源碼發現toMap方法默認使用了個throwingMerger

public static <T, K, U>
Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
                                Function<? super T, ? extends U> valueMapper) {
    return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new);
}
 
 
private static <T> BinaryOperator<T> throwingMerger() {
    return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); };
}

 

那么這個throwingMerger是哪里用的呢？

public static <T, K, U, M extends Map<K, U>>
Collector<T, ?, M> toMap(Function<? super T, ? extends K> keyMapper,
                            Function<? super T, ? extends U> valueMapper,
                            BinaryOperator<U> mergeFunction,
                            Supplier<M> mapSupplier) {
    BiConsumer<M, T> accumulator
            = (map, element) -> map.merge(keyMapper.apply(element),
                                          valueMapper.apply(element), mergeFunction);
    return new CollectorImpl<>(mapSupplier, accumulator, mapMerger(mergeFunction), CH_ID);
}

 

這里傳進去的是HashMap，所以最終走的是HashMap的merge方法。merge方法里面有這么一段代碼：

if (old != null) {
    V v;
    if (old.value != null)
        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;
}

 

相信只看變量名就能知道這段代碼啥意思了。。如果要put的key已存在，那么就調用傳進來的方法。而throwingMerger的做法就是拋了個異常。所以到這里就可以知道寫的代碼為什么呲了。。

 

如果不想拋異常的話，自己傳進去一個方法即可，上述代碼可以改成：

Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName,(oldValue, newValue) -> newValue));

 

這樣就做到了使用新的value替換原有value。

 

寫代碼調方法時，多看源碼實現，注意踩坑！