Go语言教程：高效且地道地从文件读取整数到切片

package main

import (
    "fmt"
    "io"
    "os"
)

func readFile(filePath string) (numbers []int) {
    fd, err := os.Open(filePath)
    if err != nil {
        panic(fmt.Sprintf("open %s: %v", filePath, err)) // 直接panic，不够优雅
    }
    defer fd.Close() // 确保文件关闭

    var line int
    for {
        _, err := fmt.Fscanf(fd, "%d\n", &line) // 每次读取一行并解析整数

        if err != nil {
            if err == io.EOF { // 文件结束
                return
            }
            panic(fmt.Sprintf("Scan Failed %s: %v", filePath, err)) // 再次panic
        }
        numbers = append(numbers, line)
    }
}

func main() {
    // 假设 numbers.txt 文件内容为：
    // 1
    // 2
    // 3
    // ...
    numbers := readFile("numbers.txt")
    fmt.Println("读取到的整数数量:", len(numbers))
    fmt.Println("部分整数:", numbers[:3])
}

package main

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

// ReadInts 从 io.Reader 中读取空格分隔的整数。
// 如果在读取过程中遇到错误（包括非整数格式），它会返回已成功读取的整数切片和相应的错误。
func ReadInts(r io.Reader) ([]int, error) {
    scanner := bufio.NewScanner(r) // 创建一个新的Scanner
    scanner.Split(bufio.ScanWords)  // 设置分词函数为按单词（whitespace-separated）分割

    var result []int
    for scanner.Scan() { // 遍历所有分词
        text := scanner.Text() // 获取当前分词的文本
        x, err := strconv.Atoi(text) // 尝试将文本转换为整数
        if err != nil {
            // 如果转换失败，返回已读取的整数和错误
            return result, fmt.Errorf("无法将 '%s' 转换为整数: %w", text, err)
        }
        result = append(result, x) // 将整数添加到结果切片
    }

    // 循环结束后，检查Scanner本身是否遇到错误（如底层I/O错误）
    if err := scanner.Err(); err != nil {
        return result, fmt.Errorf("Scanner 在读取过程中遇到错误: %w", err)
    }

    return result, nil // 成功读取所有整数
}

func main() {
    // 示例1: 从字符串读取
    dataString := "1\n2   3\n4 5\n6" // 包含换行符和空格分隔的整数
    reader := strings.NewReader(dataString)
    ints, err := ReadInts(reader)
    if err != nil {
        fmt.Printf("从字符串读取整数时发生错误: %v\n", err)
    } else {
        fmt.Printf("从字符串读取到的整数: %v\n", ints) // 输出: [1 2 3 4 5 6]
    }

    // 示例2: 从文件读取
    // 假设存在一个名为 "numbers.txt" 的文件，内容与 dataString 类似
    // 为了演示，这里先创建一个虚拟文件
    fileContent := "10\n20\n30 40\n50"
    err = os.WriteFile("numbers.txt", []byte(fileContent), 0644)
    if err != nil {
        fmt.Printf("创建测试文件失败: %v\n", err)
        return
    }
    defer os.Remove("numbers.txt") // 程序结束时清理文件

    file, err := os.Open("numbers.txt")
    if err != nil {
        fmt.Printf("打开文件失败: %v\n", err)
        return
    }
    defer file.Close() // 确保文件关闭

    intsFromFile, err := ReadInts(file)
    if err != nil {
        fmt.Printf("从文件读取整数时发生错误: %v\n", err)
    } else {
        fmt.Printf("从文件读取到的整数: %v\n", intsFromFile) // 输出: [10 20 30 40 50]
    }

    // 示例3: 包含非整数数据，触发错误
    invalidData := "1\n2\nhello\n4"
    invalidReader := strings.NewReader(invalidData)
    intsInvalid, err := ReadInts(invalidReader)
    if err != nil {
        fmt.Printf("从包含非整数数据源读取时发生错误: %v (已读取: %v)\n", err, intsInvalid)
        // 输出: 从包含非整数数据源读取时发生错误: 无法将 'hello' 转换为整数: strconv.Atoi: parsing "hello": invalid syntax (已读取: [1 2])
    }
}