Merge pull request #50990 from geoand/quarkus-rest-close-handlers

Discard remaining REST handlers when the client connection is closed prematurely

Author: geoand

independent-projects/resteasy-reactive/server/runtime/src/main/java/org/jboss/resteasy/reactive/server/core/ResteasyReactiveRequestContext.java

@@ -212,6 +212,9 @@ public void restart(RuntimeResource target, boolean setLocatorTarget) {
     public void setupInitialMatchAndRestart(RequestMapper.RequestMatch<RestInitialHandler.InitialMatch> initialMatch) {
         this.initialMatch = initialMatch;
 
+        // add a default close handler that simply discards whatever REST handlers still remain to be run
+        serverResponse().addCloseHandler(new DiscardRemainingRunner(this));
+
         restart(initialMatch.value.handlers);
         setMaxPathParams(initialMatch.value.maxPathParams);
         setRemaining(initialMatch.remaining);
@@ -435,6 +438,14 @@ public void close() {
         super.close();
     }
 
+    /**
+     * This method ensures that no more handlers will run and that all the resources tied to the request are closed
+     */
+    private void discardRemaining() {
+        setPosition(getHandlers().length);
+        close();
+    }
+
     public LazyResponse getResponse() {
         return response;
     }
@@ -1330,4 +1341,24 @@ public PreviousResource(RuntimeResource locatorTarget, Object locatorPathParamVa
         private final PreviousResource prev;
 
     }
+
+    private static class DiscardRemainingRunner implements Runnable {
+
+        private ResteasyReactiveRequestContext context;
+
+        private DiscardRemainingRunner(ResteasyReactiveRequestContext context) {
+            this.context = context;
+        }
+
+        @Override
+        public void run() {
+            try {
+                context.discardRemaining();
+            } catch (Exception ignored) {
+
+            } finally {
+                context = null;
+            }
+        }
+    }
 }