37811b7460
Co-authored-by: Daniel Fuchs <daniel.fuchs@oracle.com> Co-authored-by: Michael McMahon <michael.x.mcmahon@oracle.com> Co-authored-by: Pavel Rappo <pavel.rappo@oracle.com> Reviewed-by: chegar, dfuchs, michaelm
282 lines
11 KiB
Java
282 lines
11 KiB
Java
/*
|
|
* Copyright (c) 2015, 2017, Oracle and/or its affiliates. All rights reserved.
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
*
|
|
* This code is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 only, as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
|
* or visit www.oracle.com if you need additional information or have any
|
|
* questions.
|
|
*/
|
|
|
|
/*
|
|
* @test
|
|
* @bug 8087112 8180044
|
|
* @modules jdk.incubator.httpclient
|
|
* java.logging
|
|
* jdk.httpserver
|
|
* @library /lib/testlibrary/ /
|
|
* @build jdk.testlibrary.SimpleSSLContext
|
|
* @compile ../../../com/sun/net/httpserver/LogFilter.java
|
|
* @compile ../../../com/sun/net/httpserver/EchoHandler.java
|
|
* @compile ../../../com/sun/net/httpserver/FileServerHandler.java
|
|
* @run main/othervm/timeout=40 -Djdk.httpclient.HttpClient.log=ssl ManyRequests
|
|
* @run main/othervm/timeout=40 -Dtest.insertDelay=true ManyRequests
|
|
* @run main/othervm/timeout=40 -Dtest.chunkSize=64 ManyRequests
|
|
* @run main/othervm/timeout=40 -Dtest.insertDelay=true -Dtest.chunkSize=64 ManyRequests
|
|
* @summary Send a large number of requests asynchronously
|
|
*/
|
|
// * @run main/othervm/timeout=40 -Djdk.httpclient.HttpClient.log=ssl ManyRequests
|
|
|
|
import com.sun.net.httpserver.HttpsConfigurator;
|
|
import com.sun.net.httpserver.HttpsParameters;
|
|
import com.sun.net.httpserver.HttpsServer;
|
|
import com.sun.net.httpserver.HttpExchange;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import jdk.incubator.http.HttpClient;
|
|
import jdk.incubator.http.HttpRequest;
|
|
import java.net.InetSocketAddress;
|
|
import java.net.URI;
|
|
import java.util.Arrays;
|
|
import java.util.Formatter;
|
|
import java.util.HashMap;
|
|
import java.util.LinkedList;
|
|
import java.util.Random;
|
|
import java.util.logging.Logger;
|
|
import java.util.logging.Level;
|
|
import java.util.concurrent.CompletableFuture;
|
|
import javax.net.ssl.SSLContext;
|
|
import jdk.testlibrary.SimpleSSLContext;
|
|
import static jdk.incubator.http.HttpRequest.BodyPublisher.fromByteArray;
|
|
import static jdk.incubator.http.HttpResponse.BodyHandler.asByteArray;
|
|
|
|
public class ManyRequests {
|
|
|
|
volatile static int counter = 0;
|
|
|
|
public static void main(String[] args) throws Exception {
|
|
Logger logger = Logger.getLogger("com.sun.net.httpserver");
|
|
logger.setLevel(Level.ALL);
|
|
logger.info("TEST");
|
|
System.out.println("Sending " + REQUESTS
|
|
+ " requests; delay=" + INSERT_DELAY
|
|
+ ", chunks=" + CHUNK_SIZE
|
|
+ ", XFixed=" + XFIXED);
|
|
SSLContext ctx = new SimpleSSLContext().get();
|
|
|
|
InetSocketAddress addr = new InetSocketAddress(0);
|
|
HttpsServer server = HttpsServer.create(addr, 0);
|
|
server.setHttpsConfigurator(new Configurator(ctx));
|
|
|
|
HttpClient client = HttpClient.newBuilder()
|
|
.sslContext(ctx)
|
|
.build();
|
|
try {
|
|
test(server, client);
|
|
System.out.println("OK");
|
|
} finally {
|
|
server.stop(0);
|
|
}
|
|
}
|
|
|
|
//static final int REQUESTS = 1000;
|
|
static final int REQUESTS = 20;
|
|
static final boolean INSERT_DELAY = Boolean.getBoolean("test.insertDelay");
|
|
static final int CHUNK_SIZE = Math.max(0,
|
|
Integer.parseInt(System.getProperty("test.chunkSize", "0")));
|
|
static final boolean XFIXED = Boolean.getBoolean("test.XFixed");
|
|
|
|
static class TestEchoHandler extends EchoHandler {
|
|
final Random rand = new Random();
|
|
@Override
|
|
public void handle(HttpExchange e) throws IOException {
|
|
System.out.println("Server: received " + e.getRequestURI());
|
|
super.handle(e);
|
|
}
|
|
@Override
|
|
protected void close(HttpExchange t, OutputStream os) throws IOException {
|
|
if (INSERT_DELAY) {
|
|
try { Thread.sleep(rand.nextInt(200)); }
|
|
catch (InterruptedException e) {}
|
|
}
|
|
System.out.println("Server: close outbound: " + t.getRequestURI());
|
|
super.close(t, os);
|
|
}
|
|
@Override
|
|
protected void close(HttpExchange t, InputStream is) throws IOException {
|
|
if (INSERT_DELAY) {
|
|
try { Thread.sleep(rand.nextInt(200)); }
|
|
catch (InterruptedException e) {}
|
|
}
|
|
System.out.println("Server: close inbound: " + t.getRequestURI());
|
|
super.close(t, is);
|
|
}
|
|
}
|
|
|
|
static void test(HttpsServer server, HttpClient client) throws Exception {
|
|
int port = server.getAddress().getPort();
|
|
URI baseURI = new URI("https://127.0.0.1:" + port + "/foo/x");
|
|
server.createContext("/foo", new TestEchoHandler());
|
|
server.start();
|
|
|
|
RequestLimiter limiter = new RequestLimiter(40);
|
|
Random rand = new Random();
|
|
CompletableFuture<?>[] results = new CompletableFuture<?>[REQUESTS];
|
|
HashMap<HttpRequest,byte[]> bodies = new HashMap<>();
|
|
|
|
for (int i=0; i<REQUESTS; i++) {
|
|
byte[] buf = new byte[(i+1)*CHUNK_SIZE+i+1]; // different size bodies
|
|
rand.nextBytes(buf);
|
|
URI uri = new URI(baseURI.toString() + String.valueOf(i+1));
|
|
HttpRequest r = HttpRequest.newBuilder(uri)
|
|
.header("XFixed", "true")
|
|
.POST(fromByteArray(buf))
|
|
.build();
|
|
bodies.put(r, buf);
|
|
|
|
results[i] =
|
|
limiter.whenOkToSend()
|
|
.thenCompose((v) -> {
|
|
System.out.println("Client: sendAsync: " + r.uri());
|
|
return client.sendAsync(r, asByteArray());
|
|
})
|
|
.thenCompose((resp) -> {
|
|
limiter.requestComplete();
|
|
if (resp.statusCode() != 200) {
|
|
String s = "Expected 200, got: " + resp.statusCode();
|
|
System.out.println(s + " from "
|
|
+ resp.request().uri().getPath());
|
|
return completedWithIOException(s);
|
|
} else {
|
|
counter++;
|
|
System.out.println("Result (" + counter + ") from "
|
|
+ resp.request().uri().getPath());
|
|
}
|
|
return CompletableFuture.completedStage(resp.body())
|
|
.thenApply((b) -> new Pair<>(resp, b));
|
|
})
|
|
.thenAccept((pair) -> {
|
|
HttpRequest request = pair.t.request();
|
|
byte[] requestBody = bodies.get(request);
|
|
check(Arrays.equals(requestBody, pair.u),
|
|
"bodies not equal:[" + bytesToHexString(requestBody)
|
|
+ "] [" + bytesToHexString(pair.u) + "]");
|
|
|
|
});
|
|
}
|
|
|
|
// wait for them all to complete and throw exception in case of error
|
|
CompletableFuture.allOf(results).join();
|
|
}
|
|
|
|
static <T> CompletableFuture<T> completedWithIOException(String message) {
|
|
return CompletableFuture.failedFuture(new IOException(message));
|
|
}
|
|
|
|
static String bytesToHexString(byte[] bytes) {
|
|
if (bytes == null)
|
|
return "null";
|
|
|
|
StringBuilder sb = new StringBuilder(bytes.length * 2);
|
|
|
|
Formatter formatter = new Formatter(sb);
|
|
for (byte b : bytes) {
|
|
formatter.format("%02x", b);
|
|
}
|
|
|
|
return sb.toString();
|
|
}
|
|
|
|
static final class Pair<T,U> {
|
|
Pair(T t, U u) {
|
|
this.t = t; this.u = u;
|
|
}
|
|
T t;
|
|
U u;
|
|
}
|
|
|
|
/**
|
|
* A simple limiter for controlling the number of requests to be run in
|
|
* parallel whenOkToSend() is called which returns a CF<Void> that allows
|
|
* each individual request to proceed, or block temporarily (blocking occurs
|
|
* on the waiters list here. As each request actually completes
|
|
* requestComplete() is called to notify this object, and allow some
|
|
* requests to continue.
|
|
*/
|
|
static class RequestLimiter {
|
|
|
|
static final CompletableFuture<Void> COMPLETED_FUTURE =
|
|
CompletableFuture.completedFuture(null);
|
|
|
|
final int maxnumber;
|
|
final LinkedList<CompletableFuture<Void>> waiters;
|
|
int number;
|
|
boolean blocked;
|
|
|
|
RequestLimiter(int maximum) {
|
|
waiters = new LinkedList<>();
|
|
maxnumber = maximum;
|
|
}
|
|
|
|
synchronized void requestComplete() {
|
|
number--;
|
|
// don't unblock until number of requests has halved.
|
|
if ((blocked && number <= maxnumber / 2) ||
|
|
(!blocked && waiters.size() > 0)) {
|
|
int toRelease = Math.min(maxnumber - number, waiters.size());
|
|
for (int i=0; i<toRelease; i++) {
|
|
CompletableFuture<Void> f = waiters.remove();
|
|
number ++;
|
|
f.complete(null);
|
|
}
|
|
blocked = number >= maxnumber;
|
|
}
|
|
}
|
|
|
|
synchronized CompletableFuture<Void> whenOkToSend() {
|
|
if (blocked || number + 1 >= maxnumber) {
|
|
blocked = true;
|
|
CompletableFuture<Void> r = new CompletableFuture<>();
|
|
waiters.add(r);
|
|
return r;
|
|
} else {
|
|
number++;
|
|
return COMPLETED_FUTURE;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void check(boolean cond, Object... msg) {
|
|
if (cond)
|
|
return;
|
|
StringBuilder sb = new StringBuilder();
|
|
for (Object o : msg)
|
|
sb.append(o);
|
|
throw new RuntimeException(sb.toString());
|
|
}
|
|
|
|
static class Configurator extends HttpsConfigurator {
|
|
public Configurator(SSLContext ctx) {
|
|
super(ctx);
|
|
}
|
|
|
|
public void configure(HttpsParameters params) {
|
|
params.setSSLParameters(getSSLContext().getSupportedSSLParameters());
|
|
}
|
|
}
|
|
}
|