按照常規(guī)思維，往一個(gè)map里put一個(gè)已經(jīng)存在的key，會(huì)把原有的key對應(yīng)的value值覆蓋，然而通過一次線上問題，發(fā)現(xiàn)Java8中的Collectors.toMap反其道而行之，它默認(rèn)給拋異常，拋異常...
線上業(yè)務(wù)代碼出現(xiàn)Duplicate Key的異常，影響了業(yè)務(wù)邏輯，查看拋出異常部分的代碼，類似以下寫法：
|
1
|
Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName)); |
然后list里面有id相同的對象，結(jié)果轉(zhuǎn)map的時(shí)候居然直接拋異常了。。查源碼發(fā)現(xiàn)toMap方法默認(rèn)使用了個(gè)throwingMerger
|
1
2
3
4
5
6
7
8
9
10
|
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)); };} |
那么這個(gè)throwingMerger是哪里用的呢？
|
1
2
3
4
5
6
7
8
9
10
|
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);} |
這里傳進(jìn)去的是HashMap，所以最終走的是HashMap的merge方法。merge方法里面有這么一段代碼：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
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已存在，那么就調(diào)用傳進(jìn)來的方法。而throwingMerger的做法就是拋了個(gè)異常。所以到這里就可以知道寫的代碼為什么呲了。。
如果不想拋異常的話，自己傳進(jìn)去一個(gè)方法即可，上述代碼可以改成：
|
1
|
Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName,(oldValue, newValue) -> newValue)); |
這樣就做到了使用新的value替換原有value。
寫代碼調(diào)方法時(shí)，多看源碼實(shí)現(xiàn)，注意踩坑！
到此這篇關(guān)于Java8 Collectors.toMap的坑的文章就介紹到這了,更多相關(guān)Java8 Collectors.toMap內(nèi)容請搜索服務(wù)器之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持服務(wù)器之家！
原文鏈接：https://blog.csdn.net/u013805360/article/details/82686009
