使用Java Stream从一组集合中收集所有对象

Question

我尝试学习Java Streams并尝试从HashSet<Person>获取HashSet<SortedSet<Person>>。

HashSet<Person> students = getAllStudents();
HashSet<SortedSet<Person>> teachersForStudents = students.stream().map(Person::getTeachers).collect(Collectors.toCollection(HashSet::new));
HashSet<Person> = //combine teachers and students in one HashSet

我真正希望将所有教师和所有学生合并为一个HashSet<Person>。当我收集我的信息流时，我想我做错了什么？

Answer 1

您可以flatMap将每个学生分成由学生和他们的老师组成的小组：

HashSet<Person> combined = 
    students.stream()
            .flatMap(student -> Stream.concat(Stream.of(student), student.getTeachers().stream()))
            .collect(Collectors.toCollection(HashSet::new));

concat用于连接教师的Stream，由学生自己组成的Stream，使用of获得。