Пример #1
0
  private Registry getEffectiveRegistry(final DefaultHandlingResult.ResultsHolder results) {

    ClientErrorHandler clientErrorHandler =
        (context, statusCode) -> {
          results.setClientError(statusCode);
          context.getResponse().status(statusCode);
          results.getLatch().countDown();
        };

    ServerErrorHandler serverErrorHandler =
        (context, throwable1) -> {
          results.setThrowable(throwable1);
          results.getLatch().countDown();
        };

    final Registry userRegistry =
        Registry.builder()
            .add(ClientErrorHandler.class, clientErrorHandler)
            .add(ServerErrorHandler.class, serverErrorHandler)
            .build();
    return Exceptions.uncheck(
        () -> {
          ServerConfig serverConfig = serverConfigBuilder.build();
          DefaultExecController execController =
              new DefaultExecController(serverConfig.getThreads());
          return ServerRegistry.serverRegistry(
              new TestServer(),
              Impositions.none(),
              execController,
              serverConfig,
              r -> userRegistry.join(registryBuilder.build()));
        });
  }
Пример #2
0
 @Override
 public void execute(RatpackServerSpec server) throws Exception {
   RatpackDslClosures closures = new RatpackDslClosures(null);
   RatpackDslBacking backing = new RatpackDslBacking(closures);
   ClosureUtil.configureDelegateFirst(backing, closure);
   server.registry(Guice.registry(ClosureUtil.delegatingAction(closures.getBindings())));
   server.handlers(Groovy.chainAction(closures.getHandlers()));
   ServerConfigBuilder builder = ServerConfig.builder().development(true);
   if (ephemeralPort) {
     builder.port(0);
   }
   ClosureUtil.configureDelegateFirst(builder, closures.getServerConfig());
   server.serverConfig(builder);
 }
Пример #3
0
 @Override
 public TransformablePublisher<? extends ByteBuf> getBodyStream() {
   return getBodyStream(serverConfig.getMaxContentLength());
 }
Пример #4
0
 @Override
 public Promise<TypedData> getBody(Block onTooLarge) {
   return getBody(serverConfig.getMaxContentLength(), onTooLarge);
 }
Пример #5
0
 @Override
 public Promise<TypedData> getBody() {
   return getBody(serverConfig.getMaxContentLength());
 }
 @Provides
 HttpClient httpClient(ByteBufAllocator byteBufAllocator, ServerConfig serverConfig) {
   return HttpClient.httpClient(byteBufAllocator, serverConfig.getMaxContentLength());
 }
Пример #7
0
  public static Registry buildBaseRegistry(
      RatpackServer ratpackServer, ExecController execController, ServerConfig serverConfig) {
    ErrorHandler errorHandler =
        serverConfig.isDevelopment()
            ? new DefaultDevelopmentErrorHandler()
            : new DefaultProductionErrorHandler();

    RegistryBuilder baseRegistryBuilder;
    try {
      baseRegistryBuilder =
          Registry.builder()
              .add(ServerConfig.class, serverConfig)
              .add(ByteBufAllocator.class, PooledByteBufAllocator.DEFAULT)
              .add(ExecController.class, execController)
              .add(MimeTypes.class, new ActivationBackedMimeTypes())
              .add(
                  PublicAddress.class,
                  new DefaultPublicAddress(
                      serverConfig.getPublicAddress(),
                      serverConfig.getSSLContext() == null ? HTTP_SCHEME : HTTPS_SCHEME))
              .add(Redirector.class, new DefaultRedirector())
              .add(ClientErrorHandler.class, errorHandler)
              .add(ServerErrorHandler.class, errorHandler)
              .with(new DefaultFileRenderer().register())
              .with(new PromiseRenderer().register())
              .with(new PublisherRenderer().register())
              .with(new RenderableRenderer().register())
              .with(new CharSequenceRenderer().register())
              .add(FormParser.class, FormParser.multiPart())
              .add(FormParser.class, FormParser.urlEncoded())
              .add(FormNoOptParser.class, FormNoOptParser.multiPart())
              .add(FormNoOptParser.class, FormNoOptParser.urlEncoded())
              .add(RatpackServer.class, ratpackServer)
              // TODO remove Stopper, and just use RatpackServer instead (will need to update perf
              // and gradle tests)
              .add(
                  Stopper.class,
                  () ->
                      uncheck(
                          () -> {
                            ratpackServer.stop();
                            return null;
                          }))
              .add(
                  HttpClient.class,
                  HttpClient.httpClient(
                      PooledByteBufAllocator.DEFAULT, serverConfig.getMaxContentLength()))
              .add(
                  ServerSentEventStreamClient.class,
                  ServerSentEventStreamClient.sseStreamClient(PooledByteBufAllocator.DEFAULT))
              .add(HealthCheckResultsRenderer.class, new HealthCheckResultsRenderer())
              .add(RequestId.Generator.class, new UuidBasedRequestIdGenerator());

      addConfigObjects(serverConfig, baseRegistryBuilder);
    } catch (Exception e) {
      // Uncheck because it really shouldn't happen
      throw uncheck(e);
    }

    if (serverConfig.isHasBaseDir()) {
      baseRegistryBuilder.add(FileSystemBinding.class, serverConfig.getBaseDir());
    }

    return baseRegistryBuilder.build();
  }
