Java Stream - 在過濾器中拋出異常並返回一個列表

Question

我創建了一個 Java stream：

List<Student> getStudentWithSameGrade(List<Student> students, Grade grade) {
    return students.stream().filter(s -> s.getGradeId().equals(grade.getGid())).collect(Collectors.toList());
}

上面的代碼返回一個滿足filter檢查的List<Student>但我想為不滿足此檢查的學生/學生列表拋出異常。 我試着做這樣的事情，

return students.stream().filter(s -> s.getGradeId().equals(grade.getGid())).findAny()
                .orElseThrow(() -> new EntityNotFoundException(ENTITY_NOT_FOUND));

但是，上面的代碼片段返回一個學生 object 而不是一個列表。 我對Stream概念相對較新，因此非常感謝您對此的幫助。

Answer 1

最簡單的方法是簡單地檢查：

List<Student> getStudentWithSameGrade(List<Student> students, Grade grade) {
    List<Student> students = students.stream().filter(s -> s.getGradeId().equals(grade.getGid())).collect(Collectors.toList());
    if (students.isEmpty()) throw new EntityNotFoundException(ENTITY_NOT_FOUND);
    else return students;
}

如果你真的需要堅持 stream api，你可以定義自己的收集器：

public static <T> Collector<T, ?, T> toNonEmptyList() {
    return Collectors.collectingAndThen(
        Collectors.toList(),
        list -> {
            if (list.isEmpty()) {
                throw new IllegalStateException();
            }
            return list;
        }
    );
}

接着：

return students.stream()
    .filter(s -> s.getGradeId().equals(grade.getGid()))
    .collect(toNonEmptyList());