apply timeout to both method signatures

cryostatio · Apr 22, 2023 · 7f60a77 · 7f60a77
1 parent 3c1f031
commit 7f60a77
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/io/cryostat/net/TargetConnectionManager.java b/src/main/java/io/cryostat/net/TargetConnectionManager.java
@@ -148,8 +148,7 @@ public class TargetConnectionManager {
 
     public <T> CompletableFuture<T> executeConnectedTaskAsync(
             ConnectionDescriptor connectionDescriptor, ConnectedTask<T> task) {
-        return executeConnectedTaskAsync(connectionDescriptor, task, this.executor)
-                .orTimeout(jmxConnectionTimeout, TimeUnit.SECONDS);
+        return executeConnectedTaskAsync(connectionDescriptor, task, this.executor);
     }
 
     public <T> CompletableFuture<T> executeConnectedTaskAsync(
@@ -168,7 +167,8 @@ public <T> CompletableFuture<T> executeConnectedTaskAsync(
                                 throw new CompletionException(e);
                             }
                         },
-                        ex);
+                        ex)
+                .orTimeout(jmxConnectionTimeout, TimeUnit.SECONDS);
     }
 
     public <T> T executeConnectedTask(