Java Code Examples for reactor.core.Fuseable#ConditionalSubscriber

The following examples show how to use reactor.core.Fuseable#ConditionalSubscriber . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: FluxArrayTest.java From reactor-core with Apache License 2.0

6 votes

@Test
public void scanConditionalSubscriptionRequested() {
	@SuppressWarnings("unchecked")
	Fuseable.ConditionalSubscriber<? super Object> subscriber = Mockito.mock(MockUtils.TestScannableConditionalSubscriber.class);
	//the mock will not drain the request, so it can be tested
	Mockito.when(subscriber.tryOnNext(Mockito.any()))
	       .thenReturn(false);

	FluxArray.ArrayConditionalSubscription<Object> test =
			new FluxArray.ArrayConditionalSubscription<>(subscriber,
					new Object[]{"foo", "bar", "baz"});

	test.request(2);
	assertThat(test.scan(Scannable.Attr.REQUESTED_FROM_DOWNSTREAM)).isEqualTo(2L);
	assertThat(test.scan(Scannable.Attr.BUFFERED)).isEqualTo(3);
}

Example 2

Source File: ParallelLog.java From reactor-core with Apache License 2.0

6 votes

@Override
@SuppressWarnings("unchecked")
public void subscribe(CoreSubscriber<? super T>[] subscribers) {
	if (!validate(subscribers)) {
		return;
	}
	
	int n = subscribers.length;
	CoreSubscriber<? super T>[] parents = new CoreSubscriber[n];

	boolean conditional = subscribers[0] instanceof Fuseable.ConditionalSubscriber;

	for (int i = 0; i < n; i++) {
		if (conditional) {
			parents[i] = new FluxPeekFuseable.PeekConditionalSubscriber<>(
					(Fuseable.ConditionalSubscriber<T>)subscribers[i], log);
		}
		else {
			parents[i] = new FluxPeek.PeekSubscriber<>(subscribers[i], log);
		}
	}
	
	source.subscribe(parents);
}

Example 3

Source File: FluxUsingTest.java From reactor-core with Apache License 2.0

6 votes

@Test
  public void scanConditionalSubscriber() {
@SuppressWarnings("unchecked")
Fuseable.ConditionalSubscriber<Integer> actual = Mockito.mock(MockUtils.TestScannableConditionalSubscriber.class);
FluxUsing.UsingConditionalSubscriber<Integer, String> test =
		new FluxUsing.UsingConditionalSubscriber<>(actual, s -> {}, "", true);
      Subscription parent = Operators.emptySubscription();
      test.onSubscribe(parent);

      Assertions.assertThat(test.scan(Scannable.Attr.PARENT)).isSameAs(parent);
      Assertions.assertThat(test.scan(Scannable.Attr.ACTUAL)).isSameAs(actual);

      Assertions.assertThat(test.scan(Scannable.Attr.CANCELLED)).isFalse();
      Assertions.assertThat(test.scan(Scannable.Attr.TERMINATED)).isFalse();
      test.onComplete();
      Assertions.assertThat(test.scan(Scannable.Attr.TERMINATED)).isTrue();
      Assertions.assertThat(test.scan(Scannable.Attr.CANCELLED)).isTrue();
  }

Example 4

Source File: ParallelPeek.java From reactor-core with Apache License 2.0

6 votes

@Override
@SuppressWarnings("unchecked")
public void subscribe(CoreSubscriber<? super T>[] subscribers) {
	if (!validate(subscribers)) {
		return;
	}
	
	int n = subscribers.length;

	CoreSubscriber<? super T>[] parents = new CoreSubscriber[n];

	boolean conditional = subscribers[0] instanceof Fuseable.ConditionalSubscriber;

	for (int i = 0; i < n; i++) {
		if (conditional) {
			parents[i] = new FluxPeekFuseable.PeekConditionalSubscriber<>(
					(Fuseable.ConditionalSubscriber<T>)subscribers[i], this);
		}
		else {
			parents[i] = new FluxPeek.PeekSubscriber<>(subscribers[i], this);
		}
	}
	
	source.subscribe(parents);
}

Example 5

Source File: FluxDistinctTest.java From reactor-core with Apache License 2.0

