更优雅地将整数文件读取到 Go 数组中-Golang-PHP中文网

更优雅地将整数文件读取到 Go 数组中

聖光之護

发布： 2025-08-29 19:31:16

原创

749人浏览过

更优雅地将整数文件读取到 go 数组中

本文介绍了一种更简洁、更符合 Go 语言习惯的方式，将包含整数的文件读取到数组中。通过使用 bufio.Scanner 和 io.Reader 接口，可以简化代码并提高其灵活性，使其能够处理各种文件来源，而不仅仅是磁盘上的文件。

在 Go 语言中，读取文件内容并将其转换为数组是一项常见的任务。原始方法可能涉及大量的错误处理代码，使得代码显得冗长且难以阅读。本文将展示如何利用 bufio.Scanner 和 io.Reader 接口，以更简洁、更符合 Go 语言习惯的方式实现相同的功能。

使用 bufio.Scanner 和 io.Reader

bufio.Scanner 提供了一种方便的方式来逐行或逐字扫描输入流。 io.Reader 是一个接口，允许从各种来源读取数据，包括文件、网络连接和内存中的字符串。

以下是一个示例代码，展示了如何使用 bufio.Scanner 和 io.Reader 将整数文件读取到数组中：

package main

import (
    "bufio"
    "fmt"
    "io"
    "strconv"
    "strings"
)

// ReadInts reads whitespace-separated ints from r. If there's an error, it
// returns the ints successfully read so far as well as the error value.
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)
}

登录后复制

代码解释：

LibLib AI

中国领先原创AI模型分享社区，拥有LibLib等于拥有了超多模型的模型库、免费的在线生图工具，不考虑配置的模型训练工具

647

查看详情

ReadInts(r io.Reader) ([]int, error) 函数:
- 接受一个 io.Reader 接口作为输入，这使得该函数可以从任何实现了 io.Reader 接口的来源读取数据。
- 创建一个 bufio.Scanner 对象，并将其与 io.Reader 关联。
- 使用 scanner.Split(bufio.ScanWords) 将扫描器配置为按空格分隔单词。如果想按行读取，则使用 bufio.ScanLines。
- 循环扫描输入流，直到遇到错误或到达文件末尾。
- 在每次迭代中，使用 scanner.Text() 获取当前扫描到的文本，并使用 strconv.Atoi() 将其转换为整数。
- 如果转换失败，则返回已成功读取的整数数组和错误信息。
- 将转换后的整数添加到结果数组中。
- 最后，返回结果数组和扫描器可能遇到的任何错误（例如，文件末尾错误）。
main() 函数:
- 创建一个包含整数的字符串。
- 使用 strings.NewReader() 将字符串转换为 io.Reader。
- 调用 ReadInts() 函数读取字符串中的整数。
- 打印结果数组和错误信息。

优点：

简洁性： 使用 bufio.Scanner 简化了读取文件的代码，减少了错误处理的复杂性。
灵活性： io.Reader 接口允许从各种来源读取数据，而不仅仅是磁盘上的文件。例如，可以使用 strings.NewReader 从字符串读取数据，或者使用网络连接读取数据。
可读性： 代码结构清晰，易于理解和维护。