content: /*
   * Copyright 2016-present 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 <T1> the first value type
   * @param <T2> the second value type
   * @param <R> the result type
   */
  final class Plan2<T1, T2, R> extends Plan<R> {
      protected final Pattern2<T1, T2> expression;
      protected final BiFunction<T1, T2, R> selector;
  
      Plan2(Pattern2<T1, T2> expression, BiFunction<T1, T2, R> selector) {
          this.expression = expression;
          this.selector = selector;
      }
  
      @Override
      public ActivePlan0 activate(Map<Object, JoinObserver> externalSubscriptions,
              final Observer<R> observer, final Consumer<ActivePlan0> deactivate) {
          Consumer<Throwable> onError = onErrorFrom(observer);
  
          final JoinObserver1<T1> jo1 = createObserver(externalSubscriptions, expression.o1(), onError);
          final JoinObserver1<T2> jo2 = createObserver(externalSubscriptions, expression.o2(), onError);
  
          final AtomicReference<ActivePlan2<T1, T2>> self = new AtomicReference<>();
  
          ActivePlan2<T1, T2> activePlan = new ActivePlan2<>(jo1, jo2, new BiConsumer<T1, T2>() {
              @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;
      }
  
  }
  