使用 Java Streams 过滤嵌套列表

Question

我需要过滤一些嵌套列表。 考虑一个包含List<B>的List<A> ，其中包含List<C> 。 我需要过滤，返回包含至少一个满足给定条件的 B 和至少一个满足给定条件的 C 的所有 A 对象。

我在下面有一个人为的例子，它基本上是在做我在我的实际例子中试图实现的事情。 拿一个城市的学校列表，每个学校有很多学科，每个学科有很多老师。 我想保留一份包含 subject.subjectName = 'Math' 的学校列表，并且该科目中至少有一名教师 teacher.age > 65。

我已经使用自定义谓词实现了这个，所以我也在我的示例中做了一个 this 。 我收到“无法从 boolean 转换为 Stream”的错误，但老实说，我在这种情况下很挣扎。

    @Getter
    @Setter
    class School {
        private String schoolId;
        private List<Subject> classes;
        
    }
        
    @Getter
    @Setter
    class Subject {
        String subjectName;
        List<Teacher> teachers;
    }
    
    @Getter
    @Setter
    class Teacher {
        private String teacherName;
        private Integer age;
    }

    public class TestClass {
        public static void main( String[] args ){
            List<School> schools;
            
            // Get All schools where A mathTeacher is over retirement age
            List<School> schoolsWithRetirementAgeMathTeachers = schools.stream()
                    .filter(school -> null != school.getClasses())
                    .flatMap(school -> {
                        return school.getClasses().stream()
                                .filter(subject -> subject.getSubjectName().equalsIgnoreCase("Math"))
                                .filter(subject -> null != subject.getTeachers())
                                .flatMap(subject -> {
                                    return subject.getTeachers().stream()
                                            .filter(teacher -> teacher != null)
                                            .anyMatch(isRetirementAge);
                                });
                    }).collect(Collectors.toList());
                    
        }
                
        public static Predicate<Teacher> isRetirementAge = teacher -> teacher.getAge() > 65;
    
    }

Answer 1

如果你试图返回父对象，你不想使用flatMap() 。 您只需要嵌套一些anyMatch()调用：

List<A> filtered = listA.stream()
        .filter(a -> a.getListB()
                .stream()
                .anyMatch(b -> testB(b) && b.getListC()
                        .stream()
                        .anyMatch(c -> testC(c))))
        .collect(Collectors.toList());

Answer 2

由于您正在寻找School作为 output，因此您不需要将map / flatmap您的 stream 平面映射到另一个 object。 filter功能必须足够智能，通过嵌套的anyMatch根据您的要求确认 88354248716628 .

主要在下面的代码中，我们过滤了所有存在名称为“数学”的科目并且其中一位教授数学的老师超过上述退休年龄的学校。

// Get All schools where a Math teacher is over retirement age
List<School> schoolsWithRetirementAgeMathTeachers = schools.stream()
        .filter(school -> school.getSubjects()
                .stream()
                .anyMatch(subject ->
                        subject.getName().equalsIgnoreCase("Math") &&
                                subject.getTeachers()
                                        .stream()
                                        .anyMatch(isRetirementAge)
                ))
        .collect(Collectors.toList());