首页 > 解决方案 > Java Stream - 在过滤器中抛出异常并返回一个列表

问题描述

我创建了一个 Java 流:

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

上面的代码返回List<Student>满足filter检查的 a ,但我想为不满足此检查的学生/学生列表抛出异常。我试着做这样的事情,

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

但是,上面的代码片段返回单个学生对象而不是列表。我对这个Stream概念比较陌生,因此非常感谢您对此提供一些帮助。

标签: javalistexceptionlambdajava-stream

解决方案


最简单的方法是简单地检查:

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;
}

如果你真的需要坚持使用流 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());

推荐阅读