com.hivemq.client.internal.rx.operators.FlowableWithSingleMapError Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hivemq-mqtt-client Show documentation
HiveMQ MQTT Client is an MQTT 5.0 and MQTT 3.1.1 compatible and feature-rich high-performance Java client library with different API flavours and backpressure support
The newest version!
/*
 * Copyright 2018-present HiveMQ and the HiveMQ Community
 *
 * 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 com.hivemq.client.internal.rx.operators;

import com.hivemq.client.internal.rx.WithSingleConditionalSubscriber;
import com.hivemq.client.internal.util.Checks;
import com.hivemq.client.rx.FlowableWithSingle;
import com.hivemq.client.rx.FlowableWithSingleSubscriber;
import com.hivemq.client.rx.reactivestreams.WithSingleSubscriber;
import io.reactivex.FlowableSubscriber;
import io.reactivex.exceptions.CompositeException;
import io.reactivex.exceptions.Exceptions;
import io.reactivex.functions.Function;
import io.reactivex.internal.fuseable.ConditionalSubscriber;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.reactivestreams.Subscriber;
import org.reactivestreams.Subscription;

/**
 * @author Silvio Giebl
 */
public class FlowableWithSingleMapError extends FlowableWithSingleOperator {

    private final @NotNull Function errorMapper;

    public FlowableWithSingleMapError(
            final @NotNull FlowableWithSingle source,
            final @NotNull Function errorMapper) {

        super(source);
        this.errorMapper = errorMapper;
    }

    @Override
    protected void subscribeActual(final @NotNull Subscriber subscriber) {
        if (subscriber instanceof ConditionalSubscriber) {
            //noinspection unchecked
            final ConditionalSubscriber conditional = (ConditionalSubscriber) subscriber;
            source.subscribe(new MapErrorSubscriber.Conditional<>(conditional, errorMapper));
        } else {
            source.subscribe(new MapErrorSubscriber<>(subscriber, errorMapper));
        }
    }

    @Override
    protected void subscribeBothActual(final @NotNull WithSingleSubscriber subscriber) {
        if (subscriber instanceof WithSingleConditionalSubscriber) {
            //noinspection unchecked
            final WithSingleConditionalSubscriber conditional =
                    (WithSingleConditionalSubscriber) subscriber;
            source.subscribeBoth(new WithSingleMapErrorSubscriber.Conditional<>(conditional, errorMapper));
        } else {
            source.subscribeBoth(new WithSingleMapErrorSubscriber<>(subscriber, errorMapper));
        }
    }

    private static class MapErrorSubscriber>
            implements FlowableSubscriber, Subscription {

        final @NotNull T subscriber;
        private final @NotNull Function errorMapper;
        private @Nullable Subscription subscription;

        MapErrorSubscriber(
                final @NotNull T subscriber,
                final @NotNull Function errorMapper) {

            this.subscriber = subscriber;
            this.errorMapper = errorMapper;
        }

        @Override
        public void onSubscribe(final @NotNull Subscription subscription) {
            this.subscription = subscription;
            subscriber.onSubscribe(this);
        }

        @Override
        public void onNext(final @NotNull F f) {
            subscriber.onNext(f);
        }

        @Override
        public void onError(final @NotNull Throwable error) {
            Throwable em;
            try {
                em = Checks.notNull(errorMapper.apply(error), "Mapped exception");
            } catch (final Throwable throwable) {
                Exceptions.throwIfFatal(throwable);
                em = new CompositeException(error, throwable);
            }
            subscriber.onError(em);
        }

        @Override
        public void onComplete() {
            subscriber.onComplete();
        }

        @Override
        public void request(final long n) {
            assert subscription != null;
            subscription.request(n);
        }

        @Override
        public void cancel() {
            assert subscription != null;
            subscription.cancel();
        }

        private static class Conditional> extends MapErrorSubscriber
                implements ConditionalSubscriber {

            Conditional(
                    final @NotNull T subscriber,
                    final @NotNull Function errorMapper) {

                super(subscriber, errorMapper);
            }

            @Override
            public boolean tryOnNext(final @NotNull F f) {
                return subscriber.tryOnNext(f);
            }
        }
    }

    private static class WithSingleMapErrorSubscriber
            extends MapErrorSubscriber>
            implements FlowableWithSingleSubscriber {

        WithSingleMapErrorSubscriber(
                final @NotNull WithSingleSubscriber subscriber,
                final @NotNull Function errorMapper) {

            super(subscriber, errorMapper);
        }

        @Override
        public void onSingle(final @NotNull S s) {
            subscriber.onSingle(s);
        }

        private static class Conditional
                extends MapErrorSubscriber.Conditional>
                implements WithSingleConditionalSubscriber {

            Conditional(
                    final @NotNull WithSingleConditionalSubscriber subscriber,
                    final @NotNull Function errorMapper) {

                super(subscriber, errorMapper);
            }

            @Override
            public void onSingle(final @NotNull S s) {
                subscriber.onSingle(s);
            }
        }
    }
}