我创建了一个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
,但是我想为不满足此检查条件的一个学生/一组学生抛出异常。我尝试做这样的事情,
return students.stream().filter(s -> s.getGradeId().equals(grade.getGid())).findAny()
.orElseThrow(() -> new EntityNotFoundException(ENTITY_NOT_FOUND));
但是,上面的代码片段返回单个学生对象而不是列表。我对Stream
概念比较陌生,因此非常感谢您提供一些帮助。
答案 0 :(得分:4)
最简单的方法就是检查:
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());