Examples with AbstractServletWebServerFactory cn.taketoday.framework.web.servlet.server.AbstractServletWebServerFactory used on opensource projects

Example 1 with AbstractServletWebServerFactory

use of cn.taketoday.framework.web.servlet.server.AbstractServletWebServerFactory in project today-infrastructure by TAKETODAY.

the class UndertowServletWebServerFactoryTests method whenServerIsShuttingDownGracefullyThenRequestsAreRejectedWithServiceUnavailable.

@Test
void whenServerIsShuttingDownGracefullyThenRequestsAreRejectedWithServiceUnavailable() throws Exception {
    AbstractServletWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingServlet blockingServlet = new BlockingServlet();
    this.webServer = factory.getWebServer((context) -> {
        Dynamic registration = context.addServlet("blockingServlet", blockingServlet);
        registration.addMapping("/blocking");
        registration.setAsyncSupported(true);
    });
    this.webServer.start();
    int port = this.webServer.getPort();
    Future<Object> request = initiateGetRequest(port, "/blocking");
    blockingServlet.awaitQueue();
    AtomicReference<GracefulShutdownResult> result = new AtomicReference<>();
    this.webServer.shutDownGracefully(result::set);
    assertThat(result.get()).isNull();
    blockingServlet.admitOne();
    assertThat(request.get()).isInstanceOf(HttpResponse.class);
    Object rejectedResult = initiateGetRequest(port, "/").get();
    assertThat(rejectedResult).isInstanceOf(HttpResponse.class);
    assertThat(((HttpResponse) rejectedResult).getStatusLine().getStatusCode()).isEqualTo(HttpStatus.SERVICE_UNAVAILABLE.value());
    this.webServer.stop();
}

Example 2 with AbstractServletWebServerFactory

use of cn.taketoday.framework.web.servlet.server.AbstractServletWebServerFactory in project today-infrastructure by TAKETODAY.

the class TomcatServletWebServerFactoryTests method whenServerIsShuttingDownARequestOnAnIdleConnectionResultsInConnectionReset.

@Test
void whenServerIsShuttingDownARequestOnAnIdleConnectionResultsInConnectionReset() throws Exception {
    AbstractServletWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingServlet blockingServlet = new BlockingServlet();
    this.webServer = factory.getWebServer((context) -> {
        Dynamic registration = context.addServlet("blockingServlet", blockingServlet);
        registration.addMapping("/blocking");
        registration.setAsyncSupported(true);
    });
    HttpClient httpClient = HttpClients.createMinimal();
    this.webServer.start();
    int port = this.webServer.getPort();
    Future<Object> keepAliveRequest = initiateGetRequest(httpClient, port, "/blocking");
    blockingServlet.awaitQueue();
    blockingServlet.admitOne();
    assertThat(keepAliveRequest.get()).isInstanceOf(HttpResponse.class);
    Future<Object> request = initiateGetRequest(port, "/blocking");
    blockingServlet.awaitQueue();
    this.webServer.shutDownGracefully((result) -> {
    });
    Object idleConnectionRequestResult = Awaitility.await().until(() -> {
        Future<Object> idleConnectionRequest = initiateGetRequest(httpClient, port, "/");
        Object result = idleConnectionRequest.get();
        return result;
    }, (result) -> result instanceof Exception);
    assertThat(idleConnectionRequestResult).isInstanceOfAny(SocketException.class, NoHttpResponseException.class);
    if (idleConnectionRequestResult instanceof SocketException) {
        assertThat((SocketException) idleConnectionRequestResult).hasMessage("Connection reset");
    }
    blockingServlet.admitOne();
    Object response = request.get();
    assertThat(response).isInstanceOf(HttpResponse.class);
    this.webServer.stop();
}

Example 3 with AbstractServletWebServerFactory

use of cn.taketoday.framework.web.servlet.server.AbstractServletWebServerFactory in project today-infrastructure by TAKETODAY.

the class JettyServletWebServerFactoryTests method whenServerIsShuttingDownGracefullyThenResponseToRequestOnIdleConnectionWillHaveAConnectionCloseHeader.

