java - 如何处理java流api中的可选对象？

Question

我正在做简单的测试。

@Test
    public void whenFilterEmployees_thenGetFilteredEmployees(){
        Integer[] empIds = {1,2,3};
        List<Optional<Employee>> employees = Stream.of(empIds)
                .map(employeeRepository::findById)
                .collect(Collectors.toList());

        List<Employee> employeeList = employees
                .stream().filter(Optional::isPresent)
                .map(Optional::get)
                .filter(e->e !=null)
                .filter(e->e.getSalary()>200000)
                .collect(Collectors.toList());

        assertEquals(Arrays.asList(arrayOfEmps[2]), employeeList);


    }

我的员工表包含数据：

1   Jeff Bezos  100000
2   Bill Gates  200000
3   Mark Zuckerberg 300000

当前测试运行成功。

如您所见，我准备了两个员工列表，即员工和员工列表

我这样做是因为findById 方法返回可选。我如何使用流 api 以便我可以简单地获取员工列表

List<Employee> employeeList= ....

Answer 1

只需合并两个流管道，即

List<Employee> employees = Stream.of(empIds)
                                 .map(employeeRepository::findById)
                                 .filter(Optional::isPresent)
                                 .map(Optional::get)
                               //.filter(e->e !=null) not needed as it's redundant
                                 .filter(e->e.getSalary()>200000)
                                 .collect(Collectors.toList());

Answer 2

我们可以使用Optional.stream（需要Java 9+）编写更短的代码：

List<Employee> employees = Stream.of(empIds)
        .flatMap(id -> employeeRepository.findById(id).stream())
        .filter(e -> e.getSalary() > 200000)
        .collect(Collectors.toList());

根据JavaDocOptional.stream

返回仅包含该值的顺序Stream，否则 返回一个空的Stream。

因此我们可以立即进行filter 操作。