Java Stream：在多个列表上的嵌套流查找抛出 NPE

Question

我在 Java 7 中有以下代码：

List<Integer> idMappers= new ArrayList<>();

//getting information from a Map<String, List<String>>
List<String> ids= idDataStore.lookupId(id); 

 for (int i = 0; i < ids.size(); i++) {

 //getting information from a Map<String, List<Integer>>
  List<Integer> mappers= idDataStore.lookupMappers(ids.get(i));

  if (mappers!= null) {
    for (int j = 0; j < x.size(); j++) {
      idMappers.add(mappers.get(j));
    }
  }
}

我正在尝试将其更改为 Streams

List<Integer> idMappers= new ArrayList<>();
idDataStore.lookupIdMappings(id).forEach(id-> {
  idDataStore.lookupSegments(id).forEach(mapper->{
    idSegments.add(segment);
  });
});

我的问题是idDataStore.lookupSegments(id) 有时会抛出 null，所以我的流中断了。如何在 Stream 中进行空检查？

Answer 1

首先，在 lambda 中使用的变量 (id) 不能与方法相同范围内的变量同名。

Lambda 表达式的参数 id 不能重新声明在封闭范围内定义的另一个局部变量。

我看到您使用嵌套的 for 循环，为什么不使用 Stream::flatMap？

idDataStore.lookupIdMappings(id).stream()
                                .map(i -> idDataStore.lookupSegments(id))
                                .filter(Objects::nonNull)
                                .flatMap(List::stream)
                                .collect(Collectors.toList());

Answer 2

只需将idDataStore.lookupSegments(id).stream().filter(Objects::notNull) 添加到您的嵌套循环中。

但是，您拥有的是 side effect（请参阅副作用部分），不推荐填充 idMappers 列表的方法。让我尝试使用flatMap进行转换

List<Integer> idMappers = idDataStore.lookupIdMappings(id)
           .stream() // stream of LookupId's
           .flatMap(idMapping -> idDataStore
                                .lookupSegments(id)
                                .stream()
                                .filter(Objects::notNull)
                                // get stream of corresponding lookupSegments
                                // and filter out all nulls
           )
           .collect(Collectors.toList());

我希望这会有所帮助。