Java 8 Stream API — Java 9 Collectors.flatMapping, переписанный в Java 8

Я узнал о новой функции, так как java- 9 под названием Collectors.flatMapping, который происходит после группировки или разделения. Например, (пример взят из здесь ):

List<List<Integer>> list = Arrays.asList(
    Arrays.asList(1, 2, 3, 4, 5, 6), 
    Arrays.asList(7, 8, 9, 10));

Map<Integer, List<Integer>> map =list.stream()
    .collect(Collectors.groupingBy(
         Collection::size,
         Collectors.flatMapping(
             l -> l.stream().filter(i -> i % 2 == 0),
             Collectors.toList())));

{4=[8, 10], 6=[2, 4, 6]}

Это довольно элегантный способ, использующий всего 3 коллектора. Мне нужно переписать сборщик в java-8 где еще не поддерживается. Моя попытка использовать 6 коллекторов, что является довольно обширным использованием, и я не могу найти способ использовать меньше из них:

Map<Integer, List<Integer>> map = list.stream()
    .collect(Collectors.groupingBy(
        Collection::size,
        Collectors.collectingAndThen(
            Collectors.mapping(
                l -> l.stream().filter(i -> i % 2 == 0).collect(Collectors.toList()),
                Collectors.toList()),
            i -> i.stream().flatMap(j -> j.stream()).collect(Collectors.toList()))));

Есть ли более короткий лучший способ, используя только java-8?


java java-8 collectors java-9 java-stream
person Nikolas Charalambidis    schedule 21.01.2019    source источник
comment
@Michael согласился, вы должны опубликовать версию этого обратного порта на самом деле, был бы рад проголосовать. Я думал, что у Хольгера уже есть один, но я не могу найти его в переполнении стека...   -  person Eugene    schedule 21.01.2019
comment
нашел...   -  person Eugene    schedule 21.01.2019

Ответы (2)


arrow_upward
11
arrow_downward

Только для этого конкретного случая, я думаю, это будет более простая версия:

Map<Integer, List<Integer>> map =
        list.stream()
            .collect(Collectors.toMap(
                Collection::size,
                x -> x.stream().filter(y -> y % 2 == 0).collect(Collectors.toList())
            ));

Если бы было задействовано слияние (две коллекции одинакового размера), я бы добавил функцию merge, которая довольно тривиальна:

 Map<Integer, List<Integer>> map =
        list.stream()
            .collect(Collectors.toMap(
                Collection::size,
                x -> x.stream().filter(y -> y % 2 == 0).collect(Collectors.toCollection(ArrayList::new)),
                (left, right) -> {
                    left.addAll(right);
                    return left;
                }
            ));

В остальном, я согласен с Майклом в этом комментарии, это не так сложно обратно портировать на java-8.

person Eugene    schedule 21.01.2019

arrow_upward
18
arrow_downward

Я бы просто перенес flatMapping. Для этого требуется только 2 метода и 1 класс, без других зависимостей.

Кроме того, когда придет время перейти на Java 9, вы можете просто отказаться от своей версии и заменить любое ее использование соответствующей версией.

Следующий код взят из JDK. Я этого не писал. Я проверил это на вашем примере, и он возвращает тот же результат.

class Nikollectors
{
   public static <T, U, A, R> Collector<T, ?, R> flatMapping(Function<? super T, ? extends Stream<? extends U>> mapper, Collector<? super U, A, R> downstream) {
        BiConsumer<A, ? super U> downstreamAccumulator = downstream.accumulator();
        return new CollectorImpl<>(downstream.supplier(),
            (r, t) -> {
                try (Stream<? extends U> result = mapper.apply(t)) {
                    if (result != null)
                        result.sequential().forEach(u -> downstreamAccumulator.accept(r, u));
                }
            },
            downstream.combiner(), downstream.finisher(),
            downstream.characteristics());
    }

   private static class CollectorImpl<T, A, R> implements Collector<T, A, R>
   {
        private final Supplier<A> supplier;
        private final BiConsumer<A, T> accumulator;
        private final BinaryOperator<A> combiner;
        private final Function<A, R> finisher;
        private final Set<Characteristics> characteristics;

        CollectorImpl(Supplier<A> supplier,
                      BiConsumer<A, T> accumulator,
                      BinaryOperator<A> combiner,
                      Function<A,R> finisher,
                      Set<Characteristics> characteristics) {
            this.supplier = supplier;
            this.accumulator = accumulator;
            this.combiner = combiner;
            this.finisher = finisher;
            this.characteristics = characteristics;
        }

        CollectorImpl(Supplier<A> supplier,
                      BiConsumer<A, T> accumulator,
                      BinaryOperator<A> combiner,
                      Set<Characteristics> characteristics) {
            this(supplier, accumulator, combiner, castingIdentity(), characteristics);
        }

        @Override
        public BiConsumer<A, T> accumulator() {
            return accumulator;
        }

        @Override
        public Supplier<A> supplier() {
            return supplier;
        }

        @Override
        public BinaryOperator<A> combiner() {
            return combiner;
        }

        @Override
        public Function<A, R> finisher() {
            return finisher;
        }

        @Override
        public Set<Characteristics> characteristics() {
            return characteristics;
        }
    }

   private static <I, R> Function<I, R> castingIdentity() {
       return i -> (R) i;
   }
}

Пример использования:

Map<Integer, List<Integer>> map =list.stream()
    .collect(Collectors.groupingBy(
         Collection::size,
         Nikollectors.flatMapping( // <<<
             l -> l.stream().filter(i -> i % 2 == 0),
             Collectors.toList()
        )
    )
);
person Michael    schedule 21.01.2019
comment
вы заслуживаете двух голосов за Nikollectors... вам также следует добавить пример использования, и это может быть статический фабричный метод, который вернет этот Collectors::flatMapping, чтобы сделать его полным примером ИМХО - person Eugene; 21.01.2019
comment
Вы можете шутить, но в моем проекте у нас есть пакет xxx.HolgerUtil — просто чтобы быть уверенным, откуда это взято, без шуток. :) - person Eugene; 21.01.2019
comment
Я застрял в Java 8, и это ответ, который я не осознавал, что искал. Спасибо! - person moraleboost; 06.10.2020