Go语言：从标准输入读取整数并处理格式错误

本文探讨了在Go语言中如何健壮地从标准输入（stdin）读取整数序列，同时优雅地处理文件结束符（EOF）和报告或跳过格式错误。通过区分`io.EOF`与其他输入错误，并采取相应的恢复策略（如读取并丢弃无效输入），我们可以构建一个能够持续处理有效数据并报告异常的强大输入解析器。

Go语言中从标准输入读取整数的挑战

在Go语言中，使用fmt.Scan或fmt.Fscan函数可以方便地从标准输入或其他io.Reader读取格式化的数据。然而，当输入流中包含非预期格式的数据（例如，期望整数却遇到字符串）时，fmt.Scan的行为可能不尽如人意。它会停止读取，并返回一个错误，但不会自动跳过无效的输入。如果仅仅检查io.EOF来判断输入结束，那么其他类型的格式错误将导致程序在不报告具体问题的情况下提前终止读取，或者只处理部分有效数据。

例如，以下代码片段展示了这种基本读取方式：

package main

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

func main() {
    nums := make([]int, 0)
    var d int

    fmt.Println("请输入整数，按Ctrl+D (Unix/Linux) 或 Ctrl+Z (Windows) 结束输入:")

    for {
        _, err := fmt.Scan(&d) // 从os.Stdin读取整数
        if err != nil {
            if err == io.EOF {
                break // 遇到EOF，正常退出循环
            }
            // 其他错误，例如格式错误，会在此处被捕获
            log.Fatal(err) // 默认行为：遇到非EOF错误时终止程序
        }
        nums = append(nums, d)
    }
    fmt.Printf("读取到的整数: %v\n", nums)
}

当输入为 1 2 3 f4 5 时，上述代码在遇到 f4 时会因格式错误而调用 log.Fatal 终止程序。如果我们将 log.Fatal(err) 替换为 break，那么程序将只读取 [1 2 3]，并忽略 f4 5，且不报告任何错误，这在许多场景下并非理想行为。

立即学习“go语言免费学习笔记（深入）”；

沁言学术

你的论文写作AI助理，永久免费文献管理工具，认准沁言学术

30

查看详情

实现健壮的错误处理与恢复

为了构建一个更加健壮的输入解析器，我们需要区分io.EOF与其他错误，并在遇到格式错误时，不仅报告错误，还要采取措施跳过无效输入，以便继续处理后续的有效数据。

核心思路是：当fmt.Scan返回非io.EOF的错误时，表明它遇到了一个无法解析为目标类型（例如int）的令牌。这个无效令牌仍然停留在输入缓冲区中。为了让后续的fmt.Scan调用能够读取下一个有效的令牌，我们必须显式地将这个无效令牌从输入流中读取出来并丢弃。通常，可以将其读取到一个字符串变量中。

下面是一个实现此功能的示例代码：

package main

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

func main() {
    nums := make([]int, 0)
    var d int

    fmt.Println("请输入整数，按Ctrl+D (Unix/Linux) 或 Ctrl+Z (Windows) 结束输入。格式错误将跳过并报告。")

    for {
        // 尝试从标准输入读取一个整数
        _, err := fmt.Scan(&d) 

        if err != nil {
            if err == io.EOF {
                // 遇到文件结束符，正常退出循环
                break 
            }

            // 处理其他类型的错误，通常是格式错误
            var invalidToken string
            // 尝试将导致错误的无效令牌读取为字符串，以便继续处理后续输入
            _, scanErr := fmt.Scan(&invalidToken) 
            if scanErr != nil {
                // 如果连读取无效令牌也失败了，说明输入流可能已损坏或遇到EOF
                fmt.Fprintf(os.Stderr, "错误：无法读取无效令牌后的输入: %v\n", scanErr)
                break 
            }
            fmt.Fprintf(os.Stderr, "警告：跳过无效输入: %q\n", invalidToken)
            // 继续下一次循环，尝试读取下一个有效数据
            continue 
        }

        // 如果没有错误，将读取到的整数添加到切片
        nums = append(nums, d)
    }

    fmt.Printf("最终读取到的有效整数: %v\n", nums)
}

