如何在flatmap中过滤流中的空对象

Question

我已经为NPE保护了第一个流，但是第二个流也可以抛出NPE，下面是代码片段：

 Set<TestObj> vals = Optional.ofNullable(testObjs)
                    .orElse(Collections.emptySet())
                    .stream()
                    .filter(Objects::nonNull)
                    .flatMap(testObj -> getObjVal(testObj.getId()).getObj().stream())
                    .collect(Collectors.toSet());

如果getObjVal（String id）返回null，第二个流将抛出NPE。

下面的代码阻止了第二次NPE，但我觉得这不是一个好方法：

Set<TestObj> vals = Optional.ofNullable(testObjs)
                    .orElse(Collections.emptySet())
                    .stream()
                    .filter(Objects::nonNull)
                    .filter(testObj -> getObjVal(testObj.getId()) != null)
                    .flatMap(testObj->getObjVal(testObj.getId()).getObj().stream())
                    .collect(Collectors.toSet());

需要在另一个地方使用VAL，如果VAL不为空，将在该地方进行后续调用。有没有更好的方法来防止NPE在这里，我尝试使用可选的内部平面图，但它抛出语法错误。

Answer 1

您可以删除对getObjVal的重复调用，方法是在过滤器之前将其作为map执行。

Set<TestObj> vals = Optional.ofNullable(testObjs)
                    .orElse(Collections.emptySet())
                    .stream()
                    .filter(Objects::nonNull)
                    .map(testObj -> getObjVal(testObj.getId()))
                    .filter(Objects::nonNull)
                    .map(TestObject::getObj)
                    .flatMap(Collection::stream)
                    .collect(Collectors.toSet());