目的
将 List 转为 Map,如果有多个值对应同一个key,则保留最后一个。
一、准备
1⃣️、构造几个user对象,转为一个user的List,注意其中user2
与user0
的id是相同的
/**
* @description: 用户信息
* @author: wx
* @create: 2019-09-15 18:27
*/
public class User {
private Integer id;
private String userName;
private String password;
public User() {
}
public User(Integer id, String password) {
this.id = id;
this.password = password;
}
}
public static void main(String[] args) {
User user0 = new User(0, "000");
User user1 = new User(1, "111");
User user2 = new User(0, "222");
List<User> list = Arrays.asList(user0, user1, user2);
}
2⃣️、然后将List转为,以id为key,整个user对象为value的Map,加入如下代码
Map<Integer, User> collect = list.stream()
.collect(Collectors.toMap(User::getId, listSub -> listSub));
3⃣️、会发现报如下异常
Exception in thread "main" java.lang.IllegalStateException:
Duplicate key User(id=0, userName=null, password=000)
二、解决办法
修改代码List转Map代码如下:
Map<Integer, User> collect = list.stream().collect(
Collectors.toMap(User::getId, listSub -> listSub,
(listSubOld, listSubNew) -> listSubNew)
);
得到的结果:
{
0=User(id=0, userName=null, password=222),
1=User(id=1, userName=null, password=111)
}
三、分析
1、打开Collectors
源码看,会发现toMap
一共有三个重写的方法,其中根方法如下,注意看看中文注释:
/**
* @param keyMapper a mapping function to produce keys
* 指定key
* @param valueMapper a mapping function to produce values
* 指定values
* @param mergeFunction a merge function, used to resolve collisions between
* values associated with the same key, as supplied
* to {@link Map#merge(Object, Object, BiFunction)}
* 一个合并函数,有相同的key的时候如何选取value
* @param mapSupplier a function which returns a new, empty {@code Map} into
* which the results will be inserted
* 一个函数,它返回一个新的、带有结果的空Map
* 指定返回的是HashMap or ConcurrentHashMap等等
*/
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);
}
2、而我一开始用的是如下重写的toMap:
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);
}
可以看到使用了throwingMerger()
这样一个合并的方法,再看看这方法的源码:
private static <T> BinaryOperator<T> throwingMerger() {
return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); };
}
可以看到刚刚异常抛出的信息了。
总结一下:就是一旦出现一个key值有两个对应的value,就需要进行合并处理,但是默认的合并方法是throwingMerger()
,只要需要合并就抛出异常。
那么就找到解决办法了,不使用默认的方法,自己写一个合并的方法。
如上就是lambda表达式(listSubOld, listSubNew) -> listSubNew
,就是两个输入参数(一个相同的key对应的两个value),返回后一个值。