Fix bug in FastFileInflator

Needed to set the cleanHookAtomic reference to the cleanHook thread.
aaiezza · May 3, 2016 · e13acc5 · e13acc5
1 parent 447fd69
commit e13acc5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/edu/rit/flick/genetics/FastFileInflator.java b/src/edu/rit/flick/genetics/FastFileInflator.java
@@ -263,6 +263,7 @@ public File inflate( final Configuration configuration, final File fileIn, final
 
             // Make cleaning hook
             final Thread cleanHook = new Thread( () -> {
+                interrupted = true;
                 try
                 {
                     // Clean up IO
@@ -279,6 +280,8 @@ public File inflate( final Configuration configuration, final File fileIn, final
                 }
             }, "Inflation_Cleaning_Thread" );
 
+            cleanHookAtomic.set( cleanHook );
+
             Runtime.getRuntime().addShutdownHook( cleanHook );
 
             inflateToDirectoryThread.start();