说明:
①在for表达式中,“file <- files”叫做生成器(generator)
②嵌套循环可以写在一个for里
③可以为每个生成器添加if过滤器(filter)
④trimmed是一个中途绑定的变量
⑤for表达式可以有返回值,将yeild放在代码体之前。
代码
import java.io.File import scala.io.Source object Test { def main(args: Array[String]): Unit = { println(grep(".+").mkString(" ")) } def fileLines(file: File): List[String] = { Source.fromFile(file).getLines().toList } def grep(pattern: String): Array[String] = { val files: Array[File] = new File(".").listFiles() for { file <- files if file.getName.contains(".") if file.getName.endsWith(".xml") line <- fileLines(file) trimmed = line.trim if trimmed.matches(pattern) }yield file + ": " + trimmed } }
控制台输出
.pom.xml: <?xml version="1.0" encoding="UTF-8"?> .pom.xml: <project xmlns="http://maven.apache.org/POM/4.0.0" .pom.xml: xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" .pom.xml: xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> .pom.xml: <modelVersion>4.0.0</modelVersion> .pom.xml: <groupId>org.example</groupId> .pom.xml: <artifactId>scala-demo</artifactId> .pom.xml: <version>1.0-SNAPSHOT</version> .pom.xml: </project>