io.fusionauth.http.server.HTTPServer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-http Show documentation

An HTTP library for Java that provides a lightweight server (currently) and client (eventually) both with a goal of high-performance and simplicity

There is a newer version: 0.4.0-RC.3

Show newest version

/*
 * Copyright (c) 2022, FusionAuth, All Rights Reserved
 *
 * 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.fusionauth.http.server;

import java.io.Closeable;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import io.fusionauth.http.log.Logger;
import io.fusionauth.http.util.ThreadPool;

/**
 * The server bro!
 *
 * @author Brian Pontarelli
 */
public class HTTPServer implements Closeable, Configurable {
  private final List threads = new ArrayList<>();

  private HTTPServerConfiguration configuration = new HTTPServerConfiguration();

  private volatile HTTPContext context;

  private Logger logger;

  private ThreadPool threadPool;

  @Override
  public void close() {
    logger.info("HTTP server shutdown requested. Attempting to close each listener. This could take a while.");

    for (HTTPServerThread thread : threads) {
      thread.close();

      try {
        thread.join(10_000);
      } catch (InterruptedException e) {
        // Ignore so we join on all the threads
      }
    }

    if (threadPool.shutdown()) {
      logger.info("HTTP server shutdown successfully.");
    } else {
      logger.error("HTTP server shutdown failed. It's harshing my mellow!");
    }
  }

  @Override
  public HTTPServerConfiguration configuration() {
    return configuration;
  }

  /**
   * @return The HTTP Context or null if the server hasn't been started yet.
   */
  public HTTPContext getContext() {
    return context;
  }

  public HTTPServer start() {
    if (context != null) {
      return this;
    }

    logger = configuration.getLoggerFactory().getLogger(HTTPServer.class);
    logger.info("Starting the HTTP server. Buckle up!");

    context = new HTTPContext(configuration.getBaseDir());

    // Start the thread pool for the workers
    threadPool = new ThreadPool(configuration.getNumberOfWorkerThreads(), "HTTP Server Worker Thread", configuration.getShutdownDuration());

    try {
      for (HTTPListenerConfiguration listener : configuration.getListeners()) {
        HTTPServerThread thread = new HTTPServerThread(configuration, listener, threadPool);
        thread.start();
        threads.add(thread);

        logger.info("HTTP server listening on port [{}]", listener.getPort());
      }

      logger.info("HTTP server started successfully");
    } catch (IOException e) {
      logger.error("Unable to start the HTTP server because one of the listeners threw an exception.", e);

      // Clean up the threads that did start
      close();
    }

    return this;
  }

  /**
   * Specify the full configuration object for the server rather than using the {@code with} builder methods.
   *
   * @param configuration The configuration for the server.
   * @return This.
   */
  public HTTPServer withConfiguration(HTTPServerConfiguration configuration) {
    this.configuration = configuration;
    this.logger = configuration.getLoggerFactory().getLogger(HTTPServer.class);
    return this;
  }
}