在使用 Go 语言进行用户交互时，我们经常需要从标准输入（stdin）读取数据。fmt.Scanf 函数是一个常用的选择，但它在处理无效输入时可能会遇到问题，导致程序进入无限循环。本文将介绍如何使用 bufio 包来清空 stdin 输入流，从而避免这种情况。
问题描述
当使用 fmt.Scanf 读取整数时，如果用户输入了非数字字符，fmt.Scanf 会返回一个错误，并且不会从输入流中移除无效字符。这意味着在循环中重复调用 fmt.Scanf 会一直读取到相同的无效字符，导致程序陷入无限循环。
解决方案
解决此问题的关键在于清空 stdin 输入流中的无效字符。我们可以使用 bufio.NewReader 创建一个带缓冲的读取器，然后使用 ReadString(‘n’) 方法读取并丢弃剩余的输入，直到遇到换行符。
示例代码
package mainimport ( "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)}
代码解释
stdin := bufio.NewReader(os.Stdin): 创建一个新的 bufio.Reader，它从标准输入读取数据。bufio.Reader 提供了缓冲功能，可以更有效地读取数据。fmt.Fscan(stdin, &userI): 使用 fmt.Fscan 函数从 stdin 读取一个整数，并将其存储到 userI 变量中。与 fmt.Scanf 类似，但它从 io.Reader 读取，而不是直接从标准输入读取。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 输入流的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1409053.html
微信扫一扫 
支付宝扫一扫 
