code: /*
   * 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.rxjava3.internal.operators.flowable;
  
  import java.util.concurrent.atomic.AtomicInteger;
  
  import org.reactivestreams.Subscriber;
  
  import io.reactivex.rxjava3.core.Flowable;
  import io.reactivex.rxjava3.disposables.Disposable;
  import io.reactivex.rxjava3.flowables.ConnectableFlowable;
  import io.reactivex.rxjava3.functions.Consumer;
  
  /**
   * Wraps a {@link ConnectableFlowable} and calls its {@code connect()} method once
   * the specified number of {@link Subscriber}s have subscribed.
   *
   * @param <T> the value type of the chain
   */
  public final class FlowableAutoConnect<T> extends Flowable<T> {
      final ConnectableFlowable<? extends T> source;
      final int numberOfSubscribers;
      final Consumer<? super Disposable> connection;
      final AtomicInteger clients;
  
      public FlowableAutoConnect(ConnectableFlowable<? extends T> source,
              int numberOfSubscribers,
              Consumer<? super Disposable> connection) {
          this.source = source;
          this.numberOfSubscribers = numberOfSubscribers;
          this.connection = connection;
          this.clients = new AtomicInteger();
      }
  
      @Override
      public void subscribeActual(Subscriber<? super T> child) {
          source.subscribe(child);
          if (clients.incrementAndGet() == numberOfSubscribers) {
              source.connect(connection);
          }
      }
  }
  