如何限制groupBy java流

Question

这是我在课堂上名为Course的课程模型：

public class Course{
    private int courseId;
    private String courseName;
    private Teacher teacher;
}

这是我在课堂上名为Teacher的老师模型：

public class Teacher{
    private int teacherId;
    private String name;
}

我想获得Map<String, List<Course>>，但如果teacherId 重复，只需将Course 添加到地图列表中。

我正在使用groupBy

Map<Integer, List<Course>>  result = courses.stream()
        .collect(Collectors.groupingBy(c -> c.getTeacher().getTeacherId(), Collectors.toList()));

它按预期给出了结果。

但是我想在这里limit，一旦找到5位老师就停止处理并返回结果。

怎么办？？

Answer 1

没有直接支持这一点，因为停止和使用可能不完整的数据是相当不寻常的。

完全收集前五组的直接解决方案是

Set<Integer> firstFive = courses.stream()
    .map(c -> c.getTeacher().getTeacherId())
    .distinct().limit(5)
    .collect(Collectors.toSet());
Map<Integer, List<Course>> result = courses.stream()
    .filter(c -> firstFive.contains(c.getTeacher().getTeacherId()))
    .collect(Collectors.groupingBy(c -> c.getTeacher().getTeacherId()));

到此，前五个教师 ID 的 Course 列表已完成。

---

遇到第 5 个教师 ID 后真正停止的解决方案，使用循环会更简单：

Map<Integer, List<Course>> result = new HashMap<>();
for(Course c: courses) {
    result.computeIfAbsent(c.getTeacher().getTeacherId(), x -> new ArrayList<>()).add(c);
    if(result.size() == 5) break;
}

但是收集Courses 的列表并没有多大意义，因为事后您不能信任这些列表。请记住，即使源列表的最后一个元素也可能属于第一个遇到的教师 ID，因此即使您只对一位教师的完整课程列表感兴趣，您也需要处理整个列表。

Answer 2

不确定这是否是您要问的。

Map<Integer, List<Course>> map = new HashMap<>();

courses.stream().filter(course -> map.keySet().size() < 10)
    .forEach(entry -> {
       // The code below can be simplified
      int teacherId = entry.getTeacher().getTeacherId();
      if(map.get(teacherId) != null)
        map.get(teacherId).add(entry);
      else
        map.put(teacherId, Lists.newArrayList(entry));
    });