<gh_stars>1-10
/*
 * Copyright 2017 StreamSets Inc.
 *
 * 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.streamsets.pipeline.lib.httpsource;

import com.streamsets.pipeline.api.PushSource;
import com.streamsets.pipeline.api.StageException;
import com.streamsets.pipeline.lib.http.HttpConfigs;
import com.streamsets.pipeline.lib.http.HttpReceiver;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;

public abstract class AbstractHttpServerPushSource<R extends HttpReceiver>
    extends AbstractHttpServerProtoSource<R, PushSource.Context> implements PushSource {

  private static final Logger LOG = LoggerFactory.getLogger(AbstractHttpServerPushSource.class);

  public AbstractHttpServerPushSource(HttpConfigs httpConfigs, R receiver) {
    super(httpConfigs, receiver);
  }

  @Override
  public int getNumberOfThreads() {
    return getHttpConfigs().getMaxConcurrentRequests();
  }

  @Override
  public void produce(Map<String, String> map, int i) throws StageException {
    try {
      server.startServer();

      while (!getContext().isStopped()) {
        dispatchHttpReceiverErrors(100);
      }
    } finally {
      LOG.trace("Destroying server");
      server.destroy();
    }
  }
}
