在日常的软件开发中，将文本文件的内容按行读取到内存中的字符串切片（或数组），以及将内存中的字符串切片内容写入到文本文件，是一项非常常见的需求。这不仅在处理配置、日志或小型数据集时尤为方便，也避免了初期引入复杂数据库的必要。go语言标准库提供了强大且高效的工具来满足这一需求，尤其是bufio包，它通过缓冲机制显著提升了文件i/o的性能。
读取文本文件到字符串切片
Go语言的bufio.Scanner是处理按行读取文本文件的理想选择。它提供了一个简单而高效的接口来迭代文件中的每一行，而无需一次性将整个文件加载到内存中（尽管在此示例中我们最终将其全部收集到切片中）。
以下是实现readLines函数的详细步骤：
- 打开文件： 使用os.Open(path)函数打开指定路径的文件。如果文件不存在或权限不足，os.Open会返回一个错误。
- 延迟关闭： 使用defer file.Close()确保文件在函数返回前被关闭，这是一种良好的资源管理实践，可以防止文件句柄泄露。
- 创建扫描器： 通过bufio.NewScanner(file)创建一个新的Scanner实例，它将与打开的文件关联。
- 逐行扫描： 使用for scanner.Scan()循环迭代文件的每一行。scanner.Scan()方法会读取下一行并返回一个布尔值，指示是否成功读取（直到文件末尾或遇到错误）。
- 获取行内容： 在每次成功扫描后，通过scanner.Text()方法获取当前行的字符串内容。
- 追加到切片： 将获取到的行内容追加到预先声明的lines字符串切片中。
- 检查错误： 循环结束后，务必调用scanner.Err()来检查在扫描过程中是否发生了任何错误。
package main
import (
"bufio"
"fmt"
"log"
"os"
)
// readLines reads a whole file into memory
// and returns a slice of its lines.
func readLines(path string) ([]string, error) {
file, err := os.Open(path)
if err != nil {
return nil, err
}
defer file.Close() // 确保文件在函数返回时关闭
var lines []string
scanner := bufio.NewScanner(file)
for scanner.Scan() {
lines = append(lines, scanner.Text())
}
return lines, scanner.Err() // 返回扫描过程中可能发生的错误
}将字符串切片写入文本文件
将字符串切片的内容写入到文本文件同样高效，我们可以使用bufio.NewWriter配合fmt.Fprintln来实现。bufio.Writer会缓冲写入操作，直到缓冲区满或显式调用Flush，这大大减少了系统调用的次数，从而提高了写入性能。
以下是实现writeLines函数的详细步骤：
立即学习“go语言免费学习笔记（深入）”；
- 创建文件： 使用os.Create(path)函数创建或截断（如果文件已存在）指定路径的文件。如果创建失败，会返回错误。
- 延迟关闭： 同样，使用defer file.Close()确保文件在函数返回前被关闭。
- 创建写入器： 通过bufio.NewWriter(file)创建一个新的Writer实例，它将与创建的文件关联。
- 逐行写入： 遍历字符串切片中的每一行。
- 写入内容： 使用fmt.Fprintln(w, line)将当前行内容写入到bufio.Writer中。Fprintln会自动在每行末尾添加换行符。
- 刷新缓冲区： 在所有内容写入完毕后，务必调用w.Flush()。这将把缓冲区中所有剩余的数据写入到实际的文件中。如果忘记调用Flush()，部分或全部数据可能不会被写入文件。
// writeLines writes the lines to the given file.
func writeLines(lines []string, path string) error {
file, err := os.Create(path)
if err != nil {
return err
}
defer file.Close() // 确保文件在函数返回时关闭
w := bufio.NewWriter(file)
for _, line := range lines {
// Fprintln 会在每行末尾添加换行符
if _, err := fmt.Fprintln(w, line); err != nil {
return err // 检查写入错误
}
}
return w.Flush() // 刷新缓冲区，确保所有数据写入文件
}完整示例
为了演示上述函数的用法，以下是一个完整的main函数示例，它首先从一个输入文件foo.in.txt读取内容，然后将读取到的内容打印到控制台，最后将这些内容写入到另一个输出文件foo.out.txt中。
func main() {
// 假设存在一个 foo.in.txt 文件，内容如下：
// Hello Go!
// This is a test file.
// Line 3.
lines, err := readLines("foo.in.txt")
if err != nil {
log.Fatalf("readLines: %s", err) // 如果读取失败，终止程序
}
fmt.Println("--- Content from foo.in.txt ---")
for i, line := range lines {
fmt.Printf("%d: %s\n", i, line)
}
fmt.Println("---------------------------------")
// 将读取到的内容写入到 foo.out.txt
if err := writeLines(lines, "foo.out.txt"); err != nil {
log.Fatalf("writeLines: %s", err) // 如果写入失败，终止程序
}
fmt.Println("Content successfully written to foo.out.txt")
// 可以再次读取 foo.out.txt 验证写入结果
outputLines, err := readLines("foo.out.txt")
if err != nil {
log.Fatalf("readLines from output: %s", err)
}
fmt.Println("\n--- Content from foo.out.txt (verification) ---")
for i, line := range outputLines {
fmt.Printf("%d: %s\n", i, line)
}
fmt.Println("-----------------------------------------------")
}在运行上述代码之前，请确保在同一目录下创建一个名为foo.in.txt的文本文件，并填充一些内容，例如：
Hello Go! This is a test file. Line 3.
运行程序后，将会在同一目录下生成一个foo.out.txt文件，其内容与foo.in.txt相同。
注意事项
- 错误处理： 在所有的文件I/O操作中，错误处理是至关重要的。本教程中的示例都包含了对os.Open、os.Create、scanner.Err()、fmt.Fprintln和w.Flush()返回的错误进行检查，并及时返回或处理。在实际应用中，应根据具体需求进行更细致的错误日志记录或用户提示。
- 资源管理： defer file.Close()是Go语言中管理文件句柄（以及其他需要关闭的资源）的惯用方式。它确保即使在函数执行过程中发生错误，文件也能被正确关闭，避免资源泄露。
- 性能考量： bufio包的使用对于大文件的读写性能提升显著，因为它减少了底层系统调用的次数。对于极小的文件，直接使用ioutil.ReadFile和ioutil.WriteFile可能更简洁，但它们会将整个文件一次性读入或写入，不适合超大文件。
- 内存使用： readLines函数会将整个文件内容加载到内存中的[]string切片。对于非常大的文件（例如几GB），这可能会导致内存耗尽（OOM）。在这种情况下，更好的做法是逐行处理文件内容，而不是一次性全部加载。
- 文件路径： 示例中使用的是相对路径。在生产环境中，应考虑使用绝对路径或通过配置管理文件路径，以确保程序的健壮性。
总结
通过本教程，我们学习了如何利用Go语言的bufio包高效且健壮地实现文本文件与字符串切片之间的相互转换。bufio.Scanner和bufio.Writer是处理此类I/O任务的强大工具，它们提供了良好的性能和易用性。掌握这些基本的文件操作技能，将为Go语言开发者处理各种文本数据奠定坚实的基础。
