Java Stream - выбросить исключение в фильтр и вернуть список - PullRequest
Новая
       6

Java Stream - выбросить исключение в фильтр и вернуть список

2 голосов
/ 16 апреля 2020

Я создал поток 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, поэтому буду очень признателен за помощь.

1 Ответ

4 голосов
/ 16 апреля 2020

Самый простой способ - просто проверить: 

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());
...