6 votes

@Test
public void distinctPredicateThrowsConditionalOnNext() {
	IllegalStateException error = new IllegalStateException("forced failure");

	@SuppressWarnings("unchecked")
	Fuseable.ConditionalSubscriber<Integer> actualConditional = Mockito.mock(Fuseable.ConditionalSubscriber.class);
	when(actualConditional.currentContext()).thenReturn(Context.empty());

	DistinctConditionalSubscriber<Integer, Integer, Set<Integer>> conditionalSubscriber =
			new DistinctConditionalSubscriber<>(
					actualConditional,
					new HashSet<>(),
					k -> k,
					(c, k) -> { throw error; },
					Set::clear);

	conditionalSubscriber.onNext(1);

	verify(actualConditional, times(1)).onError(error);
}

Example 6

Source File: FluxMapTest.java From reactor-core with Apache License 2.0

6 votes

@Test
public void scanFuseableConditionalSubscriber() {
 @SuppressWarnings("unchecked")
 Fuseable.ConditionalSubscriber<String> actual = Mockito.mock(MockUtils.TestScannableConditionalSubscriber.class);
    FluxMapFuseable.MapFuseableConditionalSubscriber<Integer, String> test =
    		new FluxMapFuseable.MapFuseableConditionalSubscriber<>(actual, i -> String.valueOf(i));
    Subscription parent = Operators.emptySubscription();
    test.onSubscribe(parent);

    assertThat(test.scan(Scannable.Attr.PARENT)).isSameAs(parent);
    assertThat(test.scan(Scannable.Attr.ACTUAL)).isSameAs(actual);

    assertThat(test.scan(Scannable.Attr.TERMINATED)).isFalse();
    test.onError(new IllegalStateException("boom"));
    assertThat(test.scan(Scannable.Attr.TERMINATED)).isTrue();
}

Example 7

Source File: FluxPublishOnTest.java From reactor-core with Apache License 2.0

5 votes

@Test
  public void scanConditionalSubscriber() {
@SuppressWarnings("unchecked")
Fuseable.ConditionalSubscriber<Integer> actual = Mockito.mock(MockUtils.TestScannableConditionalSubscriber.class);
      FluxPublishOn.PublishOnConditionalSubscriber<Integer> test =
      		new FluxPublishOn.PublishOnConditionalSubscriber<>(actual, Schedulers.single(),
      				Schedulers.single().createWorker(), true, 123, 123, Queues.unbounded());
      Subscription parent = Operators.emptySubscription();
      test.onSubscribe(parent);

      assertThat(test.scan(Scannable.Attr.PARENT)).isSameAs(parent);
      assertThat(test.scan(Scannable.Attr.ACTUAL)).isSameAs(actual);
      assertThat(test.scan(Scannable.Attr.DELAY_ERROR)).isTrue();
      assertThat(test.scan(Scannable.Attr.PREFETCH)).isEqualTo(123);
      test.requested = 35;
      assertThat(test.scan(Scannable.Attr.REQUESTED_FROM_DOWNSTREAM)).isEqualTo(35L);
      test.queue.add(1);
      assertThat(test.scan(Scannable.Attr.BUFFERED)).isEqualTo(1);

      assertThat(test.scan(Scannable.Attr.ERROR)).isNull();
      assertThat(test.scan(Scannable.Attr.TERMINATED)).isFalse();
      test.onError(new IllegalStateException("boom"));
      assertThat(test.scan(Scannable.Attr.ERROR)).hasMessage("boom");
      assertThat(test.scan(Scannable.Attr.TERMINATED)).isTrue();

      assertThat(test.scan(Scannable.Attr.CANCELLED)).isFalse();
      test.cancel();
      assertThat(test.scan(Scannable.Attr.CANCELLED)).isTrue();
  }

Example 8

Source File: OperatorsTest.java From reactor-core with Apache License 2.0

5 votes

@Test
public void convertConditionalToConditionalShouldReturnTheSameInstance() {
	@SuppressWarnings("unchecked")
	Fuseable.ConditionalSubscriber<String> original = Mockito.mock(Fuseable.ConditionalSubscriber.class);

	Assertions.assertThat(Operators.toConditionalSubscriber(original))
	          .isSameAs(original);
}

