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

…n fails
smallrye · Nov 15, 2023 · cf10ab8 · cf10ab8
1 parent 598a42f
commit cf10ab8
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
@@ -438,7 +438,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
@@ -186,7 +186,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(() -> {