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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of server Show documentation
The newest version!
/*
 * 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.Objects;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

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

/**
 * @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 (IOException 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 HttpServerBuilderAt at(String path) {
      Objects.requireNonNull(path);
      Path thePath = Path.parse(path);
      return new At(thePath);
    }

    @Override
    public HttpServerBuilderWhen when(Method method) {
      Objects.requireNonNull(method);
      return new When(method);
    }

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

    private class At implements HttpServerBuilderAt {

      private final Path path;

      public At(Path path) {
        this.path = path;
      }

      @Override
      public HttpServerBuilderPort root(Directory directory) {
        PathSpec.Builder pathSpec = PathSpec.builder();
        while (path.hasNextPart()) {
          pathSpec.fixed(path.nextPart());
        }
        pathSpec.catchAllArgument();

        routeList.add(Route.builder()
            .method(Method.GET)
            .spec(pathSpec.build())
            .executor(new DirectoryRouteExecutor(directory))
            .build());
        return Builder.this;
      }

    }

    private class When
        implements
        HttpServerBuilderWhen,
        HttpServerBuilderWhenAt {

      private final Method method;
      private Path path;

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

      @Override
      public HttpServerBuilderWhenAt at(String path) {
        this.path = Path.parse(path);
        return this;
      }

      @Override
      public HttpServerBuilderPort execute(RouteExecutor executor) {
        PathSpec.Builder pathSpec = PathSpec.builder();
        while (path.hasNextPart()) {
          pathSpec.fixed(path.nextPart());
        }

        routeList.add(Route.builder()
            .method(method)
            .spec(pathSpec.build())
            .executor(executor)
            .build());
        return Builder.this;
      }

    }

  }

}