首页 >后端开发 >Golang >为什么我的 Go `reader.ReadString` 函数无法正确处理用户输入?

为什么我的 Go `reader.ReadString` 函数无法正确处理用户输入?

DDD
DDD原创
2024-12-14 03:21:10670浏览

Why is my Go `reader.ReadString` function not correctly processing user input?

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中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn