Go 中使用 bufio.NewReader 读取带空格输入的方法

本文介绍了如何使用 bufio.NewReader 从标准输入读取包含空格的字符串。重点讲解了 bufio.NewReader 的正确使用方法，以及如何避免常见的初始化错误。通过本文，你将学会如何在 Go 语言中可靠地获取用户输入的完整字符串，而无需担心空格分隔的问题。

在 Go 语言中，使用 fmt.Scan 函数可以方便地读取用户输入，但它默认以空格和换行符作为分隔符。如果需要读取包含空格的完整字符串，bufio.NewReader 就派上用场了。

bufio.NewReader 结合 ReadString 方法，可以按行读取用户输入，并将整行内容作为一个字符串返回。下面是一个示例：

package main

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

func main() {
    reader := bufio.NewReader(os.Stdin)
    fmt.Println("请输入字符串：")
    line, err := reader.ReadString('\n')
    if err != nil {
        fmt.Println("读取输入时发生错误:", err)
        return
    }
    fmt.Println("你输入的是:", line)
}

代码解析：

1. 引入必要的包： bufio 用于缓冲读取，fmt 用于格式化输出，os 用于访问操作系统功能，包括标准输入。
2. 创建 bufio.Reader： reader := bufio.NewReader(os.Stdin) 创建了一个新的 bufio.Reader，它从标准输入 os.Stdin 读取数据。 这是关键的一步，也是容易出错的地方。 必须将 os.Stdin 作为参数传递给 bufio.NewReader，否则 reader 将无法读取任何数据。
3. 提示用户输入： fmt.Println("请输入字符串：") 提示用户输入内容。
4. 读取字符串： line, err := reader.ReadString('\n') 从 reader 读取字符串，直到遇到换行符 \n。 ReadString 方法返回读取到的字符串 line 和一个错误 err。
5. 错误处理： if err != nil { ... } 检查是否发生了错误。 如果发生错误，则打印错误信息并退出程序。
6. 输出结果： fmt.Println("你输入的是:", line) 将读取到的字符串输出到控制台。

注意事项：

• ReadString 方法会将换行符 \n 也包含在返回的字符串中。如果不需要换行符，可以使用 strings.TrimSpace(line) 函数去除字符串首尾的空白字符，包括换行符。
• bufio.NewReader 内部维护了一个缓冲区。 如果读取的数据量超过缓冲区的大小，ReadString 方法可能会返回一个不完整的字符串，并返回 io.EOF 错误。 可以通过 bufio.NewReaderSize 函数创建具有指定大小缓冲区的 bufio.Reader。

总结：

使用 bufio.NewReader 结合 ReadString 方法是 Go 语言中读取包含空格的完整用户输入的常用方法。 通过正确初始化 bufio.Reader 并处理可能发生的错误，可以确保程序的健壮性和可靠性。 记住要将 os.Stdin 传递给 bufio.NewReader，并且根据需要处理换行符。