Go 中“reader.ReadString”问题排查
在 Go 中,bufio.Reader 提供了 ReadString 方法来从流,在指定的分隔符处停止。但是,当使用此方法检索用户输入时,有时会发现第一次出现的分隔符并未从返回的字符串中删除。
请考虑以下代码:
package main import ( "bufio" "fmt" "os" ) func main() { reader := bufio.NewReader(os.Stdin) fmt.Print("Who are you? \nEnter your name: ") text, _ := reader.ReadString('\n') if aliceOrBob(text) { fmt.Printf("Hello, ", text) } else { fmt.Printf("You're not allowed in here! Get OUT!!") } } func aliceOrBob(text string) bool { if text == "Alice" { return true } else if text == "Bob" { return true } else { return false } }
当这个程序运行时,它会提示用户输入他们的名字,但即使用户输入“Alice”或“Bob”,程序也会错误地打印“You're not allowed in this!”滚出去!!”。
问题在于 ReadString 方法会读取输入行,直到遇到指定的分隔符。但是,在其默认行为中,它不会从返回的字符串中删除分隔符。这意味着当用户输入“Alice”时,文本的值实际上是“Alicen”,它与 aliceOrBob 函数中的任何名称都不匹配。
要解决此问题,可以使用字符串.TrimSpace 在 aliceOrBob 函数中检查文本之前从文本中删除换行符:
import ( .... "strings" .... ) ... if aliceOrBob(strings.TrimSpace(text)) { ...
或者,可以使用 ReadLine 方法而不是 ReadString 从 os.Stdin 读取一行输入。 ReadLine 返回表示该行的字节切片,不带换行符。但是,由于 aliceOrBob 需要一个字符串,因此需要使用字符串函数将字节切片转换为字符串:
... text, _, _ := reader.ReadLine() if aliceOrBob(string(text)) { ...
以上是为什么我的 Go `reader.ReadString` 函数无法正确处理用户输入?的详细内容。更多信息请关注PHP中文网其他相关文章!