最近在用jdk8的stream 的Collectors.toMap 方法中,出现NullPointerException,查看源码发现用的是HashMap的merge方法,如果value为null,则抛出空指针异常。而key不允许有重复存在的,否则抛出IllegalStateException异常,并提示有重复的key已存在。
List<UserInfo> userInfos = new ArrayList<>();
for(int i=0;i<3;i++){
userInfos.add(new UserInfo("uuid"+i,null));
}
Map<String, String> map = userInfos.stream().collect(Collectors.toMap(k->k.getUuid(),v->v.getName()));
抛出NullPointerException;
List<UserInfo> userInfos = new ArrayList<>();
for(int i=0;i<1;i++){
userInfos.add(new UserInfo(null,"uuid"+i));
}
Map<String, String> map = userInfos.stream().collect(Collectors.toMap(k->k.getUuid(),v->v.getName()));
System.out.println(map);
正常输出;
List<UserInfo> userInfos = new ArrayList<>();
for(int i=0;i<3;i++){
userInfos.add(new UserInfo(null,"uuid"+i));
}
Map<String, String> map = userInfos.stream().collect(Collectors.toMap(k->k.getUuid(),v->v.getName()));
System.out.println(map);
抛出IllegalStateException;
如果要解决这个问题,可以加两个过滤:
Map<String, String> map = userInfos.stream().filter(distinctByKey(u->u.getUuid())).filter(u->u.getUuid()!=null&&u.getName()!=null).collect(Collectors.toMap(k->k.getUuid(),v->v.getName()));
过滤器配置
static <T> Predicate<T> distinctByKey(Function<? super T, ?> keyExtractor) {
Map<Object,Boolean> seen = new ConcurrentHashMap<>();
return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null;
}
这个重复过滤的只保存第一个已存在的key,根据需求可以进行更改
或者下面这个如果重复则取第一个key的value
Map<String, String> map = userInfos.stream().filter(u->u.getUuid()!=null&&u.getName()!=null).collect(Collectors.toMap(k->k.getUuid(),v->v.getName(),(k1,k2)->k1));
或者下面这个如果重复则取最后一个key的value
Map<String, String> map = userInfos.stream().filter(u->u.getUuid()!=null&&u.getName()!=null).collect(Collectors.toMap(k->k.getUuid(),v->v.getName(),(k1,k2)->k2));
或者不用stream处理。