hu.akarnokd.rxjava3.joins.Plan2 Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rxjava3-extensions Show documentation

RxJava 3.x extra sources, operators and components and ports of many 1.x companion libraries.

There is a newer version: 3.1.1

/*
 * Copyright 2016-2019 David Karnok
 *
 * 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 hu.akarnokd.rxjava3.joins;

import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;

import io.reactivex.rxjava3.core.Observer;
import io.reactivex.rxjava3.functions.*;

/**
 * Represents an execution plan for join patterns.
 * @param  the first value type
 * @param  the second value type
 * @param  the result type
 */
final class Plan2 extends Plan {
    protected final Pattern2 expression;
    protected final BiFunction selector;

    Plan2(Pattern2 expression, BiFunction selector) {
        this.expression = expression;
        this.selector = selector;
    }

    @Override
    public ActivePlan0 activate(Map externalSubscriptions,
            final Observer observer, final Consumer deactivate) {
        Consumer onError = onErrorFrom(observer);

        final JoinObserver1 jo1 = createObserver(externalSubscriptions, expression.o1(), onError);
        final JoinObserver1 jo2 = createObserver(externalSubscriptions, expression.o2(), onError);

        final AtomicReference> self = new AtomicReference>();

        ActivePlan2 activePlan = new ActivePlan2(jo1, jo2, new BiConsumer() {
            @Override
            public void accept(T1 t1, T2 t2) {
                R result;
                try {
                    result = selector.apply(t1, t2);
                } catch (Throwable t) {
                    observer.onError(t);
                    return;
                }
                observer.onNext(result);
            }
        },
        new Action() {
            @Override
            public void run() throws Throwable  {
                ActivePlan0 ap = self.get();
                jo1.removeActivePlan(ap);
                jo2.removeActivePlan(ap);
                deactivate.accept(ap);
            }
        });

        self.set(activePlan);

        jo1.addActivePlan(activePlan);
        jo2.addActivePlan(activePlan);

        return activePlan;
    }

}