Collectors.toMap空指针问题
在工作中遇到了一个List转Map的时候的一个NullPointException.
情形很简单，问题出在Collectors.toMap，当key值冲突的时候理论上会按照我们的代码来替换value，但是这里有个小坑
|
1
|
list.stream().collect(Collectors.toMap(it -> it.getCategoryId(), it -> it.getCategoryImage() ,(k1,k2) -> k2)); |
可以看到map在key值冲突merge的时候会要求新的value不能为null.
这意味着，只要传入了(k1,k2) -> k2处理key冲突的function，那么当value里存在Null的时候必然会抛NullPointException
Collectors.toMap的坑
按照常规思维，往一个map里put一个已经存在的key，会把原有的key对应的value值覆盖，然而通过一次线上问题，发现Java8中的Collectors.toMap反其道而行之，它默认给抛异常，抛异常...
线上业务代码出现Duplicate Key的异常，影响了业务逻辑，查看抛出异常部分的代码，类似以下写法：
|
1
|
Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName)); |
然后list里面有id相同的对象，结果转map的时候居然直接抛异常了。。查源码发现toMap方法默认使用了个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)); };} |
那么这个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);} |
这里传进去的是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已存在，那么就调用传进来的方法。而throwingMerger的做法就是抛了个异常。所以到这里就可以知道写的代码为什么呲了。。
如果不想抛异常的话，自己传进去一个方法即可，上述代码可以改成：
|
1
|
Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName,(oldValue, newValue) -> newValue)); |
这样就做到了使用新的value替换原有value。
写代码调方法时，多看源码实现，注意踩坑！
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/weixin_43944305/article/details/119913697
