.collect (Collectors.toList()) и потоки по Java-методу

Question

У меня есть коллекция (как hashmap) of the Doctors, в общий класс больницы.

Map<Integer, Doctor> doctors = new HashMap<Integer, Doctor>(); 

Для каждого врача у меня есть некоторая информация, например, в коде класса (фокус на пациентах):

public class Doctor extends Person { 
    private int id; 
    private String specialization; 
    private List<Person> patients = new LinkedList<Person>(); 

Моя цель состоит в том, чтобы написать эту функцию, возвращение занятых врачей: врачей, который имеет число пациентов больше среднего.

/** 
* returns the collection of doctors that has a number of patients larger than the average. 
*/ 
Collection<Doctor> busyDoctors(){ 

    Collection<Doctor> doctorsWithManyPatients = 
      doctors.values().stream() 
      .map(doctor -> doctor.getPatients()) 
      .filter(patientsList -> { return patientsList.size() >= AvgPatientsPerDoctor; }) 
      .collect(Collectors.toList()); 

    return null; 
} 

Я хочу использовать потоки, как указано выше, для выполнения этой операции. Проблема заключается в методе collect, поскольку в этой точке использования doctorsWithManyPatients имеет тип List<Collection<Person>>, а не Collection<Doctor>. Как я мог это сделать?

Предположим, что AvgPatientsPerDoctor уже определен где-то.

Answer 1

Вам не нужно использовать map (Doctor -> List<Person>), он будет использоваться в filter:

doctors 
    .values() 
    .stream() 
    .filter(d -> d.getPatients().size() >= AvgPatientsPerDoctor) 
    .collect(Collectors.toList()); 

В вашем случае, map(doctor -> doctor.getPatients()) возвращается Stream<List<Person>> и вы должны преобразовать его в Stream<Doctor> снова после того, как filter Инг и перед вызовом метод collect.

---

Существует другой способ, который не самый лучший. Имейте в виду, что он изменяет коллекцию происхождения.

doctors.values().removeIf(d -> d.getPatients().size() < AvgPatientsPerDoctor);