test/jdk/java/net/httpclient/http3/H3SimpleGet.java - master - githubmirror/panama-vector

test/jdk/java/net/httpclient/http3/H3SimpleGet.java
374 строки17 KB
Daisuke Yamazaki
8372325: Refactor tests under jdk/java/net/httpclient to use ${test.main.class}
09 апр 2026, 13:51
09 апр 2026, 13:51261011a
/*
 * Copyright (c) 2015, 2026, 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 id=with-continuations
 * @bug 8087112
 * @requires os.family != "windows" | ( os.name != "Windows 10" & os.name != "Windows Server 2016"
 *                                      & os.name != "Windows Server 2019" )
 * @requires os.family != "aix"
 * @library /test/lib /test/jdk/java/net/httpclient/lib
 * @build jdk.test.lib.net.SimpleSSLContext jdk.httpclient.test.lib.common.TestUtil
 *        jdk.httpclient.test.lib.http2.Http2TestServer
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.httpclient.retryOnStreamlimit=20
 *                     -Djdk.httpclient.redirects.retrylimit=21
 *                     -Dsimpleget.repeat=1 -Dsimpleget.chunks=1 -Dsimpleget.requests=1000
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Dsimpleget.requests=150
 *                     -Dsimpleget.chunks=16384
 *                     -Djdk.httpclient.retryOnStreamlimit=5
 *                     -Djdk.httpclient.redirects.retrylimit=6
 *                     -Djdk.httpclient.quic.defaultMTU=16336
 *                      ${test.main.class}
 */

/*
 * @test id=with-continuations-aix
 * @bug 8087112
 * @requires os.family == "aix"
 * @library /test/lib /test/jdk/java/net/httpclient/lib
 * @build jdk.test.lib.net.SimpleSSLContext jdk.httpclient.test.lib.common.TestUtil
 *        jdk.httpclient.test.lib.http2.Http2TestServer
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.httpclient.retryOnStreamlimit=20
 *                     -Djdk.httpclient.redirects.retrylimit=21
 *                     -Dsimpleget.repeat=1 -Dsimpleget.chunks=1 -Dsimpleget.requests=1000
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Dsimpleget.requests=150
 *                     -Dsimpleget.chunks=16384
 *                     -Djdk.httpclient.retryOnStreamlimit=5
 *                     -Djdk.httpclient.redirects.retrylimit=6
 *                     -Djdk.httpclient.quic.defaultMTU=8192
 *                      ${test.main.class}
 */

/*
 * @test id=without-continuation
 * @bug 8087112
 * @requires os.family == "windows" & ( os.name == "Windows 10" | os.name == "Windows Server 2016"
 *                                     | os.name == "Windows Server 2019" )
 * @library /test/lib /test/jdk/java/net/httpclient/lib
 * @build jdk.test.lib.net.SimpleSSLContext jdk.httpclient.test.lib.common.TestUtil
 *        jdk.httpclient.test.lib.http2.Http2TestServer
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -XX:+UnlockExperimentalVMOptions -XX:-VMContinuations
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -XX:+UnlockExperimentalVMOptions -XX:-VMContinuations
 *                     -Djdk.httpclient.retryOnStreamlimit=20
 *                     -Djdk.httpclient.redirects.retrylimit=21
 *                     -Dsimpleget.repeat=1 -Dsimpleget.chunks=1 -Dsimpleget.requests=1000
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -XX:+UnlockExperimentalVMOptions -XX:-VMContinuations
 *                     -Dsimpleget.requests=150
 *                     -Dsimpleget.chunks=16384
 *                     -Djdk.httpclient.retryOnStreamlimit=5
 *                     -Djdk.httpclient.redirects.retrylimit=6
 *                     -Djdk.httpclient.quic.defaultMTU=16336
 *                      ${test.main.class}
 */

/*
 * @test id=useNioSelector
 * @bug 8087112
 * @requires os.family != "aix"
 * @library /test/lib /test/jdk/java/net/httpclient/lib
 * @build jdk.test.lib.net.SimpleSSLContext jdk.httpclient.test.lib.common.TestUtil
 *        jdk.httpclient.test.lib.http2.Http2TestServer
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.internal.httpclient.quic.useNioSelector=true
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.internal.httpclient.quic.useNioSelector=true
 *                     -Djdk.httpclient.retryOnStreamlimit=20
 *                     -Djdk.httpclient.redirects.retrylimit=21
 *                     -Dsimpleget.repeat=1 -Dsimpleget.chunks=1 -Dsimpleget.requests=1000
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.internal.httpclient.quic.useNioSelector=true
 *                     -Dsimpleget.requests=150
 *                     -Dsimpleget.chunks=16384
 *                     -Djdk.httpclient.retryOnStreamlimit=5
 *                     -Djdk.httpclient.redirects.retrylimit=6
 *                     -Djdk.httpclient.quic.defaultMTU=16336
 *                      ${test.main.class}
 */

