Attempt reconnect when connection to websocket fails, and pass down c…

…onnection exceptions to created Multi
smallrye · Sep 11, 2023 · 2dcc266 · 2dcc266
1 parent fd73c21
commit 2dcc266
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...rtx/src/main/java/io/smallrye/graphql/client/vertx/dynamic/VertxDynamicGraphQLClient.java b/...rtx/src/main/java/io/smallrye/graphql/client/vertx/dynamic/VertxDynamicGraphQLClient.java
@@ -409,6 +409,7 @@ private Uni<WebSocketSubprotocolHandler> webSocketHandler() {
                                         handlerEmitter.complete(handler);
                                         log.debug("Using websocket subprotocol handler: " + handler);
                                     } else {
+                                        webSocketHandler.set(null);
                                         handlerEmitter.fail(result.cause());
                                     }
                                 });
@@ -460,7 +461,7 @@ private Multi<Response> executeSubscriptionOverWebsocket(JsonObject json) {
             webSocketHandler().subscribe().with(handler -> {
                 handlerRef.set(handler);
                 operationId.set(handler.executeMulti(json, rawEmitter));
-            });
+            }, rawEmitter::fail);
         });
         return rawMulti
                 .onCancellation().invoke(() -> {