java - 在 java flux 中按对象属性分组

Question

给定以下数据结构 Data和 Flux<Data>基于某些属性实现分组到一系列列表的惯用方法是什么:

import org.reactivestreams.Publisher;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.util.List;
import java.util.Objects;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.Function;
import java.util.function.Predicate;

class Scratch {
    private static class Data {
        private Integer key;
        private String value;

        public Data(Integer key, String value) {
            this.key = key;
            this.value = value;
        }

        public Integer getKey() {
            return key;
        }

        public String getValue() {
            return value;
        }

        public static Data of(Integer key, String value) {
            return new Data(key, value);
        }

        @Override
        public String toString() {
            return value;
        }
    }

    public static void main(String[] args) {
        Flux<Data> test = Flux.just(
                Data.of(1, "Hello"),
                Data.of(1, "world"),
                Data.of(2, "How"),
                Data.of(2, "are"),
                Data.of(2, "you"),
                Data.of(3, "Bye"));
        test.bufferUntil(new Predicate<Data>() {
            Integer prev = null;
            @Override
            public boolean test(Data next) {
                boolean collect = prev != null && !Objects.equals(prev, next.getKey());
                prev = next.getKey();
                return collect;
            }
        }, true).subscribe(e -> System.out.println(e.toString()));
    }
} 

输出:

[Hello, world]
[How, are, you]
[Bye]

我知道 Flux 上的 groupBy 函数，但这又给了我一个 Flux，而不是列表。我上面描述的当前解决方案有效，但感觉不是 100% 地道，因为我不得不使用匿名类而不是 lambda。我本可以在 lambda 之外使用 lambda 和 AtomicReference，但这也不是 100% 正确的。有什么建议吗？

Answer 1

您还可以使用 collectMultimap这让你有 Map<K, Collection<T> .在这种情况下 collectMultimap将返回:Mono<Map<Integer,Collection<Data>>> :

 test.collectMultimap( Data::getKey )
     .subscribe( dataByKey -> System.out.println( dataByKey.toString() ) );

输出:

{1=[Hello, world], 2=[How, are, you], 3=[Bye]}