fix the reconnect logic

edgedb · Oct 18, 2023 · 3be463b · 3be463b
1 parent dfb97dd
commit 3be463b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/driver/src/main/java/com/edgedb/driver/binary/duplexers/ChannelDuplexer.java b/src/driver/src/main/java/com/edgedb/driver/binary/duplexers/ChannelDuplexer.java
@@ -350,9 +350,9 @@ public CompletionStage<Void> disconnect() {
 
         if(this.channel.isOpen()) {
             return send(new Terminate())
-                    .thenCompose(v -> ChannelCompletableFuture.completeFrom(this.channel.disconnect()));
+                    .thenCompose(v -> ChannelCompletableFuture.completeFrom(this.channel.close()));
         }
 
-        return ChannelCompletableFuture.completeFrom(this.channel.disconnect());
+        return ChannelCompletableFuture.completeFrom(this.channel.close());
     }
 }
diff --git a/src/driver/src/main/java/com/edgedb/driver/clients/EdgeDBBinaryClient.java b/src/driver/src/main/java/com/edgedb/driver/clients/EdgeDBBinaryClient.java
@@ -299,7 +299,13 @@ public final CompletionStage<Void> executeQuery(
             @NotNull ExecutionArguments args
     ) {
         logger.debug("Execute request: is connected? {}", getDuplexer().isConnected());
-
+
+        if(!getDuplexer().isConnected()) {
+            // TODO: check for recursion
+            return connect()
+                    .thenCompose(v -> executeQuery(args));
+        }
+
         final var hasReleased = new AtomicBoolean();
 
         return CompletableFuture.runAsync(() -> {