All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.reactivex.internal.operators.parallel.ParallelMapTry Maven / Gradle / Ivy

The newest version!
/**
 * 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.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;

/**
 * Maps each 'rail' of the source ParallelFlowable with a mapper function
 * and handle any failure based on a handler function.
 * 

History: 2.0.8 - experimental * @param the input value type * @param the output value type * @since 2.2 */ public final class ParallelMapTry extends ParallelFlowable { final ParallelFlowable source; final Function mapper; final BiFunction errorHandler; public ParallelMapTry(ParallelFlowable source, Function mapper, BiFunction errorHandler) { this.source = source; this.mapper = mapper; this.errorHandler = errorHandler; } @Override public void subscribe(Subscriber[] 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 ParallelMapTryConditionalSubscriber((ConditionalSubscriber)a, mapper, errorHandler); } else { parents[i] = new ParallelMapTrySubscriber(a, mapper, errorHandler); } } source.subscribe(parents); } @Override public int parallelism() { return source.parallelism(); } static final class ParallelMapTrySubscriber implements ConditionalSubscriber, Subscription { final Subscriber downstream; final Function mapper; final BiFunction errorHandler; Subscription upstream; boolean done; ParallelMapTrySubscriber(Subscriber actual, Function mapper, BiFunction errorHandler) { this.downstream = actual; this.mapper = mapper; 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 (;;) { R v; try { v = ObjectHelper.requireNonNull(mapper.apply(t), "The mapper returned a null value"); } 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; } } downstream.onNext(v); 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 ParallelMapTryConditionalSubscriber implements ConditionalSubscriber, Subscription { final ConditionalSubscriber downstream; final Function mapper; final BiFunction errorHandler; Subscription upstream; boolean done; ParallelMapTryConditionalSubscriber(ConditionalSubscriber actual, Function mapper, BiFunction errorHandler) { this.downstream = actual; this.mapper = mapper; 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 (;;) { R v; try { v = ObjectHelper.requireNonNull(mapper.apply(t), "The mapper returned a null value"); } 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 downstream.tryOnNext(v); } } @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(); } } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy