Fix ThreadNexter shutdown killing nexter thread already executing a different task

jmalves · headius · commit 3011d3d89fdd · 2017-12-11T21:08:48.000-06:00
Signed-off-by: Charles Oliver Nutter &lt;headius@headius.com&gt;
diff --git a/core/src/main/java/org/jruby/RubyEnumerator.java b/core/src/main/java/org/jruby/RubyEnumerator.java
@@ -745,20 +745,28 @@ public synchronized void shutdown() {
 
             // mark for death
             die = true;
+            if (dissociateNexterThread(true)) doneObject = null;
+        }
+
+        private synchronized boolean dissociateNexterThread(boolean interrupt) {
+            Thread nexterThread = thread;
 
-            Thread myThread = thread;
-            if (myThread != null) {
-                if (DEBUG) System.out.println("clearing for shutdown");
+            if (nexterThread != null) {
+                if (DEBUG) System.out.println("dissociating nexter thread, interrupt: " + interrupt);
 
-                // we interrupt twice, to break out of iteration and
-                // (potentially) break out of final exchange
-                myThread.interrupt();
-                myThread.interrupt();
+                if (interrupt) {
+                    // we interrupt twice, to break out of iteration and
+                    // (potentially) break out of final exchange
+                    nexterThread.interrupt();
+                    nexterThread.interrupt();
+                }
 
                 // release references
                 thread = null;
-                doneObject = null;
+                return true;
             }
+
+            return false;
         }
 
         @Override
@@ -910,9 +918,8 @@ public IRubyObject call(ThreadContext context, IRubyObject[] args, Block block)
                     if (!die) out.put(finalObject);
                 }
                 catch (InterruptedException ie) { /* ignore */ }
-            }
-            finally {
-                thread = null; // disassociate this Nexter with the thread running it
+            } finally {
+                dissociateNexterThread(false); // disassociate this Nexter with the thread running it
             }
         }
     }
