Avoid retaining references after successful cancelation

Fixes #139
jbossas · Feb 13, 2024 · 93c7f80 · 93c7f80
1 parent 56ebd23
commit 93c7f80
Showing 1 changed file with 34 additions and 6 deletions.
diff --git a/src/main/java/org/jboss/threads/EnhancedQueueExecutor.java b/src/main/java/org/jboss/threads/EnhancedQueueExecutor.java
@@ -2609,6 +2609,7 @@ public boolean cancel(final boolean mayInterruptIfRunning) {
                     case ASF_ST_SUBMITTED: {
                         this.state = ASF_ST_CANCELLED;
                         notifyAll();
+                        doCancel();
                         return true;
                     }
                     case ASF_ST_RUNNING: {
@@ -2627,6 +2628,9 @@ public boolean cancel(final boolean mayInterruptIfRunning) {
             }
         }
 
+        void doCancel() {
+        }
+
         public V get() throws InterruptedException, ExecutionException {
             int state;
             synchronized (this) {
@@ -2864,33 +2868,49 @@ static int wrongType() throws ClassCastException {
     }
 
     final class RunnableScheduledFuture extends AbstractScheduledFuture<Void> {
-        final Runnable runnable;
+        Runnable runnable;
 
         RunnableScheduledFuture(final Runnable runnable, final long delay, final TimeUnit unit) {
             super(delay, unit);
             this.runnable = runnable;
         }
 
         Void performTask() {
-            runnable.run();
+            try {
+                runnable.run();
+            } finally {
+                runnable = null;
+            }
             return null;
         }
 
+        void doCancel() {
+            runnable = null;
+        }
+
         StringBuilder toString(final StringBuilder b) {
             return super.toString(b).append(runnable);
         }
     }
 
     final class CallableScheduledFuture<V> extends AbstractScheduledFuture<V> {
-        final Callable<V> callable;
+        Callable<V> callable;
 
         CallableScheduledFuture(final Callable<V> callable, final long delay, final TimeUnit unit) {
             super(delay, unit);
             this.callable = callable;
         }
 
         V performTask() throws Exception {
-            return callable.call();
+            try {
+                return callable.call();
+            } finally {
+                callable = null;
+            }
+        }
+
+        void doCancel() {
+            callable = null;
         }
 
         StringBuilder toString(final StringBuilder b) {
@@ -2940,7 +2960,7 @@ StringBuilder toString(final StringBuilder b) {
     }
 
     final class FixedRateRunnableScheduledFuture extends RepeatingScheduledFuture<Void> {
-        final Runnable runnable;
+        Runnable runnable;
 
         FixedRateRunnableScheduledFuture(final Runnable runnable, final long delay, final long period, final TimeUnit unit) {
             super(delay, period, unit);
@@ -2957,13 +2977,17 @@ Void performTask() {
             return null;
         }
 
+        void doCancel() {
+            runnable = null;
+        }
+
         StringBuilder toString(final StringBuilder b) {
             return super.toString(b).append(runnable);
         }
     }
 
     final class FixedDelayRunnableScheduledFuture extends RepeatingScheduledFuture<Void> {
-        final Runnable runnable;
+        Runnable runnable;
 
         FixedDelayRunnableScheduledFuture(final Runnable runnable, final long delay, final long period, final TimeUnit unit) {
             super(delay, period, unit);
@@ -2979,6 +3003,10 @@ Void performTask() {
             return null;
         }
 
+        void doCancel() {
+            runnable = null;
+        }
+
         StringBuilder toString(final StringBuilder b) {
             return super.toString(b).append(runnable);
         }