Пример #8
0
 private static void addConfigObjects(
     ServerConfig serverConfig, RegistryBuilder baseRegistryBuilder) {
   for (ConfigObject<?> configObject : serverConfig.getRequiredConfig()) {
     addConfigObject(baseRegistryBuilder, configObject);
   }
 }
Пример #9
0
/**
 * @see ratpack.test.handling.RequestFixture#handle(ratpack.handling.Handler, ratpack.func.Action)
 */
@SuppressWarnings("UnusedDeclaration")
public class DefaultRequestFixture implements RequestFixture {

  private final ByteBuf requestBody = unreleasableBuffer(buffer());
  private final MutableHeaders requestHeaders =
      new NettyHeadersBackedMutableHeaders(new DefaultHttpHeaders());
  private final NettyHeadersBackedMutableHeaders responseHeaders =
      new NettyHeadersBackedMutableHeaders(new DefaultHttpHeaders());

  private String method = "GET";
  private String protocol = "HTTP/1.1";
  private String uri = "/";
  private HostAndPort remoteHostAndPort = HostAndPort.fromParts("localhost", 45678);
  private HostAndPort localHostAndPort =
      HostAndPort.fromParts("localhost", ServerConfig.DEFAULT_PORT);
  private int timeout = 5;

  private RegistryBuilder registryBuilder = Registry.builder();

  private ServerConfigBuilder serverConfigBuilder = ServerConfig.builder();
  private DefaultPathBinding pathBinding;

  @Override
  public RequestFixture body(byte[] bytes, String contentType) {
    requestHeaders.add(HttpHeaderNames.CONTENT_TYPE, contentType);
    requestHeaders.add(HttpHeaderNames.CONTENT_LENGTH, bytes.length);
    requestBody.capacity(bytes.length).writeBytes(bytes);
    return this;
  }

  @Override
  public RequestFixture body(String text, String contentType) {
    return body(text.getBytes(CharsetUtil.UTF_8), contentType);
  }

  @Override
  public RegistrySpec getRegistry() {
    return registryBuilder;
  }

  @Override
  public HandlingResult handle(Handler handler) throws HandlerTimeoutException {
    final DefaultHandlingResult.ResultsHolder results = new DefaultHandlingResult.ResultsHolder();
    return invoke(handler, getEffectiveRegistry(results), results);
  }

  @Override
  public HandlingResult handleChain(Action<? super Chain> chainAction) throws Exception {
    final DefaultHandlingResult.ResultsHolder results = new DefaultHandlingResult.ResultsHolder();
    Registry registry = getEffectiveRegistry(results);
    ServerConfig serverConfig = registry.get(ServerConfig.class);
    Handler handler = Handlers.chain(serverConfig, registry, chainAction);
    return invoke(handler, registry, results);
  }

  private HandlingResult invoke(
      Handler handler, Registry registry, DefaultHandlingResult.ResultsHolder results)
      throws HandlerTimeoutException {
    ServerConfig serverConfig = registry.get(ServerConfig.class);

    DefaultRequest request =
        new DefaultRequest(
            Instant.now(),
            requestHeaders,
            HttpMethod.valueOf(method.toUpperCase()),
            HttpVersion.valueOf(protocol),
            uri,
            new InetSocketAddress(remoteHostAndPort.getHostText(), remoteHostAndPort.getPort()),
            new InetSocketAddress(localHostAndPort.getHostText(), localHostAndPort.getPort()),
            serverConfig,
            new RequestBodyReader() {
              @Override
              public Promise<? extends ByteBuf> read(long maxContentLength, Block onTooLarge) {
                return Promise.value(requestBody)
                    .route(r -> r.readableBytes() > maxContentLength, onTooLarge.action());
              }

              @Override
              public TransformablePublisher<? extends ByteBuf> readStream(long maxContentLength) {
                return Streams.<ByteBuf>yield(
                    r -> {
                      if (r.getRequestNum() > 0) {
                        return null;
                      } else {
                        return requestBody;
                      }
                    });
              }
            });

    if (pathBinding != null) {
      handler =
          Handlers.chain(
              ctx -> {
                ctx.getExecution().get(PathBindingStorage.TYPE).push(pathBinding);
                ctx.next();
              },
              handler);
    }

    try {
      return new DefaultHandlingResult(
          request, results, responseHeaders, registry, timeout, handler);
    } catch (Exception e) {
      throw Exceptions.uncheck(e);
    } finally {
      registry.get(ExecController.class).close();
    }
  }

