Go语言读写文本到字符串切片教程

在Go语言的日常开发中，对文本文件的读写操作是常见的需求，尤其是在处理配置文件、日志文件或小型数据集时。将文件内容读取为内存中的字符串切片，或将数据集合写入文件，是实现这些功能的基础。Go标准库提供了强大的工具来高效地完成这些任务，其中bufio包扮演了核心角色。

使用 bufio.Scanner 读取文件行

bufio.Scanner 是Go 1.1版本引入的一个API，它提供了一种简单而高效的方式来逐行（或其他分隔符）读取输入流。它非常适合处理大型文件，因为它不需要一次性将整个文件加载到内存中。

以下是一个 readLines 函数的实现，它能够打开指定路径的文件，并将其所有行读取到一个 []string 切片中：

package main

import (
    "bufio"
    "fmt"
    "log"
    "os"
)

// readLines 从指定路径的文件中读取所有行，并返回一个字符串切片。
// 如果发生错误，则返回nil和错误信息。
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) // 创建一个新的Scanner，从文件中读取
    for scanner.Scan() {           // 循环读取每一行
        lines = append(lines, scanner.Text()) // 将当前行文本添加到切片中
    }
    return lines, scanner.Err() // 返回读取到的行和Scanner在扫描过程中遇到的任何错误
}

关键点解析：

• os.Open(path): 用于打开文件。如果文件不存在或没有读取权限，将返回错误。
• defer file.Close(): 这是一个非常重要的语句。它确保了在 readLines 函数执行完毕（无论正常返回还是发生panic）之前，文件资源会被正确关闭，防止资源泄露。
• bufio.NewScanner(file): 创建一个 Scanner 实例，它会从 file 读取数据。默认情况下，Scanner 以行为单位进行扫描。
• scanner.Scan(): 尝试读取下一行。如果成功读取到数据，返回 true；如果到达文件末尾或发生错误，返回 false。
• scanner.Text(): 返回当前行作为字符串。
• scanner.Err(): 在 for scanner.Scan() 循环结束后，应该检查此方法以确定在扫描过程中是否发生了任何非EOF错误。

使用 bufio.NewWriter 写入文件行

与读取类似，将字符串切片写入文件也需要高效的机制。bufio.NewWriter 提供了一个带缓冲的写入器，可以显著提高写入性能，因为它会积累数据直到缓冲区满或显式调用 Flush()，从而减少底层系统调用次数。

以下是一个 writeLines 函数的实现，它能够将一个字符串切片的内容逐行写入到指定路径的文件中：

// writeLines 将字符串切片中的所有行写入到指定路径的文件。
// 如果文件创建或写入失败，则返回错误。
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 {
        // fmt.Fprintln 会将字符串写入到 w，并在末尾添加一个换行符
        _, err := fmt.Fprintln(w, line) 
        if err != nil {
            return err // 如果写入过程中发生错误，返回错误
        }
    }
    return w.Flush() // 将缓冲区中所有的数据写入到文件。这一步至关重要！
}

关键点解析：

• os.Create(path): 用于创建文件。如果文件已存在，Create 会打开它并将其内容截断为零长度，相当于清空文件。
• defer file.Close(): 同样重要，确保文件资源被释放。
• bufio.NewWriter(file): 创建一个带缓冲的写入器 w，它会将数据写入到 file。
• fmt.Fprintln(w, line): 将 line 字符串写入到 w（写入器），并在末尾自动添加一个换行符。
• w.Flush(): 这是最关键的一步！ bufio.NewWriter 会将数据暂存在内存缓冲区中，直到缓冲区满或调用 Flush()。如果没有调用 Flush()，即使写入操作成功，数据也可能不会被真正写入到磁盘文件中。因此，在所有数据写入完毕后，务必调用 Flush()。

完整示例与注意事项

将上述 readLines 和 writeLines 函数结合起来，可以在 main 函数中进行测试和演示：

func main() {
    // 1. 创建一个测试输入文件
    initialContent := []string{"Hello, Go!", "This is a test file.", "Line 3."}
    if err := writeLines(initialContent, "foo.in.txt"); err != nil {
        log.Fatalf("创建输入文件失败: %s", err)
    }
    fmt.Println("成功创建 'foo.in.txt'")

    // 2. 读取文件内容
    lines, err := readLines("foo.in.txt")
    if err != nil {
        log.Fatalf("读取文件失败: %s", err) // 使用log.Fatalf在发生致命错误时退出程序
    }
    fmt.Println("\n从 'foo.in.txt' 读取到的内容:")
    for i, line := range lines {
        fmt.Printf("%d: %s\n", i, line)
    }

    // 3. 修改内容（可选，此处直接使用读取到的内容）
    // lines = append(lines, "New line added.")

    // 4. 将内容写入新文件
    if err := writeLines(lines, "foo.out.txt"); err != nil {
        log.Fatalf("写入文件失败: %s", err)
    }
    fmt.Println("\n成功将内容写入 'foo.out.txt'")
}

注意事项：

1. 错误处理: 在Go语言中，错误处理是强制性的。每个可能返回错误的操作（如 os.Open, os.Create, scanner.Err, fmt.Fprintln, w.Flush）都必须检查其返回的 error 值。
2. 资源管理: 使用 defer file.Close() 确保文件句柄在函数返回时被关闭，避免资源泄露。
3. bufio 的重要性: 对于文本文件的逐行操作，bufio.Scanner 和 bufio.NewWriter 是首选，它们提供了高效的缓冲机制，显著优于不带缓冲的直接 os.Read 或 os.Write。
4. w.Flush(): 再次强调，写入操作完成后，必须调用 w.Flush() 将缓冲区中的数据真正写入到磁盘。
5. log.Fatalf: 在示例中，我们使用了 log.Fatalf 来处理致命错误。这会在打印错误信息后终止程序。在生产环境中，你可能需要更精细的错误处理逻辑，例如返回错误给调用者，或者进行重试等。

通过上述方法，你可以在Go语言中以专业且高效的方式处理文本文件的读写，满足大多数常见的应用场景。