Examples with Dynamic jakarta.servlet.ServletRegistration.Dynamic used on opensource projects

Example 6 with Dynamic

use of jakarta.servlet.ServletRegistration.Dynamic in project spring-boot by spring-projects.

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 7 with Dynamic

use of jakarta.servlet.ServletRegistration.Dynamic in project spring-boot by spring-projects.

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 8 with Dynamic

use of jakarta.servlet.ServletRegistration.Dynamic in project spring-boot by spring-projects.

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();
}