递归流

Question

我想使用Java8递归列出我计算机上的所有文件。

Java8提供了一个listFiles方法，该方法返回所有的文件和目录，但没有递归。如何使用它来获得完整的文件递归列表(不使用变异的集合)？

我试过下面的代码，但它只深入一层：

static Function<Path, Stream<Path>> listFiles = p -> {
    if (p.toFile().isDirectory()) {
        try { return Files.list(p); }
        catch (Exception e) { return Stream.empty(); }
    } else {
        return Stream.of(p);
    }
};

public static void main(String[] args) throws IOException {
    Path root = Paths.get("C:/temp/");
    Files.list(root).flatMap(listFiles).forEach(System.out::println);
}

并且使用return Files.list(p).flatMap(listFiles);不能编译(不确定原因)...

注意:我对涉及FileVisitors或外部库的解决方案不感兴趣。

Answer 1

通过递归遍历文件系统来生成路径流的新API是Files.walk。

如果您确实希望以递归方式生成流(不一定遍历文件树，但我将继续使用它作为示例)，那么使用方法引用来完成递归可能会更简单一些：

class RecursiveStream {
    static Stream<Path> listFiles(Path path) {
        if (Files.isDirectory(path)) {
            try { return Files.list(path).flatMap(RecursiveStream::listFiles); }
            catch (Exception e) { return Stream.empty(); }
        } else {
            return Stream.of(path);
        }
    }

    public static void main(String[] args) {
        listFiles(Paths.get(".")).forEach(System.out::println);
    }
}

事实证明，方法引用对于将具有与函数接口相同的“形状”(参数和返回类型)的命名方法适配到该函数接口非常有用。这也避免了在实例或静态变量中存储lambda并递归调用自身的潜在初始化循环。