Added missing #disconnectEndpoint method to AsyncConnectExec

ok2c · ok2c · commit a46c9e56568b · 2023-11-30T13:33:04.000+01:00
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/httpclient5/src/main/java/org/apache/hc/client5/http/impl/async/InternalH2AsyncExecRuntime.java b/httpclient5/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/httpclient5/src/main/java/org/apache/hc/client5/http/impl/async/InternalHttpAsyncExecRuntime.java b/httpclient5/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);
