Skip to content

Commit c45caf5

Browse files
committed
Use the Upgrade flow by default
Signed-off-by: Andriy Redko <andriy.redko@aiven.io>
1 parent f63b0a2 commit c45caf5

8 files changed

+16
-13
lines changed
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
ce913081e592ee8eeee35c4e577d7dce13cba7a4
1+
ce913081e592ee8eeee35c4e577d7dce13cba7a4
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
eaf64237945d7d0f301d48420e8bdb7f565a7b0e
1+
eaf64237945d7d0f301d48420e8bdb7f565a7b0e
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
760c34db3ba41b0ffa07e956bc308d3a12356915
1+
760c34db3ba41b0ffa07e956bc308d3a12356915
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
c4c0c3c7bbcb0db54aa7ddd39e34a835428c99c0
1+
c4c0c3c7bbcb0db54aa7ddd39e34a835428c99c0

client/rest/src/main/java/org/opensearch/client/RestClientBuilder.java

+1-6
Original file line numberDiff line numberDiff line change
@@ -308,16 +308,11 @@ private CloseableHttpAsyncClient createHttpClient() {
308308
// default timeouts are all infinite
309309
RequestConfig.Builder requestConfigBuilder = RequestConfig.custom()
310310
.setConnectTimeout(Timeout.ofMilliseconds(DEFAULT_CONNECT_TIMEOUT_MILLIS))
311-
.setResponseTimeout(Timeout.ofMilliseconds(DEFAULT_RESPONSE_TIMEOUT_MILLIS))
312-
.setProtocolUpgradeEnabled(false);
311+
.setResponseTimeout(Timeout.ofMilliseconds(DEFAULT_RESPONSE_TIMEOUT_MILLIS));
313312
if (requestConfigCallback != null) {
314313
requestConfigBuilder = requestConfigCallback.customizeRequestConfig(requestConfigBuilder);
315314
}
316315
RequestConfig requestConfig = requestConfigBuilder.build();
317-
if (requestConfig.isProtocolUpgradeEnabled()) {
318-
throw new IllegalArgumentException("protocol upgrade is not supported");
319-
}
320-
321316
try {
322317
final TlsStrategy tlsStrategy = ClientTlsStrategyBuilder.create()
323318
.setSslContext(SSLContext.getDefault())

client/rest/src/test/java/org/opensearch/client/RestClientSingleHostIntegTests.java

+9-1
Original file line numberDiff line numberDiff line change
@@ -382,6 +382,10 @@ public void testHeaders() throws Exception {
382382
if (method.equals("HEAD") == false) {
383383
standardHeaders.add("Content-length");
384384
}
385+
if (method.equals("HEAD") == true || method.equals("GET") == true || method.equals("OPTIONS") == true) {
386+
standardHeaders.add("Upgrade");
387+
}
388+
385389
final Header[] requestHeaders = RestClientTestUtil.randomHeaders(getRandom(), "Header");
386390
final int statusCode = randomStatusCode(getRandom());
387391
Request request = new Request(method, "/" + statusCode);
@@ -400,11 +404,15 @@ public void testHeaders() throws Exception {
400404
assertEquals(method, esResponse.getRequestLine().getMethod());
401405
assertEquals(statusCode, esResponse.getStatusLine().getStatusCode());
402406
assertEquals(pathPrefix + "/" + statusCode, esResponse.getRequestLine().getUri());
407+
403408
assertHeaders(defaultHeaders, requestHeaders, esResponse.getHeaders(), standardHeaders);
409+
final Set<String> removedHeaders = new HashSet<>();
404410
for (final Header responseHeader : esResponse.getHeaders()) {
405411
String name = responseHeader.getName();
406-
if (name.startsWith("Header") == false) {
412+
// Some headers could be returned multiple times in response, like Connection fe.
413+
if (name.startsWith("Header") == false && removedHeaders.contains(name) == false) {
407414
assertTrue("unknown header was returned " + name, standardHeaders.remove(name));
415+
removedHeaders.add(name);
408416
}
409417
}
410418
assertTrue("some expected standard headers weren't returned: " + standardHeaders, standardHeaders.isEmpty());
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
ce913081e592ee8eeee35c4e577d7dce13cba7a4
1+
ce913081e592ee8eeee35c4e577d7dce13cba7a4
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
eaf64237945d7d0f301d48420e8bdb7f565a7b0e
1+
eaf64237945d7d0f301d48420e8bdb7f565a7b0e

0 commit comments

Comments
 (0)