package io.reactivex.internal.operators.parallel;
import org.reactivestreams.*;
import io.reactivex.exceptions.*;
import io.reactivex.functions.*;
import io.reactivex.internal.functions.ObjectHelper;
import io.reactivex.internal.fuseable.ConditionalSubscriber;
import io.reactivex.internal.subscriptions.SubscriptionHelper;
import io.reactivex.parallel.*;
import io.reactivex.plugins.RxJavaPlugins;
public final class ParallelFilterTry<T> extends ParallelFlowable<T> {
final ParallelFlowable<T> source;
final Predicate<? super T> predicate;
final BiFunction<? super Long, ? super Throwable, ParallelFailureHandling> errorHandler;
public ParallelFilterTry(ParallelFlowable<T> source, Predicate<? super T> predicate,
BiFunction<? super Long, ? super Throwable, ParallelFailureHandling> errorHandler) {
this.source = source;
this.predicate = predicate;
this.errorHandler = errorHandler;
}
@Override
public void subscribe(Subscriber<? super T>[] subscribers) {
if (!validate(subscribers)) {
return;
}
int n = subscribers.length;
@SuppressWarnings("unchecked")
Subscriber<? super T>[] parents = new Subscriber[n];
for (int i = 0; i < n; i++) {
Subscriber<? super T> a = subscribers[i];
if (a instanceof ConditionalSubscriber) {
parents[i] = new ParallelFilterConditionalSubscriber<T>((ConditionalSubscriber<? super T>)a, predicate, errorHandler);
} else {
parents[i] = new ParallelFilterSubscriber<T>(a, predicate, errorHandler);
}
}
source.subscribe(parents);
}
@Override
public int parallelism() {
return source.parallelism();
}
abstract static class BaseFilterSubscriber<T> implements ConditionalSubscriber<T>, Subscription {
final Predicate<? super T> predicate;
final BiFunction<? super Long, ? super Throwable, ParallelFailureHandling> errorHandler;
Subscription upstream;
boolean done;
BaseFilterSubscriber(Predicate<? super T> predicate, BiFunction<? super Long, ? super Throwable, ParallelFailureHandling> errorHandler) {
this.predicate = predicate;
this.errorHandler = errorHandler;
}
@Override
public final void request(long n) {
upstream.request(n);
}
@Override
public final void cancel() {
upstream.cancel();
}
@Override
public final void onNext(T t) {
if (!tryOnNext(t) && !done) {
upstream.request(1);
}
}
}
static final class ParallelFilterSubscriber<T> extends BaseFilterSubscriber<T> {
final Subscriber<? super T> downstream;
ParallelFilterSubscriber(Subscriber<? super T> actual, Predicate<? super T> predicate, BiFunction<? super Long, ? super Throwable, ParallelFailureHandling> errorHandler) {
super(predicate, errorHandler);
this.downstream = actual;
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validate(this.upstream, s)) {
this.upstream = s;
downstream.onSubscribe(this);
}
}
@Override
public boolean tryOnNext(T t) {
if (!done) {
long retries = 0L;
for (;;) {
boolean b;
try {
b = predicate.test(t);
} catch (Throwable ex) {
Exceptions.throwIfFatal(ex);
ParallelFailureHandling h;
try {
h = ObjectHelper.requireNonNull(errorHandler.apply(++retries, ex), "The errorHandler returned a null item");
} 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;
}
}
if (b) {
downstream.onNext(t);
return true;
}
return false;
}
}
return false;
}
@Override
public void onError(Throwable t) {
if (done) {
RxJavaPlugins.onError(t);
return;
}
done = true;
downstream.onError(t);
}
@Override
public void onComplete() {
if (!done) {
done = true;
downstream.onComplete();
}
}
}
static final class ParallelFilterConditionalSubscriber<T> extends BaseFilterSubscriber<T> {
final ConditionalSubscriber<? super T> downstream;
ParallelFilterConditionalSubscriber(ConditionalSubscriber<? super T> actual,
Predicate<? super T> predicate,
BiFunction<? super Long, ? super Throwable, ParallelFailureHandling> errorHandler) {
super(predicate, errorHandler);
this.downstream = actual;
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validate(this.upstream, s)) {
this.upstream = s;
downstream.onSubscribe(this);
}
}
@Override
public boolean tryOnNext(T t) {
if (!done) {
long retries = 0L;
for (;;) {
boolean b;
try {
b = predicate.test(t);
} catch (Throwable ex) {
Exceptions.throwIfFatal(ex);
ParallelFailureHandling h;
try {
h = ObjectHelper.requireNonNull(errorHandler.apply(++retries, ex), "The errorHandler returned a null item");
} 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 b && downstream.tryOnNext(t);
}
}
return false;
}
@Override
public void onError(Throwable t) {
if (done) {
RxJavaPlugins.onError(t);
return;
}
done = true;
downstream.onError(t);
}
@Override
public void onComplete() {
if (!done) {
done = true;
downstream.onComplete();
}
}
}}