/*
 * @test id=useNioSelector-aix
 * @bug 8087112
 * @requires os.family == "aix"
 * @library /test/lib /test/jdk/java/net/httpclient/lib
 * @build jdk.test.lib.net.SimpleSSLContext jdk.httpclient.test.lib.common.TestUtil
 *        jdk.httpclient.test.lib.http2.Http2TestServer
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.internal.httpclient.quic.useNioSelector=true
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.internal.httpclient.quic.useNioSelector=true
 *                     -Djdk.httpclient.retryOnStreamlimit=20
 *                     -Djdk.httpclient.redirects.retrylimit=21
 *                     -Dsimpleget.repeat=1 -Dsimpleget.chunks=1 -Dsimpleget.requests=1000
 *                     ${test.main.class}
 * @run junit/othervm/timeout=480 -XX:+HeapDumpOnOutOfMemoryError -XX:+CrashOnOutOfMemoryError
 *                     -Djdk.internal.httpclient.quic.useNioSelector=true
 *                     -Dsimpleget.requests=150
 *                     -Dsimpleget.chunks=16384
 *                     -Djdk.httpclient.retryOnStreamlimit=5
 *                     -Djdk.httpclient.redirects.retrylimit=6
 *                     -Djdk.httpclient.quic.defaultMTU=8192
 *                      ${test.main.class}
 */

/*
 * @test id=reno-cc
 * @bug 8087112
 * @library /test/lib /test/jdk/java/net/httpclient/lib
 * @build jdk.test.lib.net.SimpleSSLContext jdk.httpclient.test.lib.common.TestUtil
 *        jdk.httpclient.test.lib.http2.Http2TestServer
 * @run junit/othervm/timeout=480 -Djdk.internal.httpclient.quic.congestionController=reno
 *                     ${test.main.class}
 * @summary send multiple GET requests using Reno congestion controller
 */


// Interesting additional settings for debugging and manual testing:
// -----------------------------------------------------------------
// -Djdk.httpclient.HttpClient.log=requests,errors,quic:retransmit:control,http3
// -Djdk.httpclient.HttpClient.log=errors,requests,quic:all
// -Djdk.httpclient.quic.defaultMTU=64000
// -Djdk.httpclient.quic.defaultMTU=16384
// -Djdk.httpclient.quic.defaultMTU=4096
// -Djdk.httpclient.http3.maxStreamLimitTimeout=1375
// -Xmx16g
// -Djdk.httpclient.quic.defaultMTU=16384
// -Djdk.internal.httpclient.debug=err
// -XX:+HeapDumpOnOutOfMemoryError
// -Xmx768m -XX:MaxRAMPercentage=12.5
// -Djdk.httpclient.HttpClient.log=errors,requests,http3
// -Djdk.httpclient.HttpClient.log=errors,http3,quic:retransmit:control

import java.io.IOException;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpClient.Builder;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.net.http.HttpResponse.BodyHandlers;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ConcurrentSkipListSet;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.function.Predicate;
import javax.net.ssl.SSLContext;

import jdk.httpclient.test.lib.common.HttpServerAdapters;
import jdk.test.lib.net.SimpleSSLContext;

import static java.net.http.HttpClient.Version.HTTP_3;
import static java.net.http.HttpOption.Http3DiscoveryMode.HTTP_3_URI_ONLY;
import static java.net.http.HttpOption.H3_DISCOVERY;

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

public class H3SimpleGet implements HttpServerAdapters {
    static HttpTestServer httpsServer;
    static HttpClient client = null;
    private static final SSLContext sslContext = SimpleSSLContext.findSSLContext();
    static String httpsURIString;
    static ExecutorService serverExec =
            Executors.newThreadPerTaskExecutor(Thread.ofVirtual()
                    .name("server-vt-worker-", 1).factory());

    static void initialize() throws Exception {
        try {
            client = getClient();

            httpsServer = HttpTestServer.create(HTTP_3_URI_ONLY, sslContext, serverExec);
            httpsServer.addHandler(new TestHandler(), "/");
            httpsURIString = "https://" + httpsServer.serverAuthority() + "/bar/";

            httpsServer.start();
            warmup();
        } catch (Throwable e) {
            System.err.println("Throwing now");
            e.printStackTrace();
            throw e;
        }
    }

    private static void warmup() throws Exception {
        // warmup server
        try (var client2 = createClient(sslContext, Executors.newThreadPerTaskExecutor(
                Thread.ofVirtual().name("client-2-vt-worker", 1).factory()))) {
            HttpRequest request = HttpRequest.newBuilder(URI.create(httpsURIString))
                    .version(HTTP_3)
                    .setOption(H3_DISCOVERY, HTTP_3_URI_ONLY)
                    .HEAD().build();
            client2.send(request, BodyHandlers.ofByteArrayConsumer(b-> {}));
        }

        // warmup client
        var httpsServer2 = HttpTestServer.create(HTTP_3_URI_ONLY, sslContext,
                Executors.newThreadPerTaskExecutor(
                        Thread.ofVirtual().name("server-2-vt-worker", 1).factory()));
        httpsServer2.addHandler(new TestHandler(), "/");
        var httpsURIString2 = "https://" + httpsServer2.serverAuthority() + "/bar/";
        httpsServer2.start();
        try {
            HttpRequest request = HttpRequest.newBuilder(URI.create(httpsURIString2))
                    .version(HTTP_3)
                    .setOption(H3_DISCOVERY, HTTP_3_URI_ONLY)
                    .HEAD().build();
            client.send(request, BodyHandlers.ofByteArrayConsumer(b-> {}));
        } finally {
            httpsServer2.stop();
        }
    }

