使用 bufio.Scanner 读取整数文件
bufio.Scanner 是 Go 标准库中用于读取文本的强大工具。它可以按行、按单词或使用自定义分割函数来扫描输入。 在我们的例子中，我们将使用它按单词扫描，并将每个单词转换为整数。
以下是一个使用 bufio.Scanner 读取整数文件的示例代码：
package main
import (
"bufio"
"fmt"
"io"
"strconv"
"strings"
)
// ReadInts 从 r 中读取以空格分隔的整数。如果发生错误，它将返回到目前为止成功读取的整数以及错误值。
func ReadInts(r io.Reader) ([]int, error) {
scanner := bufio.NewScanner(r)
scanner.Split(bufio.ScanWords)
var result []int
for scanner.Scan() {
x, err := strconv.Atoi(scanner.Text())
if err != nil {
return result, err
}
result = append(result, x)
}
return result, scanner.Err()
}
func main() {
tf := "1\n2\n3\n4\n5\n6"
ints, err := ReadInts(strings.NewReader(tf))
fmt.Println(ints, err)
}代码解析：
-
ReadInts(r io.Reader) ([]int, error) 函数:
- 接受一个 io.Reader 接口作为输入，这意味着它可以从任何实现了 io.Reader 接口的对象读取数据，例如文件、字符串等。
- 创建一个 bufio.Scanner 对象，并将其与 io.Reader 关联。
- 使用 scanner.Split(bufio.ScanWords) 设置扫描器按单词分割输入。
- 循环调用 scanner.Scan() 读取下一个单词。
- 使用 strconv.Atoi() 将读取的单词转换为整数。如果转换失败，则返回已读取的整数数组和错误信息。
- 将转换后的整数添加到结果数组中。
- 循环结束后，返回结果数组和扫描器的错误信息（如果存在）。
-
main() 函数:
- 创建一个包含整数的字符串 tf，模拟文件内容。
- 使用 strings.NewReader(tf) 将字符串转换为 io.Reader 对象。
- 调用 ReadInts() 函数读取整数数组和错误信息。
- 打印结果数组和错误信息。
优势与注意事项
- 简洁性： 使用 bufio.Scanner 可以简化错误处理，代码更加简洁易懂。
- 通用性： 接受 io.Reader 接口作为输入，使其可以处理任何实现了 io.Reader 接口的对象。
- 错误处理： ReadInts 函数返回已读取的整数和遇到的错误，允许调用者处理部分成功读取的情况。
- 性能： 对于大型文件，bufio.Scanner 通常比 fmt.Fscanf 更有效率。
注意事项：
- 该代码假设输入文件中的整数以空格或换行符分隔。
- 如果输入文件中包含非整数内容，strconv.Atoi() 函数将返回错误。
总结
使用 bufio.Scanner 和 io.Reader 接口，可以更高效、更简洁地将包含整数的文件读取到 Go 语言的整数数组中。这种方法不仅代码可读性更高，而且更加通用，可以处理各种输入源。通过合理地利用 Go 语言的标准库，可以编写出更健壮、更易于维护的代码。