@Test
void whenServerIsShuttingDownGracefullyThenResponseToRequestOnIdleConnectionWillHaveAConnectionCloseHeader() throws Exception {
    AbstractServletWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingServlet blockingServlet = new BlockingServlet();
    this.webServer = factory.getWebServer((context) -> {
        Dynamic registration = context.addServlet("blockingServlet", blockingServlet);
        registration.addMapping("/blocking");
        registration.setAsyncSupported(true);
    });
    this.webServer.start();
    int port = this.webServer.getPort();
    HttpClient client = HttpClients.createMinimal();
    Future<Object> request = initiateGetRequest(client, port, "/blocking");
    blockingServlet.awaitQueue();
    blockingServlet.admitOne();
    Object response = request.get();
    assertThat(response).isInstanceOf(HttpResponse.class);
    assertThat(((HttpResponse) response).getStatusLine().getStatusCode()).isEqualTo(200);
    assertThat(((HttpResponse) response).getFirstHeader("Connection")).isNull();
    this.webServer.shutDownGracefully((result) -> {
    });
    request = initiateGetRequest(client, port, "/blocking");
    blockingServlet.awaitQueue();
    blockingServlet.admitOne();
    response = request.get();
    assertThat(response).isInstanceOf(HttpResponse.class);
    assertThat(((HttpResponse) response).getStatusLine().getStatusCode()).isEqualTo(200);
    assertThat(((HttpResponse) response).getFirstHeader("Connection")).isNotNull().extracting(Header::getValue).isEqualTo("close");
    this.webServer.stop();
}

Example 4 with AbstractServletWebServerFactory

use of cn.taketoday.framework.web.servlet.server.AbstractServletWebServerFactory in project today-infrastructure by TAKETODAY.

the class JettyServletWebServerFactoryTests method whenARequestCompletesAfterGracefulShutdownHasBegunThenItHasAConnectionCloseHeader.

@Test
void whenARequestCompletesAfterGracefulShutdownHasBegunThenItHasAConnectionCloseHeader() throws InterruptedException, ExecutionException {
    AbstractServletWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingServlet blockingServlet = new BlockingServlet();
    this.webServer = factory.getWebServer((context) -> {
        Dynamic registration = context.addServlet("blockingServlet", blockingServlet);
        registration.addMapping("/blocking");
        registration.setAsyncSupported(true);
    });
    this.webServer.start();
    int port = this.webServer.getPort();
    Future<Object> request = initiateGetRequest(port, "/blocking");
    blockingServlet.awaitQueue();
    AtomicReference<GracefulShutdownResult> result = new AtomicReference<>();
    this.webServer.shutDownGracefully(result::set);
    blockingServlet.admitOne();
    Awaitility.await().atMost(Duration.ofSeconds(5)).until(() -> GracefulShutdownResult.IDLE == result.get());
    Object requestResult = request.get();
    assertThat(requestResult).isInstanceOf(HttpResponse.class);
    assertThat(((HttpResponse) requestResult).getFirstHeader("Connection").getValue()).isEqualTo("close");
}

Example 5 with AbstractServletWebServerFactory

use of cn.taketoday.framework.web.servlet.server.AbstractServletWebServerFactory in project today-infrastructure by TAKETODAY.

the class JettyServletWebServerFactoryTests method whenServerIsShuttingDownGracefullyThenNewConnectionsCannotBeMade.

@Test
void whenServerIsShuttingDownGracefullyThenNewConnectionsCannotBeMade() throws Exception {
    AbstractServletWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingServlet blockingServlet = new BlockingServlet();
    this.webServer = factory.getWebServer((context) -> {
        Dynamic registration = context.addServlet("blockingServlet", blockingServlet);
        registration.addMapping("/blocking");
        registration.setAsyncSupported(true);
    });
    this.webServer.start();
    int port = this.webServer.getPort();
    Future<Object> request = initiateGetRequest(port, "/blocking");
    blockingServlet.awaitQueue();
    this.webServer.shutDownGracefully((result) -> {
    });
    Future<Object> unconnectableRequest = initiateGetRequest(port, "/");
    blockingServlet.admitOne();
    Object response = request.get();
    assertThat(response).isInstanceOf(HttpResponse.class);
    assertThat(unconnectableRequest.get()).isInstanceOf(HttpHostConnectException.class);
    this.webServer.stop();
}