  @Override
  public RequestFixture header(CharSequence name, String value) {
    requestHeaders.add(name, value);
    return this;
  }

  @Override
  public RequestFixture serverConfig(Action<? super ServerConfigBuilder> action) throws Exception {
    serverConfigBuilder = ServerConfig.builder();
    action.execute(serverConfigBuilder);
    return this;
  }

  @Override
  public RequestFixture method(String method) {
    if (method == null) {
      throw new IllegalArgumentException("method must not be null");
    }
    this.method = method.toUpperCase();
    return this;
  }

  @Override
  public RequestFixture pathBinding(Map<String, String> pathTokens) {
    return pathBinding("", "", pathTokens);
  }

  @Override
  public RequestFixture pathBinding(
      String boundTo, String pastBinding, Map<String, String> pathTokens) {
    pathBinding =
        new DefaultPathBinding(
            pastBinding, ImmutableMap.copyOf(pathTokens), new RootPathBinding(pastBinding));
    return this;
  }

  @Override
  public RequestFixture registry(Action<? super RegistrySpec> action) throws Exception {
    action.execute(registryBuilder);
    return this;
  }

  @Override
  public RequestFixture responseHeader(CharSequence name, String value) {
    responseHeaders.add(name, value);
    return this;
  }

  @Override
  public RequestFixture timeout(int timeoutSeconds) {
    if (timeoutSeconds < 0) {
      throw new IllegalArgumentException("timeout must be > 0");
    }
    this.timeout = timeoutSeconds;
    return this;
  }

  @Override
  public RequestFixture uri(String uri) {
    if (uri == null) {
      throw new NullPointerException("uri cannot be null");
    }
    if (!uri.startsWith("/")) {
      uri = "/" + uri;
    }

    this.uri = uri;
    return this;
  }

  @Override
  public RequestFixture remoteAddress(HostAndPort remote) {
    remoteHostAndPort = remote;
    return this;
  }

  @Override
  public RequestFixture localAddress(HostAndPort local) {
    localHostAndPort = local;
    return this;
  }

  @Override
  public RequestFixture protocol(String protocol) {
    this.protocol = protocol;
    return this;
  }

  private Registry getEffectiveRegistry(final DefaultHandlingResult.ResultsHolder results) {

    ClientErrorHandler clientErrorHandler =
        (context, statusCode) -> {
          results.setClientError(statusCode);
          context.getResponse().status(statusCode);
          results.getLatch().countDown();
        };

    ServerErrorHandler serverErrorHandler =
        (context, throwable1) -> {
          results.setThrowable(throwable1);
          results.getLatch().countDown();
        };

    final Registry userRegistry =
        Registry.builder()
            .add(ClientErrorHandler.class, clientErrorHandler)
            .add(ServerErrorHandler.class, serverErrorHandler)
            .build();
    return Exceptions.uncheck(
        () -> {
          ServerConfig serverConfig = serverConfigBuilder.build();
          DefaultExecController execController =
              new DefaultExecController(serverConfig.getThreads());
          return ServerRegistry.serverRegistry(
              new TestServer(),
              Impositions.none(),
              execController,
              serverConfig,
              r -> userRegistry.join(registryBuilder.build()));
        });
  }

  // TODO some kind of impl here
  private static class TestServer implements RatpackServer {

    @Override
    public String getScheme() {
      throw new UnsupportedOperationException();
    }

    @Override
    public int getBindPort() {
      throw new UnsupportedOperationException();
    }

    @Override
    public String getBindHost() {
      throw new UnsupportedOperationException();
    }

    @Override
    public boolean isRunning() {
      throw new UnsupportedOperationException();
    }

    @Override
    public void start() throws Exception {
      throw new UnsupportedOperationException();
    }

    @Override
    public void stop() throws Exception {
      throw new UnsupportedOperationException();
    }

    @Override
    public RatpackServer reload() throws Exception {
      throw new UnsupportedOperationException();
    }
  }
}
Пример #10
0
 @Override
 public RequestFixture serverConfig(Action<? super ServerConfigBuilder> action) throws Exception {
   serverConfigBuilder = ServerConfig.builder();
   action.execute(serverConfigBuilder);
   return this;
 }