使用 Java Streams 过滤嵌套列表

Question

我有一个嵌套列表上的过滤。考虑包含list＆lt; b＆gt;的lt; a＆gt;，其中包含list> list＆lt; c＆gt;。我需要过滤，返回所有包含满足给定条件的B的A对象，至少一个满足给定条件的C。

我在下面有一个人为的示例，基本上是在实际示例中尝试实现的工作。请列出每个学校的城市学校清单，每个学科都有许多学科，并且有许多老师。我想保留一个学科的学校列表。主题名='数学'，在这个主题中至少有一位老师。 65。

我已经使用自定义谓词实现了此功能，因此在我的示例中我也做了一个。 我遇到了一个错误，“无法从布尔人转换为流式传输”，但是说实话，在这种情况下我正在挣扎。

    @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

由于您正在寻找“学校”作为输出，因此不需要将流映射/平面映射到另一个对象中。过滤功能必须足够智能，能够根据您的要求遍历嵌套集合并确认是否存在anyMatch。

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

// 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());