Added missing #disconnectEndpoint method to AsyncConnectExec

apache · Nov 30, 2023 · a46c9e5 · a46c9e5
1 parent 8a2b0f5
commit a46c9e5
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/httpclient5/src/main/java/org/apache/hc/client5/http/async/AsyncExecRuntime.java b/httpclient5/src/main/java/org/apache/hc/client5/http/async/AsyncExecRuntime.java
@@ -106,6 +106,11 @@ Cancellable connectEndpoint(
             HttpClientContext context,
             FutureCallback<AsyncExecRuntime> callback);
 
+    /**
+     * Disconnects the local endpoint from the initial hop in the connection route.
+     */
+    void disconnectEndpoint();
+
     /**
      * Upgrades transport security of the active connection by using the TLS security protocol.
      *

diff --git a/...ient5/src/main/java/org/apache/hc/client5/http/impl/async/InternalH2AsyncExecRuntime.java b/...ient5/src/main/java/org/apache/hc/client5/http/impl/async/InternalH2AsyncExecRuntime.java
@@ -215,6 +215,14 @@ public void cancelled() {
 
     }
 
+    @Override
+    public void disconnectEndpoint() {
+        final Endpoint endpoint = sessionRef.get();
+        if (endpoint != null) {
+            endpoint.session.close(CloseMode.GRACEFUL);
+        }
+    }
+
     @Override
     public void upgradeTls(final HttpClientContext context) {
         throw new UnsupportedOperationException();

diff --git a/...nt5/src/main/java/org/apache/hc/client5/http/impl/async/InternalHttpAsyncExecRuntime.java b/...nt5/src/main/java/org/apache/hc/client5/http/impl/async/InternalHttpAsyncExecRuntime.java
@@ -235,6 +235,14 @@ public void completed(final AsyncConnectionEndpoint endpoint) {
 
     }
 
+    @Override
+    public void disconnectEndpoint() {
+        final AsyncConnectionEndpoint endpoint = endpointRef.get();
+        if (endpoint != null) {
+            endpoint.close(CloseMode.GRACEFUL);
+        }
+    }
+
     @Override
     public void upgradeTls(final HttpClientContext context) {
         upgradeTls(context, null);