asarkar / akka Goto Github PK

@Test
public void partitioning() throws InterruptedException {
    final int N = 10;
    Flux<Integer> source = Flux.range(1, 10000).share();
    // partition source into publishers
    Publisher<Integer>[] publishers = new Publisher[N];
    for (int i = 0; i < N; i++) {
        final int idx = i;
        publishers[idx] = source.filter(v -> v % N == idx);
    }
    // create ParallelFlux each 'rail' containing single partition
    ParallelFlux.from(publishers)
            // schedule partitions into different threads
            .runOn(Schedulers.newParallel("proc", N))
            // process each partition in its own thread, i.e. in order
            .map(it -> {
                String threadName = Thread.currentThread().getName();
                Assert.assertEquals("proc-" + (it % 10 + 1), threadName);
                return it;
            })
            // collect results
            .sequential()
            .publishOn(Schedulers.newSingle("subscriber"))
            .subscribe(it -> {
                String threadName = Thread.currentThread().getName();
                Assert.assertEquals("subscriber-1", threadName);
            });
    Thread.sleep(1000);
}