    public static void main(String[] args) throws Exception {
        new H3SimpleGet().test();
    }

    static volatile boolean waitBeforeTest = false;

    @Test
    public void test() throws Exception {
        try {
            if (waitBeforeTest) {
                Thread.sleep(20000);
            }
            long prestart = System.nanoTime();
            initialize();
            long done = System.nanoTime();
            System.out.println("Stat: Initialization and warmup took "
                    + TimeUnit.NANOSECONDS.toMillis(done-prestart)+" millis");
            HttpRequest request = HttpRequest.newBuilder(URI.create(httpsURIString))
                    .version(HTTP_3)
                    .setOption(H3_DISCOVERY, HTTP_3_URI_ONLY)
                    .GET().build();
            long start = System.nanoTime();
            var resp = client.send(request, BodyHandlers.ofByteArrayConsumer(b-> {}));
            Assertions.assertEquals(200, resp.statusCode());
            long elapsed = System.nanoTime() - start;
            System.out.println("Stat: First request took: " + elapsed
                    + " nanos (" + TimeUnit.NANOSECONDS.toMillis(elapsed) + " ms)");
            final int max = property("simpleget.requests", 50);
            List<CompletableFuture<HttpResponse<Void>>> list = new ArrayList<>(max);
            Set<String> connections = new ConcurrentSkipListSet<>();
            long start2 = System.nanoTime();
            for (int i = 0; i < max; i++) {
                var cf = client.sendAsync(request, BodyHandlers.ofByteArrayConsumer(b-> {}))
                        .whenComplete((r, t) -> Optional.ofNullable(r)
                        .flatMap(HttpResponse::connectionLabel)
                        .ifPresent(connections::add));
                list.add(cf);
                //cf.get(); // uncomment to test with serial instead of concurrent requests
            }
            try {
                CompletableFuture.allOf(list.toArray(new CompletableFuture[0])).join();
            } finally {
                long elapsed2 = System.nanoTime() - start2;
                long completed = list.stream().filter(CompletableFuture::isDone)
                        .filter(Predicate.not(CompletableFuture::isCompletedExceptionally)).count();
                connections.forEach(System.out::println);
                if (completed > 0) {
                    System.out.println("Stat: Next " + completed + " requests took: " + elapsed2 + " nanos ("
                            + TimeUnit.NANOSECONDS.toMillis(elapsed2) + "ms for " + completed + " requests): "
                            + elapsed2 / completed + " nanos per request ("
                            + TimeUnit.NANOSECONDS.toMillis(elapsed2) / completed + " ms) on "
                            + connections.size() + " connections");
                }
            }
            list.forEach((cf) -> Assertions.assertEquals(200, cf.join().statusCode()));
        } catch (Throwable tt) {
            System.err.println("tt caught");
            tt.printStackTrace();
            throw tt;
        } finally {
            httpsServer.stop();
        }
    }

    static HttpClient createClient(SSLContext sslContext, ExecutorService clientExec) {
        var builder = HttpServerAdapters.createClientBuilderForH3()
                .sslContext(sslContext)
                .version(HTTP_3)
                .proxy(Builder.NO_PROXY);
        if (clientExec != null) {
            builder = builder.executor(clientExec);
        }
        return builder.build();
    }

    static HttpClient getClient() {
        if (client == null) {
            client = createClient(sslContext, null);
        }
        return client;
    }

    static int property(String name, int defaultValue) {
        return Integer.parseInt(System.getProperty(name, String.valueOf(defaultValue)));
    }

    // 32 * 32 * 1024 * 10 chars = 10Mb responses
    // 50 requests   => 500Mb
    // 100 requests  => 1Gb
    // 1000 requests => 10Gb
    private final static int REPEAT = property("simpleget.repeat", 32);
    private final static String RESPONSE = "abcdefghij".repeat(property("simpleget.chunks", 1024*32));
    private final static byte[] RESPONSE_BYTES = RESPONSE.getBytes(StandardCharsets.UTF_8);

    private static class TestHandler implements HttpTestHandler {
        @Override
        public void handle(HttpTestExchange t) throws IOException {
            try (var in = t.getRequestBody()) {
                byte[] input = in.readAllBytes();
                t.sendResponseHeaders(200, RESPONSE_BYTES.length * REPEAT);
                try (var out = t.getResponseBody()) {
                    if (t.getRequestMethod().equals("HEAD")) return;
                    for (int i=0; i<REPEAT; i++) {
                        out.write(RESPONSE_BYTES);
                    }
                }
            }
        }
    }

}