Scala 中的 for 表達式有如一把完成迭代任務(wù)的瑞士軍刀，它允許你使用一些簡單的部件以不同的方法組合可以完成許多復(fù)雜的迭代任務(wù)。簡單的應(yīng)用比如枚舉一個整數(shù)列表，較復(fù)雜的應(yīng)用可以同時枚舉多個不同類型的列表，根據(jù)條件過濾元素，并可以生成新的集合。

枚舉集合元素

這是使用 for 表示式的一個基本用法，和 Java 的 for 非常類似，比如下面的代碼可以枚舉當前目錄下所有文件：

val filesHere = (new java.io.File(".")).listFiles
for( file <-filesHere)
  println(file)

其中如 file < – filesHere 的語法結(jié)構(gòu)，在 Scala 中稱為“生成器 (generator)”。本例中 filesHere 的類型為 Array[File]，每次迭代變量 file 會初始化為該數(shù)組中一個元素， File 的 toString()為文件的文件名，因此 println(file)打印出文件名。 Scala 的 for 表達式支持所有類型的集合類型，而不僅僅是數(shù)組，比如下面使用 for 表達式來枚舉一個 Range 類型。

 scala> for ( i      | println ("Interation " + i)
Interation 1
Interation 2
Interation 3
Interation 4

過濾

某些時候，你不想枚舉集合中的每個元素，而是只迭代某些符合條件的元素，在 Scala 中，你可以為 for 表達式添加一個過濾器–在 for 的括號內(nèi)添加一個 if 語句，例如：修改前面枚舉文件的例子，改成只列出 .scala 文件如下：

val filesHere = (new java.io.File(".")).listFiles
for( file   println(file)

如果有必要的話，你可以使用多個過濾器，只要添加多個 if 語句即可，比如，為保證前面列出的文件不是目錄，可以添加一個 if，如下面代碼：

val filesHere = (new java.io.File(".")).listFiles
for( file <-filesHere
   if file.isFile
   if file.getName.endsWith(".scala")
)  println(file)

嵌套迭代

fo r表達式支持多重迭代，下面的例子使用兩重迭代，外面的循環(huán)枚舉 filesHere，而內(nèi)部循環(huán)枚舉該文件的每一行文字。實現(xiàn)了類似 Unix　grep 命令：

val filesHere = (new java.io.File(".")).listFiles
def fileLines (file : java.io.File) =
   scala.io.Source.fromFile(file).getLines().toList
def grep (pattern: String) =
  for (
    file     if file.getName.endsWith(".scala");
    line <-fileLines(file)
    if line.trim.matches(pattern)
  ) println(file + ":" + line.trim)
grep (".*gcd.*")

注意上面代碼中兩個迭代之間使用了”;”，如果你使用{} 替代 for 的()的括號，你可以不使用“；”分隔這兩個“生成器”，這是因為 Scala 編譯器不推算包含在括號內(nèi)的省掉的“;”。使用{}改寫的代碼如下：

val filesHere = (new java.io.File(".")).listFiles
def fileLines (file : java.io.File) =
   scala.io.Source.fromFile(file).getLines().toList
def grep (pattern: String) =
  for {
    file     if file.getName.endsWith(".scala")
    line <-fileLines(file)
    if line.trim.matches(pattern)
  } println(file + ":" + line.trim)
grep (".*gcd.*")

這兩段代碼是等效的。

綁定中間變量

你可以注意到前面代碼使用了多次 line.trim，如果 trim 是個耗時的操作，你可以希望 trim 只計算一次，Scala 允許你使用=號來綁定計算結(jié)果到一個新變量。綁定的作用和 val 類似，只是不需要使用 val 關(guān)鍵字，例如，修改前面的例子，只計算一次 trim，把結(jié)果保存在 trimmed 變量中。

val filesHere = (new java.io.File(".")).listFiles
def fileLines (file : java.io.File) =
   scala.io.Source.fromFile(file).getLines().toList
def grep (pattern: String) =
  for {
    file     if file.getName.endsWith(".scala")
    line <-fileLines(file)
    trimmed=line.trim
    if trimmed.matches(pattern)
  } println(file + ":" + trimmed)
grep (".*gcd.*")

生成新集合

for 表達式也可以用來生產(chǎn)新的集合，這是 Scala 的 for 表達式比 Java 的 for 語句功能強大的地方。它的基本語法如下：

for clauses yield body

關(guān)鍵字 yield 放在 body 的前面，for 沒迭代一次，產(chǎn)生一個 body，yield 收集所有的 body 結(jié)果，返回一個 body 類型的集合。比如，前面列出所有 .scala 文件，返回這些文件的集合：

def scalaFiles =
  for {
    file     if file.getName.endsWith(".scala")
  } yield file

scalaFiles 的類型為 Array[File]。

上一篇：組合和繼承–使用組合還是繼承？下一篇：類和對象 (三)