BAEL-4755: Add Files.list example (eugenp#10356)

kwoyke · web-flow · commit ed330c5c0de4 · 2020-12-30T12:39:45.000+01:00
diff --git a/core-java-modules/core-java-io-2/src/main/java/com/baeldung/listfiles/ListFiles.java b/core-java-modules/core-java-io-2/src/main/java/com/baeldung/listfiles/ListFiles.java
@@ -24,9 +24,20 @@ public Set<String> listFilesUsingJavaIO(String dir) {
             .collect(Collectors.toSet());
     }
 
+    public Set<String> listFilesUsingFilesList(String dir) throws IOException {
+        try (Stream<Path> stream = Files.list(Paths.get(dir))) {
+            return stream
+                .filter(file -> !Files.isDirectory(file))
+                .map(Path::getFileName)
+                .map(Path::toString)
+                .collect(Collectors.toSet());
+        }
+    }
+
     public Set<String> listFilesUsingFileWalk(String dir, int depth) throws IOException {
         try (Stream<Path> stream = Files.walk(Paths.get(dir), depth)) {
-            return stream.filter(file -> !Files.isDirectory(file))
+            return stream
+                .filter(file -> !Files.isDirectory(file))
                 .map(Path::getFileName)
                 .map(Path::toString)
                 .collect(Collectors.toSet());
diff --git a/core-java-modules/core-java-io-2/src/test/java/com/baeldung/listfiles/ListFilesUnitTest.java b/core-java-modules/core-java-io-2/src/test/java/com/baeldung/listfiles/ListFilesUnitTest.java
@@ -25,10 +25,15 @@ public class ListFilesUnitTest {
     };
 
     @Test
-    public void givenDir_whenUsingJAVAIO_thenListAllFiles() throws IOException {
+    public void givenDir_whenUsingJAVAIO_thenListAllFiles() {
         assertEquals(EXPECTED_FILE_LIST, listFiles.listFilesUsingJavaIO(DIRECTORY));
     }
 
+    @Test
+    public void givenDir_whenUsingFilesList_thenListAllFiles() throws IOException {
+        assertEquals(EXPECTED_FILE_LIST, listFiles.listFilesUsingFilesList(DIRECTORY));
+    }
+
     @Test
     public void givenDir_whenWalkingTree_thenListAllFiles() throws IOException {
         assertEquals(EXPECTED_FILE_LIST, listFiles.listFilesUsingFileWalk(DIRECTORY,DEPTH));