Example 9

Source File: FluxSwitchOnFirstTest.java From reactor-core with Apache License 2.0

5 votes

@SuppressWarnings("rawtypes")
@Test
public void unitRequestsAreSerialTest() {
    @SuppressWarnings("unchecked")
    BiFunction<FluxSwitchOnFirst.AbstractSwitchOnFirstMain, CoreSubscriber, InnerOperator>[] factories = new BiFunction[] {
            (parent, assertSubscriber) -> new FluxSwitchOnFirst.SwitchOnFirstControlSubscriber((FluxSwitchOnFirst.AbstractSwitchOnFirstMain) parent, (CoreSubscriber) assertSubscriber, true),
            (parent, assertSubscriber) -> new FluxSwitchOnFirst.SwitchOnFirstConditionalControlSubscriber((FluxSwitchOnFirst.AbstractSwitchOnFirstMain) parent, (Fuseable.ConditionalSubscriber) assertSubscriber, true)
    };
    for (BiFunction<FluxSwitchOnFirst.AbstractSwitchOnFirstMain, CoreSubscriber, InnerOperator> factory : factories) {
        for (int i = 0; i < 100000; i++) {
            long[] valueHolder = new long[] { 0 };
            FluxSwitchOnFirst.AbstractSwitchOnFirstMain mockParent = Mockito.mock(FluxSwitchOnFirst.AbstractSwitchOnFirstMain.class);
            Mockito.doNothing().when(mockParent).request(Mockito.anyLong());
            Mockito.doNothing().when(mockParent).cancel();
            Subscription mockSubscription = Mockito.mock(Subscription.class);
            Mockito.doAnswer((a) -> valueHolder[0] += (long) a.getArgument(0)).when(mockSubscription).request(Mockito.anyLong());
            Mockito.doNothing().when(mockSubscription).cancel();
            AssertSubscriber<Object> subscriber = AssertSubscriber.create(0);
            InnerOperator switchOnFirstControlSubscriber = factory.apply(mockParent, Operators.toConditionalSubscriber(subscriber));

            switchOnFirstControlSubscriber.request(10);
            RaceTestUtils.race(() -> {
                        switchOnFirstControlSubscriber.request(10);
                        switchOnFirstControlSubscriber.request(10);
                        switchOnFirstControlSubscriber.request(10);
                        switchOnFirstControlSubscriber.request(10);
                    },
                    () -> switchOnFirstControlSubscriber.onSubscribe(mockSubscription),
                    Schedulers.parallel());

            switchOnFirstControlSubscriber.request(10);
            Assertions.assertThat(valueHolder[0])
                      .isEqualTo(60L);
            mockSubscription.toString();
        }
    }
}

Example 10

Source File: FluxSwitchOnFirst.java From reactor-core with Apache License 2.0

5 votes

@Override
public boolean tryOnNext(T t) {
    @SuppressWarnings("unchecked")
    final Fuseable.ConditionalSubscriber<? super T> i =
            (Fuseable.ConditionalSubscriber<? super T>) this.inner;
    if (this.done || i == Operators.EMPTY_SUBSCRIBER) {
        Operators.onNextDropped(t, currentContext());
        return false;
    }

    if (i == null) {
        final Publisher<? extends R> result;
        final CoreSubscriber<? super R> o = this.outer;

        try {
            result = Objects.requireNonNull(
                this.transformer.apply(Signal.next(t, o.currentContext()), this),
                "The transformer returned a null value"
            );
        }
        catch (Throwable e) {
            this.done = true;
            Operators.error(o, Operators.onOperatorError(this.s, e, t, o.currentContext()));
            return false;
        }

        this.first = t;
        result.subscribe(o);
        return true;
    }

    return i.tryOnNext(t);
}

Example 11

Source File: FluxMapTest.java From reactor-core with Apache License 2.0

5 votes

