go - 检查用户输入的字符串

Question

我是 GoLang 的新手，我遇到了这种情况的问题:即使用户输入的是“1”，也不会进入if语句。

package main

import (
    "bufio"
    "fmt"
    "os"
    "strconv"
    "math"
    "strings"
)

func prompt(toprint string) string{
    if(toprint == ""){
        toprint = "Enter text :";
    }
    reader := bufio.NewReader(os.Stdin);
    fmt.Println(toprint);
    text, _ := reader.ReadString('\n');
    return text;
}

func main() {
    choice := prompt("Please enter '1'");

    if(strings.Compare("1",choice)==0||choice=="1"){
        // D'ONT ENTER HERE EVEN WHEN choice=="1"
    }else{
        // Always go here
    }
}

感谢您的帮助。

Answer 1

这是因为 reader.ReadString 返回包括分隔符在内的所有文本，所以返回的字符串将是 1\n 而不仅仅是 1 .来自 the documentation (我的重点):

func (*Reader) ReadString

func (b *Reader) ReadString(delim byte) (string, error)

ReadString reads until the first occurrence of delim in the input, returning a string containing the data up to and including the delimiter. If ReadString encounters an error before finding a delimiter, it returns the data read before the error and the error itself (often io.EOF). ReadString returns err != nil if and only if the returned data does not end in delim. For simple uses, a Scanner may be more convenient.

也许你想做

return strings.TrimSpace(text)

在 prompt() 的末尾。