mirror of
https://github.com/openjdk/jdk.git
synced 2026-06-06 18:53:37 +00:00
225 lines
9.4 KiB
Java
225 lines
9.4 KiB
Java
/*
|
|
* Copyright (c) 2025, 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.
|
|
*/
|
|
|
|
import java.net.URI;
|
|
import java.net.http.HttpClient;
|
|
import java.net.http.HttpRequest;
|
|
import java.net.http.HttpRequest.BodyPublishers;
|
|
import java.net.http.HttpResponse;
|
|
import java.net.http.HttpResponse.BodyHandlers;
|
|
import java.util.HashSet;
|
|
import java.util.Set;
|
|
import java.util.stream.Stream;
|
|
import javax.net.ssl.SSLContext;
|
|
|
|
import jdk.httpclient.test.lib.common.HttpServerAdapters;
|
|
import jdk.test.lib.net.SimpleSSLContext;
|
|
|
|
import org.junit.jupiter.api.AfterAll;
|
|
import org.junit.jupiter.api.BeforeAll;
|
|
import org.junit.jupiter.api.Test;
|
|
import org.junit.jupiter.api.Assertions;
|
|
|
|
import static java.net.http.HttpClient.Builder.NO_PROXY;
|
|
import static java.net.http.HttpClient.Version.HTTP_2;
|
|
|
|
/*
|
|
* @test id=default
|
|
* @bug 8372159
|
|
* @summary Verifies whether `SelectorManager` uses virtual threads
|
|
* as expected when no explicit configuration is provided
|
|
* @library /test/lib /test/jdk/java/net/httpclient/lib
|
|
* @build jdk.test.lib.net.SimpleSSLContext
|
|
* jdk.httpclient.test.lib.common.HttpServerAdapters
|
|
* @run junit/othervm
|
|
* -Djdk.httpclient.HttpClient.log=requests,responses,headers,errors
|
|
* H2SelectorVTTest
|
|
*/
|
|
/*
|
|
* @test id=never
|
|
* @bug 8372159
|
|
* @summary Verifies that `SelectorManager` does *not* use virtual threads
|
|
when explicitly configured to "never" use them
|
|
* @library /test/lib /test/jdk/java/net/httpclient/lib
|
|
* @build jdk.test.lib.net.SimpleSSLContext
|
|
* jdk.httpclient.test.lib.common.HttpServerAdapters
|
|
* @run junit/othervm
|
|
* -Djdk.internal.httpclient.tcp.selector.useVirtualThreads=never
|
|
* -Djdk.httpclient.HttpClient.log=requests,responses,headers,errors
|
|
* H2SelectorVTTest
|
|
*/
|
|
/*
|
|
* @test id=always
|
|
* @bug 8372159
|
|
* @summary Verifies that `SelectorManager` does *always* use virtual threads
|
|
when explicitly configured to "always" use them
|
|
* @library /test/lib /test/jdk/java/net/httpclient/lib
|
|
* @build jdk.test.lib.net.SimpleSSLContext
|
|
* jdk.httpclient.test.lib.common.HttpServerAdapters
|
|
* @run junit/othervm
|
|
* -Djdk.internal.httpclient.tcp.selector.useVirtualThreads=always
|
|
* -Djdk.httpclient.HttpClient.log=requests,responses,headers,errors
|
|
* H2SelectorVTTest
|
|
*/
|
|
/*
|
|
* @test id=explicit-default
|
|
* @bug 8372159
|
|
* @summary Verifies whether `SelectorManager` uses virtual threads
|
|
* as expected when `default` is explicitly configured
|
|
* @library /test/lib /test/jdk/java/net/httpclient/lib
|
|
* @build jdk.test.lib.net.SimpleSSLContext
|
|
* jdk.httpclient.test.lib.common.HttpServerAdapters
|
|
* @run junit/othervm
|
|
* -Djdk.internal.httpclient.tcp.selector.useVirtualThreads=default
|
|
* -Djdk.httpclient.HttpClient.log=requests,responses,headers,errors
|
|
* H2SelectorVTTest
|
|
*/
|
|
/*
|
|
* @test id=garbage
|
|
* @bug 8372159
|
|
* @summary Verifies whether `SelectorManager` uses virtual threads when
|
|
it is configured using an invalid value
|
|
* @library /test/lib /test/jdk/java/net/httpclient/lib
|
|
* @build jdk.test.lib.net.SimpleSSLContext
|
|
* jdk.httpclient.test.lib.common.HttpServerAdapters
|
|
* @run junit/othervm
|
|
* -Djdk.internal.httpclient.tcp.selector.useVirtualThreads=garbage
|
|
* -Djdk.httpclient.HttpClient.log=requests,responses,headers,errors
|
|
* H2SelectorVTTest
|
|
*/
|
|
// -Djava.security.debug=all
|
|
class H2SelectorVTTest implements HttpServerAdapters {
|
|
|
|
private static final SSLContext sslContext = SimpleSSLContext.findSSLContext();
|
|
private static HttpTestServer h2Server;
|
|
private static String requestURI;
|
|
|
|
enum UseVTForSelector { ALWAYS, NEVER, DEFAULT }
|
|
private static final String PROP_NAME = "jdk.internal.httpclient.tcp.selector.useVirtualThreads";
|
|
private static final UseVTForSelector USE_VT_FOR_SELECTOR;
|
|
static {
|
|
String useVtForSelector =
|
|
System.getProperty(PROP_NAME, "default");
|
|
USE_VT_FOR_SELECTOR = Stream.of(UseVTForSelector.values())
|
|
.filter((v) -> v.name().equalsIgnoreCase(useVtForSelector))
|
|
.findFirst().orElse(UseVTForSelector.DEFAULT);
|
|
}
|
|
|
|
private static boolean isTCPSelectorThreadVirtual() {
|
|
return switch (USE_VT_FOR_SELECTOR) {
|
|
case ALWAYS -> true;
|
|
case NEVER -> false;
|
|
default -> true;
|
|
};
|
|
}
|
|
|
|
@BeforeAll
|
|
static void beforeClass() throws Exception {
|
|
// create a h2 server
|
|
h2Server = HttpTestServer.create(HTTP_2, sslContext);
|
|
h2Server.addHandler((exchange) -> exchange.sendResponseHeaders(200, 0), "/hello");
|
|
h2Server.start();
|
|
System.out.println("Server started at " + h2Server.getAddress());
|
|
requestURI = "https://" + h2Server.serverAuthority() + "/hello";
|
|
}
|
|
|
|
@AfterAll
|
|
static void afterClass() throws Exception {
|
|
if (h2Server != null) {
|
|
System.out.println("Stopping server " + h2Server.getAddress());
|
|
h2Server.stop();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Issues various HTTP/2 requests and verifies the responses are received
|
|
*/
|
|
@Test
|
|
void testBasicRequests() throws Exception {
|
|
try (final HttpClient client = HttpClient.newBuilder()
|
|
.proxy(NO_PROXY)
|
|
.sslContext(sslContext).build()) {
|
|
final URI reqURI = new URI(requestURI);
|
|
final HttpRequest.Builder reqBuilder = HttpRequest.newBuilder(reqURI);
|
|
|
|
// GET
|
|
final HttpRequest req1 = reqBuilder.copy().GET().build();
|
|
System.out.println("\nIssuing request: " + req1);
|
|
final HttpResponse<?> resp1 = client.send(req1, BodyHandlers.ofString());
|
|
Assertions.assertEquals(200, resp1.statusCode(), "unexpected response code for GET request");
|
|
assertSelectorThread(client);
|
|
|
|
// POST
|
|
final HttpRequest req2 = reqBuilder.copy().POST(BodyPublishers.ofString("foo")).build();
|
|
System.out.println("\nIssuing request: " + req2);
|
|
final HttpResponse<?> resp2 = client.send(req2, BodyHandlers.ofString());
|
|
Assertions.assertEquals(200, resp2.statusCode(), "unexpected response code for POST request");
|
|
assertSelectorThread(client);
|
|
|
|
// HEAD
|
|
final HttpRequest req3 = reqBuilder.copy().HEAD().build();
|
|
System.out.println("\nIssuing request: " + req3);
|
|
final HttpResponse<?> resp3 = client.send(req3, BodyHandlers.ofString());
|
|
Assertions.assertEquals(200, resp3.statusCode(), "unexpected response code for HEAD request");
|
|
assertSelectorThread(client);
|
|
}
|
|
}
|
|
|
|
// This method attempts to determine whether the selector thread
|
|
// is a platform thread or a virtual thread, and throws if expectations
|
|
// ar not met.
|
|
// Since we don't have access to the selector thread, the method
|
|
// uses a roundabout way to figure this out: it enumerates all
|
|
// platform threads, and if it finds a thread whose name matches
|
|
// the expected name of the selector thread it concludes that the
|
|
// selector thread is a platform thread. Otherwise, it assumes
|
|
// that the thread is virtual.
|
|
private static void assertSelectorThread(HttpClient client) {
|
|
String cname = client.toString();
|
|
String clientId = cname.substring(cname.indexOf('(') + 1, cname.length() -1);
|
|
String name = "HttpClient-" + clientId + "-SelectorManager";
|
|
Set<String> threads = new HashSet<>(Thread.getAllStackTraces().keySet().stream()
|
|
.map(Thread::getName)
|
|
.toList());
|
|
boolean found = threads.contains(name);
|
|
String status = found == isTCPSelectorThreadVirtual() ? "ERROR" : "SUCCESS";
|
|
String propval = System.getProperty(PROP_NAME);
|
|
if (propval == null) {
|
|
System.out.printf("%s not defined, virtual=%s, thread found=%s%n",
|
|
PROP_NAME, isTCPSelectorThreadVirtual(), found);
|
|
} else {
|
|
System.out.printf("%s=%s, virtual=%s, thread found=%s%n",
|
|
PROP_NAME, propval, isTCPSelectorThreadVirtual(), found);
|
|
}
|
|
final String msg;
|
|
if (found) {
|
|
msg = "%s found in %s".formatted(name, threads);
|
|
System.out.printf("%s: %s%n", status, msg);
|
|
} else {
|
|
msg = "%s not found in %s".formatted(name, threads);
|
|
System.out.printf("%s: %s%n", status, msg);
|
|
}
|
|
Assertions.assertEquals(!isTCPSelectorThreadVirtual(), found, msg);
|
|
}
|
|
}
|