java - 使用 Java 8 Stream 从根目录递归添加所有文件

Question

我有以下递归方法，它只是将给定文件夹中的所有子项添加到列表中:

private List<TemplateFile> readTemplateFiles(String nextTemplateDir, String rootTemplateDir) throws FileNotFoundException {
    List<TemplateFile> templateFiles = new ArrayList<>();

    for (File file : new File(nextTemplateDir).listFiles()) {
        if (!file.isDirectory() && !file.getName().startsWith(".")) {
            templateFiles.add(TemplateFile.create(file, rootTemplateDir));
        } else if (file.isDirectory()) {
            templateFiles.addAll(readTemplateFiles(file.getAbsolutePath(), rootTemplateDir));
        }
    }

    return templateFiles;
}

如何使用新的 Java 8 Stream API 重构此方法？

Answer 1

你可以使用 Files.walk(start, options...) 递归遍历文件树。此方法返回 Stream<Path>由所有 Path 组成从给定的根开始。

Return a Stream that is lazily populated with Path by walking the file tree rooted at a given starting file. The file tree is traversed depth-first, the elements in the stream are Path objects that are obtained as if by resolving the relative path against start.

private List<TemplateFile> readTemplateFiles(String nextTemplateDir, String rootTemplateDir) throws FileNotFoundException {
    return Files.walk(Paths.get(nextTemplateDir))
                .filter(path -> !path.getFileName().startsWith("."))
                .map(path -> TemplateFile.create(path.toFile(), rootTemplateDir))
                .collect(Collectors.toList());
}

在 options 当中, 有 FOLLOW_LINKS 这将遵循符号链接(symbolic link)。