使用 Go 语言读取文本文件数据

本文介绍了如何使用 Go 语言读取特定格式的文本文件。该文件包含头部信息、记录列表和整数值列表三个部分。我们将通过 `bufio` 包提供的 `Reader`，结合 `fmt.Fscanf` 和 `ReadString` 方法，实现对文件中数据的解析和读取，并展示了完整代码示例。

概述

在 Go 语言中，读取文本文件是一项常见的任务。bufio 包提供了缓冲 I/O 操作，可以高效地读取文件内容。本文将重点介绍如何使用 bufio.Reader 配合 fmt.Fscanf 和 ReadString 方法，读取包含特定格式数据的文本文件，并提供完整的代码示例。

文本文件格式说明

假设我们要读取的文本文件包含以下三个部分：

1. 头部 (Header): 第一行包含两个值：一个 uint64 类型的整数和一个 uint16 类型的整数。
2. 记录列表 (Records): 包含多行记录，每行记录包含一个 uint64 类型的整数，一个 uint16 类型的整数，一个字符 (X 或 -)，以及一个 UTF-8 编码的字符串，直到行尾。
3. 整数值列表 (Integer Values): 包含多行 uint64 类型的整数值。

例如：

小绿鲸英文文献阅读器

英文文献阅读器，专注提高SCI阅读效率

40

查看详情

5 4
1 2 - Yogurt
2 0 X Chicken soup
3 1 X Cheese
4 3 X Ham
2
3
4
0

代码实现

以下代码展示了如何使用 bufio.Reader 读取上述格式的文本文件：

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

func loadFile(fileName string) {
    // 打开文件并实例化一个 reader
    file, err := os.Open(fileName)
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close() // 确保文件在使用后关闭

    reader := bufio.NewReader(file)

    var (
        value0  uint64
        nbrRows uint16
    )

    // 读取头部信息
    if _, err := fmt.Fscanf(reader, "%d %d\n", &value0, &nbrRows); err != nil {
        log.Fatal(err)
    }

    // 遍历记录列表
    for i := uint16(0); i < nbrRows; i++ {
        var (
            value1 uint64
            value2 uint16
            value3 string
            value4 string
        )

        // 读取前三个值
        if _, err := fmt.Fscanf(reader, "%d %d %s\n", &value1, &value2, &value3); err != nil {
            log.Fatal(err)
        }
        // 读取剩余的行
        if value4, err = reader.ReadString('\n'); err != nil {
            log.Fatal(err)
        }
        value4 = strings.Trim(value4, " \n") // 去除首尾空格和换行符

        // 显示解析后的数据
        fmt.Printf("%d %d %s '%s'\n", value1, value2, value3, value4)
    }

    // 遍历整数值列表
    for i := uint16(0); i < nbrRows; i++ {
        var value5 uint64

        // 读取整数值
        if _, err := fmt.Fscanf(reader, "%d\n", &value5); err != nil {
            log.Fatal(err)
        }

        // 显示解析后的数据
        fmt.Printf("%d\n", value5)
    }
}

func main() {
    loadFile("data.txt") // 替换为你的文件名
}

代码解释：

1. 打开文件： 使用 os.Open 函数打开指定文件。
2. 创建 bufio.Reader： 使用 bufio.NewReader 创建一个带缓冲的读取器，提高读取效率。
3. 读取头部： 使用 fmt.Fscanf 函数从 reader 中读取头部信息，并将值赋给 value0 和 nbrRows 变量。\n 用来匹配换行符。
4. 读取记录列表： 使用循环遍历 nbrRows 次，每次循环读取一行记录。
   • fmt.Fscanf 用于读取每行记录的前三个值 (value1, value2, value3)。
   • reader.ReadString('\n') 用于读取剩余的字符串，直到遇到换行符为止。
   • strings.Trim(value4, " \n") 用于去除读取到的字符串首尾的空格和换行符。
5. 读取整数值列表： 类似地，使用循环遍历 nbrRows 次，每次循环读取一个整数值。
6. 错误处理： 代码中使用了 log.Fatal(err) 来处理读取过程中可能发生的错误。这会在发生错误时打印错误信息并退出程序。实际应用中，可以根据需要进行更精细的错误处理。

注意事项

• 文件关闭： 务必在使用完文件后关闭文件，可以使用 defer file.Close() 语句确保文件在函数退出时被关闭。
• 错误处理： 在实际应用中，需要对读取过程中可能出现的错误进行更完善的处理，例如文件不存在、文件格式错误等。
• 字符串处理： strings.Trim 函数用于去除字符串首尾的空格和换行符，可以根据实际情况调整需要去除的字符。
• 文件路径： 确保 loadFile 函数中的文件名参数 ("data.txt") 指向正确的文件路径。
• 数据类型： 确保代码中变量的数据类型与文件中实际的数据类型一致，否则可能导致读取错误。
• 假设条件： 此代码假设 value4 字符串不以空格或换行符开头或结尾。如果不是这种情况，可能需要调整 strings.Trim() 的用法。

总结

本文提供了一个使用 Go 语言读取特定格式文本文件的完整示例。通过 bufio.Reader 配合 fmt.Fscanf 和 ReadString 方法，我们可以灵活地解析和读取文件中的数据。在实际应用中，可以根据具体的文件格式和需求进行适当的调整和扩展。 掌握这些技巧，可以更有效地处理各种文本文件读取任务。

以上就是使用 Go 语言读取文本文件数据的详细内容，更多请关注php中文网其它相关文章！