Scala for 表達式

for 表達式

Scala 中的 for 表達式有如一把完成迭代任務的瑞士軍刀，它允許你使用一些簡單的部件以不同的方法組合可以完成許多復雜的迭代任務。簡單的應用比如枚舉一個整數(shù)列表，較復雜的應用可以同時枚舉多個不同類型的列表，根據(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<- 1 to 5 ){ 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]。