go 逐行读取文件
前言
文件 I/O,特别是对文件的读写是编程语言中重要的功能。通常,我们需要逐行读取文件。
GO 提供了 bufio
软件包,实现了有缓冲的 I/O。它包装一个 io.Reader
或 io.Writer
接口对象,创建另一个也实现了该接口,且同时还提供了缓冲和一些文本 I/O 的帮助函数的对象。
在读取文件之前,我们首先需要使用 os.Open()
函数将其打开,该函数返回指向文件的指针类型。代码段中显示的 D:\\go_work\\test.txt
文件需要已经存在在系统中(将路径放置到文件所在的位置)。
bufio.NewScanner(file)
函数创建并返回一个从 r
读取数据的 Scanner
类型,该类型中的函数支持读取文件,其中默认的分割函数是 ScanLines
。
要逐行读取文件,我们需要使用两种在新的 Scanner
的方法 Scan
,它会获取当前位置的 token(该 token 可以通过 Bytes 或 Text 方法获得,在本例中为新行),并让 Scanner
的扫描位置移动到下一个 token
,和 Text(或 Byte)读取调用 Scan
时生成的最新符记。
如果在读取文件时遇到任何错误,可以通过在新的 Scanner
上调用 Err()
方法来处理这些错误,该方法将返回 Scanner
遇到的第一个非文件结尾错误;除非是 io.EOF
,此时 Err
会返回 nil
。
Go 逐行读取文件的完整代码
package main
import (
"bufio"
"fmt"
"log"
"os"
)
func main() {
// open the file
file, err := os.Open("D:\\go_work\\test.txt")
// handle errors while opening
if err != nil {
log.Fatalf("Error when opening file: %s", err)
}
defer file.Close()
fileScanner := bufio.NewScanner(file)
// read line by line
for fileScanner.Scan() {
fmt.Println(fileScanner.Text())
}
// handle first encountered error while reading
if err := fileScanner.Err(); err != nil {
log.Fatalf("Error while reading file: %s", err)
}
}
配置 Scanner 行为
Scanner
类型具有 Split
函数,该函数接受 SplitFunc
函数来确定 Scanner
如何拆分给定的字节片。默认的 SplitFunc
是 ScanLines
,它将返回文本的每一行,并删除行尾标记。
例如,我们可以使用单词进行拆分,如下面的代码片段所示:
scanner.Split(bufio.ScanWords) //configure how the scanner behaves