java - 使用 java8 流从多级内部列表中检索数据

Question

我有 Employee 列表，每个员工都有一个 Department 列表，每个部门都有一个 location 列表，我想根据某些过滤条件获取位置列表。

这是我的代码 -

for (Employee empl : employeeList) {
    if(empl.getEmployeeName().equals("XXX")) {
        for (Department dept : empl.getDepartmets()) {
            if(dept.getDepartmentName().equals("Sales")) {
                return dept.getLocations();
            }
        }
    }
}

我想在 java8 steam api 上处理这个，有人能帮忙吗？

Answer 1

employeeList.stream()
          .filter(x -> "XXX".equals(x.getEmployeeName()))
          .flatMap(x -> x.getDepartmets().stream().filter(y -> "Sales".equals(y.getDepartmentName())))
          .findFirst()
          .map(Department::getLocations)
          .orElse(Collections.emptyList());

唯一的一点是 flatMap 在 java-10 之前并不是真正的懒惰。如果你真的关心这个，我假设 locations 是这里某种类型的 List。