Scala – 迭代器在目录中的文件中的所有行

import java.io.File
import scala.io.Source

for {
  file <- new File(dir).listFiles.toIterator if file.isFile
  line <- Source fromFile file getLines
} { doSomething line }

诀窍是flatMap和its for-comprehension syntactic sugar.例如,上面的或多或少等同于以下内容：

new File(dir)
  .listFiles.toIterator
  .filter(_.isFile)
  .flatMap(Source fromFile _ getLines)
  .map(doSomething)

正如Daniel Sobral在下面的评论中指出的,这种方法(和你的问题中的代码)将使文件保持打开状态.如果这是一次性脚本,或者您只是在REPL中工作,这可能不是一件大事.如果遇到问题,可以使用pimp-my-library pattern实现一些基本的资源管理：

implicit def toClosingSource(source: Source) = new {
  val lines = source.getLines
  var stillOpen = true
  def getLinesAndClose = new Iterator[String] {
    def hasNext = stillOpen && lines.hasNext
    def next = {
      val line = lines.next
      if (!lines.hasNext) { source.close() ; stillOpen = false }
      line
    }
  }
}

现在只需使用Source fromFile文件getLinesAndClose,您就不必担心文件被打开.