@Test
public void scanConditionalSubscriber() {
 @SuppressWarnings("unchecked")
 Fuseable.ConditionalSubscriber<String> actual = Mockito.mock(MockUtils.TestScannableConditionalSubscriber.class);
    FluxMap.MapConditionalSubscriber<Integer, String> test = new FluxMap.MapConditionalSubscriber<>(actual, i -> String.valueOf(i));
    Subscription parent = Operators.emptySubscription();
    test.onSubscribe(parent);

    assertThat(test.scan(Scannable.Attr.PARENT)).isSameAs(parent);
    assertThat(test.scan(Scannable.Attr.ACTUAL)).isSameAs(actual);

    assertThat(test.scan(Scannable.Attr.TERMINATED)).isFalse();
    test.onError(new IllegalStateException("boom"));
    assertThat(test.scan(Scannable.Attr.TERMINATED)).isTrue();
}

Example 12

Source File: FluxSwitchOnFirstTest.java From reactor-core with Apache License 2.0

5 votes

@SuppressWarnings("rawtypes")
@Test
public void unitCancelRacingTest() {
    @SuppressWarnings("unchecked")
    BiFunction<FluxSwitchOnFirst.AbstractSwitchOnFirstMain, CoreSubscriber, InnerOperator>[] factories = new BiFunction[] {
            (parent, assertSubscriber) -> new FluxSwitchOnFirst.SwitchOnFirstControlSubscriber((FluxSwitchOnFirst.AbstractSwitchOnFirstMain) parent, (CoreSubscriber) assertSubscriber, true),
            (parent, assertSubscriber) -> new FluxSwitchOnFirst.SwitchOnFirstConditionalControlSubscriber((FluxSwitchOnFirst.AbstractSwitchOnFirstMain) parent, (Fuseable.ConditionalSubscriber) assertSubscriber, true)
    };
    for (BiFunction<FluxSwitchOnFirst.AbstractSwitchOnFirstMain, CoreSubscriber, InnerOperator> factory : factories) {
        for (int i = 0; i < 10000; i++) {
            FluxSwitchOnFirst.AbstractSwitchOnFirstMain mockParent = Mockito.mock(FluxSwitchOnFirst.AbstractSwitchOnFirstMain.class);
            Mockito.doNothing().when(mockParent).request(Mockito.anyLong());
            Mockito.doNothing().when(mockParent).cancel();
            Subscription mockSubscription = Mockito.mock(Subscription.class);
            ArgumentCaptor<Long> longArgumentCaptor = ArgumentCaptor.forClass(Long.class);
            Mockito.doNothing().when(mockSubscription).request(longArgumentCaptor.capture());
            Mockito.doNothing().when(mockSubscription).cancel();
            AssertSubscriber<Object> subscriber = AssertSubscriber.create(0);
            InnerOperator switchOnFirstControlSubscriber = factory.apply(mockParent, Operators.toConditionalSubscriber(subscriber));

            switchOnFirstControlSubscriber.request(10);
            RaceTestUtils.race(() -> switchOnFirstControlSubscriber.cancel(), () -> switchOnFirstControlSubscriber.onSubscribe(mockSubscription), Schedulers.parallel());

            Assertions.assertThat(longArgumentCaptor.getAllValues().size()).isBetween(0, 1);
            Mockito.verify(mockParent).cancel();
            if (longArgumentCaptor.getAllValues().size() == 1) {
                Assertions.assertThat(longArgumentCaptor.getValue()).isEqualTo(10L);
            }
            else if (longArgumentCaptor.getAllValues().size() > 1) {
                Assertions.fail("Unexpected number of calls");
            }
        }
    }
}

Example 13

Source File: MonoMap.java From reactor-core with Apache License 2.0

5 votes

@Override
@SuppressWarnings("unchecked")
public CoreSubscriber<? super T> subscribeOrReturn(CoreSubscriber<? super R> actual) {
	if (actual instanceof Fuseable.ConditionalSubscriber) {
		Fuseable.ConditionalSubscriber<? super R> cs = (Fuseable.ConditionalSubscriber<? super R>) actual;
		return new FluxMap.MapConditionalSubscriber<>(cs, mapper);
	}
	return new FluxMap.MapSubscriber<>(actual, mapper);
}

Example 14

Source File: FluxFilterTest.java From reactor-core with Apache License 2.0

5 votes

