Properly fail a single-result operation over WS when the WS connectio…

…n fails
smallrye · Nov 14, 2023 · 739dbb9 · 739dbb9
1 parent 7bdfb31
commit 739dbb9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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
@@ -460,7 +460,7 @@ private Uni<Response> executeSingleResultOperationOverWebsocket(JsonObject json)
             webSocketHandler().subscribe().with(handler -> {
                 handlerRef.set(handler);
                 operationId.set(handler.executeUni(json, rawEmitter));
-            });
+            }, rawEmitter::fail);
         });
         return rawUni
                 .onCancellation().invoke(() -> {

diff --git a/.../main/java/io/smallrye/graphql/client/vertx/typesafe/VertxTypesafeGraphQLClientProxy.java b/.../main/java/io/smallrye/graphql/client/vertx/typesafe/VertxTypesafeGraphQLClientProxy.java
@@ -215,7 +215,7 @@ private Uni<Object> executeSingleResultOperationOverWebsocket(MethodInvocation m
             webSocketHandler().subscribe().with((handler) -> {
                 handlerRef.set(handler);
                 operationId.set(handler.executeUni(request, rawEmitter));
-            });
+            }, rawEmitter::fail);
         });
         return rawUni
                 .onCancellation().invoke(() -> {