GraphQL to terminate the request even if it was active

Signed-off-by: Phillip Kruger <[email protected]>
michalvavrik · Jul 26, 2022 · 08e5c31 · 08e5c31
1 parent 1eb8047
commit 08e5c31
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 17 deletions.
diff --git a/...ime/src/main/java/io/quarkus/smallrye/graphql/runtime/SmallRyeGraphQLAbstractHandler.java b/...ime/src/main/java/io/quarkus/smallrye/graphql/runtime/SmallRyeGraphQLAbstractHandler.java
@@ -58,22 +58,18 @@ public void handle(Object e) {
     @Override
     public void handle(final RoutingContext ctx) {
 
-        if (currentManagedContext.isActive()) {
-            handleWithIdentity(ctx);
-        } else {
-
+        ctx.response()
+                .endHandler(currentManagedContextTerminationHandler)
+                .exceptionHandler(currentManagedContextTerminationHandler)
+                .closeHandler(currentManagedContextTerminationHandler);
+        if (!currentManagedContext.isActive()) {
             currentManagedContext.activate();
-            ctx.response()
-                    .endHandler(currentManagedContextTerminationHandler)
-                    .exceptionHandler(currentManagedContextTerminationHandler)
-                    .closeHandler(currentManagedContextTerminationHandler);
-
-            try {
-                handleWithIdentity(ctx);
-            } catch (Throwable t) {
-                currentManagedContext.terminate();
-                throw t;
-            }
+        }
+        try {
+            handleWithIdentity(ctx);
+        } catch (Throwable t) {
+            currentManagedContext.terminate();
+            throw t;
         }
     }
 

diff --git a/...me/src/main/java/io/quarkus/smallrye/graphql/runtime/SmallRyeGraphQLExecutionHandler.java b/...me/src/main/java/io/quarkus/smallrye/graphql/runtime/SmallRyeGraphQLExecutionHandler.java
@@ -196,8 +196,8 @@ private JsonObject getJsonObjectFromBody(RoutingContext ctx) throws IOException
     }
 
     private String readBody(RoutingContext ctx) {
-        if (ctx.getBody() != null) {
-            return ctx.getBodyAsString();
+        if (ctx.body() != null) {
+            return ctx.body().asString();
         }
         return null;
     }