Examples with Schedulers reactor.core.scheduler.Schedulers used on opensource projects

Example 1 with Schedulers

use of reactor.core.scheduler.Schedulers in project reactor-core by reactor.

the class EmitterProcessorTest method testThreadAffinity.

@Test
public void testThreadAffinity() throws InterruptedException {
    int count = 10;
    Scheduler[] schedulers = new Scheduler[4];
    CountDownLatch[] latches = new CountDownLatch[schedulers.length];
    for (int i = 0; i < schedulers.length; i++) {
        schedulers[i] = Schedulers.newSingle("scheduler" + i + '-');
        int expectedCount = i == 1 ? count * 2 : count;
        latches[i] = new CountDownLatch(expectedCount);
    }
    EmitterProcessor<Integer> processor = EmitterProcessor.create();
    processor.publishOn(schedulers[0]).share();
    processor.publishOn(schedulers[1]).subscribe(i -> {
        assertThat(Thread.currentThread().getName().contains("scheduler1")).isTrue();
        latches[1].countDown();
    });
    for (int i = 0; i < count; i++) processor.onNext(i);
    processor.publishOn(schedulers[2]).map(i -> {
        assertThat(Thread.currentThread().getName().contains("scheduler2")).isTrue();
        latches[2].countDown();
        return i;
    }).publishOn(schedulers[3]).doOnNext(i -> {
        assertThat(Thread.currentThread().getName().contains("scheduler3")).isTrue();
        latches[3].countDown();
    }).subscribe();
    for (int i = 0; i < count; i++) processor.onNext(i);
    processor.onComplete();
    for (int i = 1; i < latches.length; i++) assertThat(latches[i].await(5, TimeUnit.SECONDS)).isTrue();
    assertThat(latches[0].getCount()).isEqualTo(count);
}