jdk8的straem中的Collectors.toMap报NullPointerException

最近在用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处理。

猜你喜欢

转载自blog.csdn.net/qq_34308732/article/details/86699152
今日推荐