Skip to content

Commit

Permalink
Pending requests is accessed concurrently
Browse files Browse the repository at this point in the history
Signed-off-by: Jorge Bescos Gascon <jorge.bescos.gascon@oracle.com>
  • Loading branch information
jbescos committed Jul 8, 2021
1 parent ea56d57 commit 502d8b9
Show file tree
Hide file tree
Showing 2 changed files with 36 additions and 54 deletions.
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright (c) 2015, 2019 Oracle and/or its affiliates. All rights reserved.
* Copyright (c) 2015, 2021 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0, which is available at
Expand All @@ -20,7 +20,6 @@
import java.net.CookieManager;
import java.net.URI;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Queue;
import java.util.Set;
Expand All @@ -37,7 +36,7 @@ class DestinationConnectionPool {
private final Queue<HttpConnection> idleConnections = new ConcurrentLinkedDeque<>();
private final Set<HttpConnection> connections = Collections.newSetFromMap(new ConcurrentHashMap<>());
private final Queue<RequestRecord> pendingRequests = new ConcurrentLinkedDeque<>();
private final Map<HttpConnection, RequestRecord> requestsInProgress = new HashMap<>();
private final Map<HttpConnection, RequestRecord> requestsInProgress = new ConcurrentHashMap<>();
private final CookieManager cookieManager;
private final ScheduledExecutorService scheduler;
private final ConnectionStateListener connectionStateListener;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,73 +19,51 @@
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;

import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLParameters;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.core.Application;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriBuilder;

import org.glassfish.jersey.SslConfigurator;
import org.glassfish.jersey.client.ClientConfig;
import org.glassfish.jersey.internal.guava.ThreadFactoryBuilder;
import org.glassfish.jersey.jdk.connector.JdkConnectorProperties;
import org.glassfish.jersey.jdk.connector.JdkConnectorProvider;
import org.glassfish.jersey.server.ResourceConfig;
import org.glassfish.jersey.test.JerseyTest;
import org.glassfish.jersey.test.TestProperties;
import org.junit.Ignore;
import org.junit.Test;

public class StressTest extends JerseyTest {

private static final int SSL_PORT = 29999;
private static final int ITERATIONS = 10000;
// Must be less than the server thread pool. Is there any way to get that value?.
private static final int N_REQUESTS = 10;
private static final SSLContext SSL_CONTEXT;
private static final ExecutorService executor = Executors.newFixedThreadPool(N_REQUESTS);
private static final int PARALLELISM = 10;
private static final int IDLE_TIMEOUT = 50;
private static final int ITERATIONS = 1000;
private static CountDownLatch requests;
private static CountDownLatch latch;

static {
System.setProperty("javax.net.ssl.keyStore", SslFilterTest.class.getResource("/keystore_server").getPath());
System.setProperty("javax.net.ssl.keyStorePassword", "asdfgh");
System.setProperty("javax.net.ssl.trustStore", SslFilterTest.class.getResource("/truststore_server").getPath());
System.setProperty("javax.net.ssl.trustStorePassword", "asdfgh");
try {
SSL_CONTEXT = SslConfigurator.newInstance()
.trustStoreFile(SslFilterTest.class.getResource("/truststore_server").getPath())
.trustStorePassword("asdfgh")
.keyStoreFile(SslFilterTest.class.getResource("/keystore_server").getPath())
.keyStorePassword("asdfgh").createSSLContext();
} catch (Exception e) {
e.printStackTrace();
throw new IllegalStateException(e);
}
}

@Path("/test")
public static class TestResource {
@GET
@Path("/1")
public String test1() throws InterruptedException {
requests.countDown();
if (latch.await(100, TimeUnit.SECONDS)) {
if (latch.await(10, TimeUnit.SECONDS)) {
return "test1";
} else {
throw new IllegalStateException("Timeout");
}
}

@GET
@Path("/2")
public String test2() {
Expand All @@ -102,56 +80,61 @@ protected Application configure() {

@Override
protected void configureClient(ClientConfig config) {
config.property(JdkConnectorProperties.MAX_CONNECTIONS_PER_DESTINATION, PARALLELISM);
config.property(JdkConnectorProperties.CONNECTION_IDLE_TIMEOUT, IDLE_TIMEOUT);
config.connectorProvider(new JdkConnectorProvider());
}

@Override
protected Optional<SSLContext> getSslContext() {
return Optional.of(SSL_CONTEXT);
}

@Override
protected Optional<SSLParameters> getSslParameters() {
return Optional.of(SSL_CONTEXT.createSSLEngine("localhost", SSL_PORT).getSSLParameters());
}

@Override
protected URI getBaseUri() {
return UriBuilder.fromUri("https://localhost/").port(SSL_PORT).build();
}

@Test
public void hangAllRequestsStatus200() throws InterruptedException, ExecutionException {
assertEquals("https", getBaseUri().getScheme());
ExecutorService executor = Executors.newFixedThreadPool(PARALLELISM,
new ThreadFactoryBuilder().setNameFormat("client-%d").build());
for (int i = 0; i < ITERATIONS; i++) {
requests = new CountDownLatch(N_REQUESTS);
requests = new CountDownLatch(PARALLELISM);
latch = new CountDownLatch(1);
List<Future<Response>> responses = new ArrayList<>();
for (int j = 0; j < N_REQUESTS; j++) {
for (int j = 0; j < PARALLELISM; j++) {
Future<Response> future = executor.submit(() -> target("/test/1").request().get());
responses.add(future);
}
assertTrue(requests.await(100, TimeUnit.SECONDS));
assertTrue(requests.await(20, TimeUnit.SECONDS));
latch.countDown();
for (Future<Response> response : responses) {
assertEquals(200, response.get().getStatus());
}
}
executor.shutdown();
assertTrue(executor.awaitTermination(10, TimeUnit.SECONDS));
}

@Test
public void randomnessStatus200() throws InterruptedException, ExecutionException {
assertEquals("https", getBaseUri().getScheme());
ExecutorService executor = Executors.newFixedThreadPool(PARALLELISM,
new ThreadFactoryBuilder().setNameFormat("client-%d").build());
for (int i = 0; i < ITERATIONS; i++) {
System.out.println("Iteration " + i);
List<Future<Response>> responses = new ArrayList<>();
for (int j = 0; j < N_REQUESTS; j++) {
for (int j = 0; j < 100; j++) {
Future<Response> future = executor.submit(() -> target("/test/2").request().get());
responses.add(future);
}
for (Future<Response> response : responses) {
assertEquals(200, response.get().getStatus());
}
}
executor.shutdown();
assertTrue(executor.awaitTermination(10, TimeUnit.SECONDS));
}

@Test
public void syncTest() {
Response response = target("/test/2").request().get();
assertEquals(200, response.getStatus());
}

@Test
public void asyncTest() throws InterruptedException, ExecutionException {
Future<Response> response = target("/test/2").request().async().get();
assertEquals(200, response.get().getStatus());
}
}

0 comments on commit 502d8b9

Please sign in to comment.