@Test
public void discardConditionalTryOnNextPredicateMiss() {
	List<Object> discarded = new ArrayList<>();
	Fuseable.ConditionalSubscriber<Integer> actual = new FluxPeekFuseableTest.ConditionalAssertSubscriber<>(
			Context.of(Hooks.KEY_ON_DISCARD, (Consumer<?>) discarded::add));

	FilterConditionalSubscriber<Integer> subscriber =
			new FilterConditionalSubscriber<>(actual, i -> i % 2 == 0);
	subscriber.onSubscribe(Operators.emptySubscription());

	subscriber.tryOnNext(1);
	subscriber.tryOnNext(2);

	assertThat(discarded).containsExactly(1);
}

Example 15

Source File: DefaultTestPublisher.java From reactor-core with Apache License 2.0

5 votes

@SuppressWarnings("unchecked")
TestPublisherSubscription(Subscriber<? super T> actual, DefaultTestPublisher<T> parent) {
	this.actual = actual;
	if(actual instanceof Fuseable.ConditionalSubscriber){
		this.actualConditional = (Fuseable.ConditionalSubscriber<? super T>) actual;
	}
	else {
		this.actualConditional = null;
	}
	this.parent = parent;
}

Example 16

Source File: ParallelDoOnEach.java From reactor-core with Apache License 2.0

5 votes

@Override
@SuppressWarnings("unchecked")
public void subscribe(CoreSubscriber<? super T>[] subscribers) {
	if (!validate(subscribers)) {
		return;
	}

	int n = subscribers.length;

	CoreSubscriber<? super T>[] parents = new CoreSubscriber[n];

	boolean conditional = subscribers[0] instanceof Fuseable.ConditionalSubscriber;

	for (int i = 0; i < n; i++) {
		CoreSubscriber<? super T> subscriber = subscribers[i];
		SignalPeek<T> signalPeek = new DoOnEachSignalPeek(subscriber.currentContext());

		if (conditional) {
			parents[i] = new FluxPeekFuseable.PeekConditionalSubscriber<>(
					(Fuseable.ConditionalSubscriber<T>) subscriber, signalPeek);
		}
		else {
			parents[i] = new FluxPeek.PeekSubscriber<>(subscriber, signalPeek);
		}
	}

	source.subscribe(parents);
}

Example 17

Source File: FluxFilterTest.java From reactor-core with Apache License 2.0

5 votes

@Test
public void scanConditionalSubscriber() {
 @SuppressWarnings("unchecked")
 Fuseable.ConditionalSubscriber<String> actual = Mockito.mock(MockUtils.TestScannableConditionalSubscriber.class);
    FilterConditionalSubscriber<String> test = new FilterConditionalSubscriber<>(actual, t -> true);
    Subscription parent = Operators.emptySubscription();
    test.onSubscribe(parent);

    assertThat(test.scan(Scannable.Attr.PARENT)).isSameAs(parent);
    assertThat(test.scan(Scannable.Attr.ACTUAL)).isSameAs(actual);

    assertThat(test.scan(Scannable.Attr.TERMINATED)).isFalse();
    test.onError(new IllegalStateException("boom"));
    assertThat(test.scan(Scannable.Attr.TERMINATED)).isTrue();
}

Example 18

Source File: FluxSwitchOnFirst.java From reactor-core with Apache License 2.0

4 votes

SwitchOnFirstConditionalMain(
        Fuseable.ConditionalSubscriber<? super R> outer,
        BiFunction<Signal<? extends T>, Flux<T>, Publisher<? extends R>> transformer,
        boolean cancelSourceOnComplete) {
    super(outer, transformer, cancelSourceOnComplete);
}

Example 19

Source File: DiscardOnCancelSubscriber.java From r2dbc-mysql with Apache License 2.0

4 votes

DiscardOnCancelFuseableConditionalSubscriber(Fuseable.ConditionalSubscriber<? super T> actual) {
    super(actual);
}

Example 20

Source File: DiscardOnCancelSubscriber.java From r2dbc-mysql with Apache License 2.0

4 votes

DiscardOnCancelFuseableConditionalSubscriber(Fuseable.ConditionalSubscriber<? super T> actual) {
    super(actual);
}