added in code to check for task being cancelled

bbejeck · Feb 13, 2012 · 68ad482 · 68ad482
1 parent 051eeed
commit 68ad482
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/java/bbejeck/nio/files/directory/FileDirectoryStream.java b/src/main/java/bbejeck/nio/files/directory/FileDirectoryStream.java
@@ -77,10 +77,12 @@ public Void call() throws Exception {
     private void findFiles(final File startDirectory, final FilenameFilter filenameFilter) {
         File[] files = startDirectory.listFiles(filenameFilter);
         for (File file : files) {
-            if (file.isDirectory()) {
-                findFiles(file, filenameFilter);
+            if (!fileTask.isCancelled()) {
+                if (file.isDirectory()) {
+                    findFiles(file, filenameFilter);
+                }
+                fileLinkedBlockingQueue.offer(file);
             }
-            fileLinkedBlockingQueue.offer(file);
         }
     }