Java 8 Streams - Map Несколько объектов одного и того же типа для списка с использованием потоков

Question

Возможно ли выполнить описанные ниже шаги с использованием потоков лучше?

Set<Long> memberIds = new HashSet<>(); 
marksDistribution.parallelStream().forEach(marksDistribution -> { 
     memberIds.add(marksDistribution.getStudentId()); 
     memberIds.add(marksDistribution.getTeacherId()); 
     }); 

instanceDistribution.getStudentId() и instanceDistribution.getTeacherId() оба типа Long.

Возможно, этот вопрос задан, но я не могу его понять. В простых да или нет. Если да/нет, то как и бит объяснять. И если это возможно, обсудите эффективность.

Answer 1

Вы можете использовать 3-арг версию от сбора:

Set<Long> memberIds = 
    marksDistribution.parallelStream() 
        .collect(HashSet::new, 
           (s, m) -> { 
            s.add(m.getStudentId()); 
            s.add(m.getTeacherId()); 
           }, Set::addAll); 

Ваша текущая версия может приводить к неправильным результатам, поскольку вы добавляете параллельные элементы в коллекцию, не связанную потоками. Таким образом, возможно, что у вас есть несколько раз то же значение в наборе.

Answer 2

Да, вы можете использовать flatMap для отображения одного элемента из ваших Stream в Stream из нескольких элементов, а затем сплющить их в один Stream:

Set<Long> memberIds = 
    marksDistribution.stream() 
        .flatMap (marksDistribution -> Stream.of(marksDistribution.getStudentId(), marksDistribution.getTeacherId())) 
        .collect(Collectors.toSet());