使用Java 8流仅列出特定深度的文件夹

Posted 2021-05-04

考虑到Java 8中存在“新”streams API，我可以使用Files.walk迭代文件夹。如果使用此方法或者深度= 2，我怎么才能获取给定目录的子文件夹？

我目前有这个工作示例，遗憾的是还将根路径打印为所有“子文件夹”。

Files.walk(Paths.get("/path/to/stuff/"))
     .forEach(f -> {
        if (Files.isDirectory(f)) {
            System.out.println(f.getName());
        }
     });

因此，我回到了以下approach。它将文件夹存储在内存中，然后需要处理存储的列表，我会避免使用lambdas。

File[] directories = new File("/your/path/").listFiles(File::isDirectory);

答案

仅列出给定目录的子目录：

Path dir = Paths.get("/path/to/stuff/");
Files.walk(dir, 1)
     .filter(p -> Files.isDirectory(p) && ! p.equals(dir))
     .forEach(p -> System.out.println(p.getFileName()));

另一答案

同意Andreas的回答，你也可以使用Files.list而不是Files.walk

Files.list(Paths.get("/path/to/stuff/"))
.filter(p -> Files.isDirectory(p) && ! p.equals(dir))
.forEach(p -> System.out.println(p.getFileName()));

另一答案

您可以使用Files#walk方法的第二个重载来显式设置最大深度。跳过流的第一个元素以忽略根路径，然后您只能过滤目录以最终打印每个元素。

final Path root = Paths.get("<your root path here>");

final int maxDepth = <your max depth here>;

Files.walk(root, maxDepth)
    .skip(1)
    .filter(Files::isDirectory)
    .map(Path::getFileName)
    .forEach(System.out::println);

另一答案

这是一个解决方案，适用于任意minDepth和maxDepth也大于1.假设minDepth >= 0和minDepth <= maxDepth：

final int minDepth = 2;
final int maxDepth = 3;
final Path rootPath = Paths.get("/path/to/stuff/");
final int rootPathDepth = rootPath.getNameCount();
Files.walk(rootPath, maxDepth)
        .filter(e -> e.toFile().isDirectory())
        .filter(e -> e.getNameCount() - rootPathDepth >= minDepth)
        .forEach(System.out::println);

为了完成你在列出“...只有一定深度的文件夹......”的问题中最初的问题，请确保minDepth == maxDepth。

另一答案

你也可以尝试这个：

private File getSubdirectory(File file){
    try {
        return new File(file.getAbsolutePath().substring(file.getParent().length()));
    }catch (Exception ex){

    }
    return null;
}

收集文件：

File[] directories = Arrays.stream(new File("/path/to/stuff")
          .listFiles(File::isDirectory)).map(Main::getSubdirectory)
                                        .toArray(File[]::new);