io.reactivex.rxjava3.internal.operators.parallel.ParallelDoOnNextTry Maven / Gradle / Ivy
Show all versions of rxjava Show documentation
/*
* Copyright (c) 2016-present, RxJava Contributors.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in
* compliance with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under the License is
* distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
* the License for the specific language governing permissions and limitations under the License.
*/
package io.reactivex.rxjava3.internal.operators.parallel;
import org.reactivestreams.*;
import io.reactivex.rxjava3.exceptions.*;
import io.reactivex.rxjava3.functions.*;
import io.reactivex.rxjava3.internal.subscriptions.SubscriptionHelper;
import io.reactivex.rxjava3.operators.ConditionalSubscriber;
import io.reactivex.rxjava3.parallel.*;
import io.reactivex.rxjava3.plugins.RxJavaPlugins;
import java.util.Objects;
/**
* Calls a Consumer for each upstream value passing by
* and handles any failure with a handler function.
* History: 2.0.8 - experimental
* @param the input value type
* @since 2.2
*/
public final class ParallelDoOnNextTry extends ParallelFlowable {
final ParallelFlowable source;
final Consumer onNext;
final BiFunction errorHandler;
public ParallelDoOnNextTry(ParallelFlowable source, Consumer onNext,
BiFunction errorHandler) {
this.source = source;
this.onNext = onNext;
this.errorHandler = errorHandler;
}
@Override
public void subscribe(Subscriber[] subscribers) {
subscribers = RxJavaPlugins.onSubscribe(this, subscribers);
if (!validate(subscribers)) {
return;
}
int n = subscribers.length;
@SuppressWarnings("unchecked")
Subscriber[] parents = new Subscriber[n];
for (int i = 0; i < n; i++) {
Subscriber a = subscribers[i];
if (a instanceof ConditionalSubscriber) {
parents[i] = new ParallelDoOnNextConditionalSubscriber<>((ConditionalSubscriber)a, onNext, errorHandler);
} else {
parents[i] = new ParallelDoOnNextSubscriber<>(a, onNext, errorHandler);
}
}
source.subscribe(parents);
}
@Override
public int parallelism() {
return source.parallelism();
}
static final class ParallelDoOnNextSubscriber implements ConditionalSubscriber, Subscription {
final Subscriber downstream;
final Consumer onNext;
final BiFunction errorHandler;
Subscription upstream;
boolean done;
ParallelDoOnNextSubscriber(Subscriber actual, Consumer onNext,
BiFunction errorHandler) {
this.downstream = actual;
this.onNext = onNext;
this.errorHandler = errorHandler;
}
@Override
public void request(long n) {
upstream.request(n);
}
@Override
public void cancel() {
upstream.cancel();
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validate(this.upstream, s)) {
this.upstream = s;
downstream.onSubscribe(this);
}
}
@Override
public void onNext(T t) {
if (!tryOnNext(t)) {
upstream.request(1);
}
}
@Override
public boolean tryOnNext(T t) {
if (done) {
return false;
}
long retries = 0;
for (;;) {
try {
onNext.accept(t);
} catch (Throwable ex) {
Exceptions.throwIfFatal(ex);
ParallelFailureHandling h;
try {
h = Objects.requireNonNull(errorHandler.apply(++retries, ex), "The errorHandler returned a null ParallelFailureHandling");
} catch (Throwable exc) {
Exceptions.throwIfFatal(exc);
cancel();
onError(new CompositeException(ex, exc));
return false;
}
switch (h) {
case RETRY:
continue;
case SKIP:
return false;
case STOP:
cancel();
onComplete();
return false;
default:
cancel();
onError(ex);
return false;
}
}
downstream.onNext(t);
return true;
}
}
@Override
public void onError(Throwable t) {
if (done) {
RxJavaPlugins.onError(t);
return;
}
done = true;
downstream.onError(t);
}
@Override
public void onComplete() {
if (done) {
return;
}
done = true;
downstream.onComplete();
}
}
static final class ParallelDoOnNextConditionalSubscriber implements ConditionalSubscriber, Subscription {
final ConditionalSubscriber downstream;
final Consumer onNext;
final BiFunction errorHandler;
Subscription upstream;
boolean done;
ParallelDoOnNextConditionalSubscriber(ConditionalSubscriber actual,
Consumer onNext,
BiFunction errorHandler) {
this.downstream = actual;
this.onNext = onNext;
this.errorHandler = errorHandler;
}
@Override
public void request(long n) {
upstream.request(n);
}
@Override
public void cancel() {
upstream.cancel();
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validate(this.upstream, s)) {
this.upstream = s;
downstream.onSubscribe(this);
}
}
@Override
public void onNext(T t) {
if (!tryOnNext(t) && !done) {
upstream.request(1);
}
}
@Override
public boolean tryOnNext(T t) {
if (done) {
return false;
}
long retries = 0;
for (;;) {
try {
onNext.accept(t);
} catch (Throwable ex) {
Exceptions.throwIfFatal(ex);
ParallelFailureHandling h;
try {
h = Objects.requireNonNull(errorHandler.apply(++retries, ex), "The errorHandler returned a null ParallelFailureHandling");
} catch (Throwable exc) {
Exceptions.throwIfFatal(exc);
cancel();
onError(new CompositeException(ex, exc));
return false;
}
switch (h) {
case RETRY:
continue;
case SKIP:
return false;
case STOP:
cancel();
onComplete();
return false;
default:
cancel();
onError(ex);
return false;
}
}
return downstream.tryOnNext(t);
}
}
@Override
public void onError(Throwable t) {
if (done) {
RxJavaPlugins.onError(t);
return;
}
done = true;
downstream.onError(t);
}
@Override
public void onComplete() {
if (done) {
return;
}
done = true;
downstream.onComplete();
}
}
}