br.com.objectos.http.HttpServer Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016 Objectos, Fábrica de Software LTDA.
 *
 * 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 br.com.objectos.http;

import java.io.IOException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import br.com.objectos.core.util.ImmutableList;

/**
 * @author [email protected] (Marcio Endo)
 */
public class HttpServer {

  private final Channel channel;
  private final RouteRegistry routeRegistry;

  private final ExecutorService executor;

  private HttpServer(Channel channel, RouteRegistry routeRegistry) {
    this.channel = channel;
    this.routeRegistry = routeRegistry;
    executor = Executors.newSingleThreadExecutor();
  }

  public static HttpServerBuilder builder() {
    return new Builder();
  }

  public HttpServer start() {
    executor.submit(new Lifecycle());
    return this;
  }

  public HttpServer stop() {
    try {
      channel.close();
    } catch (IOException e) {
      e.printStackTrace();
    } finally {
      executor.shutdown();
    }
    return this;
  }

  private void accept() throws IOException {
    try (Socket socket = channel.accept()) {
      socket.parse().resolve(routeRegistry).execute().respond(socket);
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  private class Lifecycle implements Runnable {

    @Override
    public void run() {
      while (!executor.isShutdown()) {
        try {
          accept();
        } catch (IOException e) {
          e.printStackTrace();
          stop();
        }
      }
    }

  }

  private static class Builder
      implements
      HttpServerBuilder,
      HttpServerBuilder.HttpServerBuilderPort {

    private int port = 80;
    private final ImmutableList.Builder routeList = ImmutableList.builder();

    @Override
    public HttpServer build() {
      try {
        Channel channel = Channel.get(port);
        RouteRegistry routeRegistry = RouteRegistry.of(routeList.build());
        return new HttpServer(channel, routeRegistry);
      } catch (IOException e) {
        throw new RuntimeException(e);
      }
    }

    @Override
    public HttpServerBuilderGet get() {
      return new RouteBuilder(Method.GET);
    }

    @Override
    public HttpServerBuilderPort port(int port) {
      this.port = port;
      return this;
    }

    private class RouteBuilder implements HttpServerBuilderGet {

      private final Method method;
      private final PathSpec.Builder pathSpec = PathSpec.builder();

      public RouteBuilder(Method method) {
        this.method = method;
      }

      @Override
      public HttpServerBuilderPort execute(RouteExecutor executor) {
        routeList.add(Route.builder()
            .method(method)
            .spec(pathSpec.build())
            .executor(executor)
            .build());
        return Builder.this;
      }

      @Override
      public HttpServerBuilderGet fixed(String part) {
        pathSpec.fixed(part);
        return this;
      }

    }

  }

}