Java8 Collectors.toMap的坑

Java8 Collectors.toMap的坑

按照常规思维，往一个map里put一个已经存在的key，会把原有的key对应的value值覆盖，然而通过一次线上问题，发现java8中的Collectors.toMap反其道而行之，它默认给抛异常，抛异常...

线上业务代码出现Duplicate Key的异常，影响了业务逻辑，查看抛出异常部分的代码，类似以下写法：

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

然后list里面有id相同的对象，结果转map的时候居然直接抛异常了。。查源码发现toMap方法默认使用了个throwingMerger

public static

Collector> toMap(Function super T, ? extends K> keyMapper,

Function http://super T, ? extends U> valueMapper) {

return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new);

}

private static throwingMerger() {

return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); };

}

那么这个throwingMerger是哪里用的呢？

public static >

Collector toMap(Function super T, ? extends K> keyMapper,

Function super T, ? extends U> valueMapper,

BinaryOperator mergeFunction,

Supplier mapSupplier) {

BiConsumer 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已存在，http://那么就调用传进来的方法。而throwingMerger的做法就是抛了个异常。所以到这里就可以知道写的代码为什么呲了。。

如果不想抛异常的话，自己传进去一个方法即可，上述代码可以改成：

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

这样就做到了使用新的value替换原有value。

写代码调方法时，多看源码实现，注意踩坑！

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。