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

io.reactivex.observable.internal.operators.MaybeConcatIterable 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.observable.internal.operators;

import java.util.Iterator;
import java.util.concurrent.atomic.*;

import io.reactivex.common.Disposable;
import io.reactivex.common.exceptions.Exceptions;
import io.reactivex.common.internal.disposables.SequentialDisposable;
import io.reactivex.common.internal.functions.ObjectHelper;
import io.reactivex.observable.*;
import io.reactivex.observable.internal.disposables.EmptyDisposable;
import io.reactivex.observable.internal.utils.NotificationLite;

/**
 * Concatenate values of each MaybeSource provided by an Iterable.
 *
 * @param  the value type
 */
public final class MaybeConcatIterable extends Observable {

    final Iterable> sources;

    public MaybeConcatIterable(Iterable> sources) {
        this.sources = sources;
    }

    @Override
    protected void subscribeActual(Observer s) {

        Iterator> it;

        try {
            it = ObjectHelper.requireNonNull(sources.iterator(), "The sources Iterable returned a null Iterator");
        } catch (Throwable ex) {
            Exceptions.throwIfFatal(ex);
            EmptyDisposable.error(ex, s);
            return;
        }

        ConcatMaybeObserver parent = new ConcatMaybeObserver(s, it);
        s.onSubscribe(parent);
        parent.drain();
    }

    static final class ConcatMaybeObserver
    extends AtomicInteger
    implements MaybeObserver, Disposable {

        private static final long serialVersionUID = 3520831347801429610L;

        final Observer actual;

        final AtomicReference current;

        final SequentialDisposable disposables;

        final Iterator> sources;

        ConcatMaybeObserver(Observer actual, Iterator> sources) {
            this.actual = actual;
            this.sources = sources;
            this.disposables = new SequentialDisposable();
            this.current = new AtomicReference(NotificationLite.COMPLETE); // as if a previous completed
        }

        @Override
        public void dispose() {
            disposables.dispose();
        }

        @Override
        public boolean isDisposed() {
            return disposables.isDisposed();
        }

        @Override
        public void onSubscribe(Disposable d) {
            disposables.replace(d);
        }

        @Override
        public void onSuccess(T value) {
            current.lazySet(value);
            drain();
        }

        @Override
        public void onError(Throwable e) {
            actual.onError(e);
        }

        @Override
        public void onComplete() {
            current.lazySet(NotificationLite.COMPLETE);
            drain();
        }

        @SuppressWarnings("unchecked")
        void drain() {
            if (getAndIncrement() != 0) {
                return;
            }

            AtomicReference c = current;
            Observer a = actual;
            Disposable cancelled = disposables;

            for (;;) {
                if (cancelled.isDisposed()) {
                    c.lazySet(null);
                    return;
                }

                Object o = c.get();

                if (o != null) {
                    if (o != NotificationLite.COMPLETE) {
                        c.lazySet(null);

                        a.onNext((T)o);
                    } else {
                        c.lazySet(null);
                    }

                    if (!cancelled.isDisposed()) {
                        boolean b;

                        try {
                            b = sources.hasNext();
                        } catch (Throwable ex) {
                            Exceptions.throwIfFatal(ex);
                            a.onError(ex);
                            return;
                        }

                        if (b) {
                            MaybeSource source;

                            try {
                                source = ObjectHelper.requireNonNull(sources.next(), "The source Iterator returned a null MaybeSource");
                            } catch (Throwable ex) {
                                Exceptions.throwIfFatal(ex);
                                a.onError(ex);
                                return;
                            }

                            source.subscribe(this);
                        } else {
                            a.onComplete();
                        }
                    }
                }

                if (decrementAndGet() == 0) {
                    break;
                }
            }
        }
    }
}