如何在 Go 中使用 bufio 清空 stdin 输入流

在使用 Go 语言进行用户交互时，我们经常需要从标准输入（stdin）读取数据。fmt.Scanf 函数是一个常用的选择，但它在处理无效输入时可能会遇到问题，导致程序进入无限循环。本文将介绍如何使用 bufio 包来清空 stdin 输入流，从而避免这种情况。

问题描述

当使用 fmt.Scanf 读取整数时，如果用户输入了非数字字符，fmt.Scanf 会返回一个错误，并且不会从输入流中移除无效字符。这意味着在循环中重复调用 fmt.Scanf 会一直读取到相同的无效字符，导致程序陷入无限循环。

解决方案

解决此问题的关键在于清空 stdin 输入流中的无效字符。我们可以使用 bufio.NewReader 创建一个带缓冲的读取器，然后使用 ReadString('\n') 方法读取并丢弃剩余的输入，直到遇到换行符。

示例代码

GPTKit

一个AI文本生成检测工具

108

查看详情

package main

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

func main() {
    stdin := bufio.NewReader(os.Stdin)

    fmt.Println("Please enter an integer: ")

    var userI int

    for {
        _, err := fmt.Fscan(stdin, &userI)
        if err == nil {
            break
        }

        stdin.ReadString('\n') // 清空输入流
        fmt.Println("Sorry, invalid input. Please enter an integer: ")
    }

    fmt.Println(userI)
}

代码解释

1. stdin := bufio.NewReader(os.Stdin): 创建一个新的 bufio.Reader，它从标准输入读取数据。bufio.Reader 提供了缓冲功能，可以更有效地读取数据。
2. fmt.Fscan(stdin, &userI): 使用 fmt.Fscan 函数从 stdin 读取一个整数，并将其存储到 userI 变量中。与 fmt.Scanf 类似，但它从 io.Reader 读取，而不是直接从标准输入读取。
3. stdin.ReadString('\n'): 如果 fmt.Fscan 返回错误，说明输入无效。这时，stdin.ReadString('\n') 会读取并丢弃 stdin 中剩余的字符，直到遇到换行符。这有效地清空了输入流，确保下一次循环可以读取新的输入。

注意事项

• 使用 bufio.Reader 可以提高读取效率，特别是在需要频繁读取输入的情况下。
• ReadString('\n') 会一直读取到换行符，因此确保用户输入以换行符结尾。
• 除了 ReadString('\n')，还可以使用其他方法来清空输入流，例如 ReadBytes('\n') 或 ReadByte()，但 ReadString('\n') 通常是最方便的选择。

总结

通过使用 bufio.NewReader 和 ReadString('\n') 方法，我们可以有效地清空 stdin 输入流，解决 fmt.Scanf 在处理无效输入时可能导致的无限循环问题。这种方法可以提高程序的健壮性和用户体验。

以上就是如何在 Go 中使用 bufio 清空 stdin 输入流的详细内容，更多请关注php中文网其它相关文章！