scala - 如何列出 Hdfs 目录和子目录中文件的路径？

Question

无法找到一种方法来列出目录和子目录中的所有文件。

这是我正在使用的代码，它列出了特定目录中的文件，但如果里面有子目录，则为文件:

val conf = new Configuration()
val fs = FileSystem.get(new java.net.URI("hdfs://servername/"), conf)
val status = fs.listStatus(new Path("path/to/folder/"))
status.foreach { x => println(x.getPath.toString()) }

上面的代码列出了目录中的所有文件，但我需要它是递归的。

Answer 1

只要发现新文件夹，就可以进行递归:

val hdfs = FileSystem.get(new Configuration())

def listFileNames(hdfsPath: String): List[String] = {

  hdfs
    .listStatus(new Path(hdfsPath))
    .flatMap { status =>
      // If it's a file:
      if (status.isFile)
        List(hdfsPath + "/" + status.getPath.getName)
      // If it's a dir and we're in a recursive option:
      else
        listFileNames(hdfsPath + "/" + status.getPath.getName)
    }
    .toList
    .sorted
}