代码解析

1. for {} 循环: 创建一个无限循环，持续尝试从输入流中读取数据，直到遇到EOF或不可恢复的错误。
2. fmt.Scan(&d): 尝试将输入流中的下一个空白分隔的令牌解析为一个整数并存储到变量 d 中。
3. if err != nil: 检查 fmt.Scan 返回的错误。
   • if err == io.EOF: 如果错误是 io.EOF，表示已经到达输入流的末尾，此时我们应该优雅地退出循环。
   • 其他错误处理: 对于非 io.EOF 的错误，这通常意味着输入格式不匹配（例如，期望整数却得到了字符串）。
     • var invalidToken string: 声明一个字符串变量来存储无效的令牌。
     • _, scanErr := fmt.Scan(&invalidToken): 这是关键一步。我们再次调用 fmt.Scan，但这次是尝试将输入流中的下一个令牌（即之前导致错误解析的那个令牌）读取为一个字符串。这样做是为了“消耗”掉这个无效的令牌，将其从输入缓冲区中移除，从而允许后续的 fmt.Scan(&d) 调用能够读取到下一个有效的令牌。
     • if scanErr != nil: 如果在尝试读取无效令牌时也发生了错误（例如，在无效令牌之后立即遇到了EOF），这可能意味着更严重的问题，此时也应退出循环。
     • fmt.Fprintf(os.Stderr, "警告：跳过无效输入: %q\n", invalidToken): 打印警告信息，告知用户哪个输入被跳过了。这里使用 os.Stderr 将警告信息输出到标准错误流，与正常输出分离。
     • continue: 执行 continue 语句，跳过当前循环的剩余部分，直接进入下一次循环迭代，再次尝试读取一个整数。
4. nums = append(nums, d): 如果 fmt.Scan 成功读取并解析了一个整数，则将其添加到 nums 切片中。

注意事项与最佳实践

• 错误报告策略: 在上述示例中，我们选择了报告警告并跳过无效输入。在实际应用中，您可能需要根据业务需求采取不同的策略，例如：
  • 终止程序: 如果任何格式错误都被认为是致命的，可以使用 log.Fatal(err) 来立即终止程序。
  • 收集错误: 将所有格式错误收集到一个错误列表中，并在程序结束时统一报告。
  • 自定义错误处理: 对于更复杂的场景，可以解析 fmt.Scan 返回的错误类型（例如，strconv.NumError）以获取更详细的错误信息。
• 输入源选择:
  • fmt.Scan 默认从 os.Stdin 读取。
  • fmt.Fscan(reader, &d) 可以从任何实现了 io.Reader 接口的源（如文件、网络连接、bytes.Buffer等）读取数据。在测试时，使用 bytes.NewBufferString 结合 fmt.Fscan 是一个方便的方式来模拟输入。
• 性能考量: 对于需要处理大量数据的场景，fmt.Scan 可能会有性能瓶颈。在这种情况下，考虑使用 bufio.Scanner 配合 strconv.Atoi 进行逐行或逐词读取和转换，可以提供更细粒度的控制和更好的性能。
• 输入格式: fmt.Scan 默认使用空白字符（空格、制表符、换行符）作为分隔符。如果您的输入使用其他分隔符，您可能需要使用 bufio.Scanner 或手动解析。

总结

在Go语言中从标准输入读取整数并处理格式错误，需要对fmt.Scan的错误返回机制有深入理解。通过区分io.EOF和格式错误，并在后者发生时主动读取并丢弃无效令牌，我们可以构建出既能优雅处理文件结束，又能健壮地处理并报告输入格式异常的程序。这种模式确保了程序在遇到部分无效数据时仍能继续处理有效数据，提高了程序的容错性和用户体验。

以上就是Go语言：从标准输入读取整数并处理格式错误的详细内容，更多请关注php中文网其它相关文章！