Merge pull request #9 from stevegury/master

FileSystemWatcher, when enabling "scan for current files" also report directories.
ReactiveX · Aug 26, 2015 · ba043ff · ba043ff
2 parents 0f325b0 + 5ec546d
commit ba043ff
Showing 1 changed file with 26 additions and 16 deletions.
diff --git a/src/main/java/rx/fileutils/FileSystemWatcher.java b/src/main/java/rx/fileutils/FileSystemWatcher.java
@@ -142,22 +142,13 @@ private List<WatchEvent<Path>> getEventsForCurrentFiles(Path directory) {
                 Files.walkFileTree(directory, new SimpleFileVisitor<Path>() {
                     @Override
                     public FileVisitResult visitFile(Path path, BasicFileAttributes attrs) {
-                        events.add(new WatchEvent<Path>() {
-                            @Override
-                            public Kind<Path> kind() {
-                                return ENTRY_CREATE;
-                            }
-
-                            @Override
-                            public int count() {
-                                return 1;
-                            }
-
-                            @Override
-                            public Path context() {
-                                return path;
-                            }
-                        });
+                        events.add(pathToWatchEvent(path));
+                        return FileVisitResult.CONTINUE;
+                    }
+
+                    @Override
+                    public FileVisitResult preVisitDirectory(Path path, BasicFileAttributes attrs) {
+                        events.add(pathToWatchEvent(path));
                         return FileVisitResult.CONTINUE;
                     }
                 });
@@ -167,6 +158,25 @@ public Path context() {
 
             return events;
         }
+
+        private WatchEvent<Path> pathToWatchEvent(Path path) {
+            return new WatchEvent<Path>() {
+                @Override
+                public Kind<Path> kind() {
+                    return ENTRY_CREATE;
+                }
+
+                @Override
+                public int count() {
+                    return 1;
+                }
+
+                @Override
+                public Path context() {
+                    return path;
+                }
+            };
+        }
     }
 
     public static class Builder {