学习Scala脚本:从文件里读取行记录
处理琐碎的,每日工作的脚本经常需要处理文件。本节中,你将建立一个从文件中读入行记录,并把行中字符个数前置到每一行,打印输出的脚本。第一版展示在代码3.10中:
51CTO编辑推荐:Scala编程语言专题
import scala.io.Source if (args.length > 0) { for (line <- Source.fromFile(args(0)).getLines) print(line.length + " " + line) } else Console.err.println("Please enter filename")
代码 3.10 从文件中读入行
此脚本开始于从包scala.io引用名为Source的类。然后检查是否命令行里定义了至少一个参数。若是,则第一个参数被解释为要打开和处理的文件名。表达式Source.fromFile(args(0)),尝试打开指定的文件并返回一个Source对象,你在其上调用getLines。函数返回Iterator[String],在每个枚举里提供一行包括行结束符的信息。for表达式枚举这些行并打印每行的长度,空格和这行记录。如果命令行里没有提供参数,最后的else子句将在标准错误流中打印一条信息。如果你把这些代码放在文件contchars1.scala,并运行它调用自己:
$ scala countchars1.scala countchars1.scala
你会看到:
23 import scala.io.Source 1 23 if (args.length > 0) { 1 50 for (line <- Source.fromFile(args(0)).getLines) 36 print(line.length + " " + line) 2 } 5 else 47 Console.err.println("Please enter filename")
尽管当前形式的脚本打印出了所需的信息,你或许希望能让数字右序排列,并加上管道符号,这样输出看上去就替换成:
23 | import scala.io.Source 1 | 23 | if (args.length > 0) { 1 | 50 | for (line <- Source.fromFile(args(0)).getLines) 36 | print(line.length + " " + line) 2 | } 5 | else 47 | Console.err.println("Please enter filename")
想要达到这一点,你可以对所有行枚举两次。第一次决定每行字符计数的最大宽度。第二次打印输出之前计算的最大宽度。因为要枚举两次,你最好把它们赋给变量:
val lines = Source.fromFile(args(0)).getLines.toList
最后的toList是必须加的,因为getLines方法返回的是枚举器。一旦你使用它完成遍历,枚举器就失效了。而通过调用toList把它转换为List,你就可以枚举任意次数,代价就是把文件中的所有行一次性贮存在内存里。lines变量因此就指向着包含了命令行指定的文件文本字串的数组。
下一步,因为要对每行字符数计算两次,每个枚举计算一次,你或许会考虑把表达式拉出来变成一个小函数,专门用来计算传入字串的字符长度:
def widthOfLength(s: String) = s.length.toString.length
有了这个函数,你就可以计算最大长度了:
var maxWidth = 0 for (line <- lines) maxWidthmaxWidth = maxWidth.max(widthOfLength(line))
这里你用一个for表达式枚举了每一行,计算这些行的宽度,并且,如果比当前最大宽度还大,就把它赋值给maxWidth,一个初始化为0的var。(max方法是你可以在任何Int上调用的,可以返回被调用者和被传入者中的较大的值。)如果你希望不用var发现最大值,替代的方法是可以首先找到最长的一行,如:
val longestLine = lines.reduceLeft( (a, b) => if (a.length > b.length) a else b ) val widths = lines.map(widthOfLength)
reduceLeft方法把传入的方法应用于lines的前两个元素,然后再应用于第一次应用的结果和lines接下去的一个元素,等等,直至整个列表。每次这样的应用,结果将是碰到的最长一行,因为传入的函数,(a, b) => if (a.length > b.length) a else b,返回两个传入字串的最长那个。reduceLeft将传回最后一次应用的结果,也就是本例lines中包含的最长字串。
得到这个结果之后,你可以通过把最长一行传给widthOfLength计算最大的宽度:
val maxWidth = widthOfLength(longestLine)
最后剩下的就是用一个合适的格式把这些行打印出来。你可以这么做:
for (line <- lines) { val numSpaces = maxWidth - widthOfLength(line) val padding = " " * numSpaces print(padding + line.length + " | " + line) }
在这个for表达式里,你再一次枚举了全部行记录。对于每一行,首先计算行长度前所需的空格并把它赋给numSpaces。然后用表达式:" " * numSpaces创建包含numSpaces个空格的字串。最终,你打印出你想要格式的信息。全部的脚本展示在代码3.11中:
import scala.io.Source def widthOfLength(s: String) = s.length.toString.length if (args.length > 0) { val lines = Source.fromFile(args(0)).getLines.toList val longestLine = lines.reduceLeft( (a, b) => if (a.length > b.length) a else b ) val maxWidth = widthOfLength(longestLine) for (line <- lines) { val numSpaces = maxWidth widthOfLength(line) val padding = " " * numSpaces print(padding + line.length +" | "+ line) } } else Console.err.println("Please enter filename")
代码 3.11 对文件的每行记录打印格式化的字符数量。