阶流处理递归文件和子目录

Question

我想申请一个功能在目录和子目录中的每个文件，内容如下：

def applyRecursively(dir: String, fn: (File) => Any) { 
    def listAndProcess(dir: File) { 
     dir.listFiles match { 
     case null => out.println("exception: dir cannot be listed: " + dir.getPath); List[File]() 
     case files => files.toList.sortBy(_.getName).foreach(file => { 
      fn(file) 
      if (file.isDirectory) listAndProcess(file) 
     }) 
     } 
    } 
    listAndProcess(new File(dir)) 
    } 

    def exampleFn(file: File) { println(s"processing $file") } 

    applyRecursively(dir, exampleFn) 

这个工程。以前他们answered如何通过使用scala迭代器来重构此代码。这里的问题是我如何通过使用scala Streams来重构此代码。 是这样的：

val stream: Stream[File] = ... // ??? 
stream.foreach(exampleFn) 

Answer 1

在这里你去：

def fileStream(dir: File): Stream[File] = 
    Option(dir.listFiles).map(_.toList.sortBy(_.getName).toStream).map { 
    files => files.append(files.filter(_.isDirectory).flatMap(fileStream)) 
    } getOrElse { 
    println("exception: dir cannot be listed: " + dir.getPath) 
    Stream.empty 
    } 

更新：以上回报文件和目录执行。下面是如何有效地排除目录：

def fileStreamNoDirs(dir: File): Stream[File] = 
    Option(dir.listFiles).map(_.toList.sortBy(_.getName).toStream.partition(_.isDirectory)) 
    .map { case (dirs, files) => 
     files.append(dirs.flatMap(fileStreamNoDirs)) 
    } getOrElse { 
    println("exception: dir cannot be listed: " + dir.getPath) 
    Stream.empty 
    } 

然而，简单fileStream("/some/path").filterNot(_.isDirectory)会做同样的工作。

UPDATE2：这种变体保留了原来的例子的遍历顺序：

def fileStream(dir: File): Stream[File] = 
    if (dir.isDirectory) 
    Option(dir.listFiles) 
     .map(_.toList.sortBy(_.getName).toStream.flatMap(file => file #:: fileStream(file))) 
     .getOrElse { 
     println("exception: dir cannot be listed: " + dir.getPath) 
     Stream.empty 
     } 
    else Stream.empty