Go语言使用bufio读取输入避免换行问题

在使用Go语言的bufio包进行输入操作时，bufio.NewReader结合ReadString(‘n’)的方式是一种常见的读取用户输入的方法。 然而，这种方法的一个常见问题是ReadString函数会连同换行符n一起读取到字符串中，导致后续输出时出现换行，影响程序的输出格式。

例如，以下代码演示了这个问题：

package main  import (     "bufio"     "fmt"     "os" )  func main() {     fmt.Println("Example")     print("example: ")     in := bufio.NewReader(os.Stdin)     input, err := in.ReadString('n')     if err != nil {         fmt.Println("Error: ", err)         return     }      fmt.Println(input, "something") }

如果用户输入 “some input” 并按下回车，程序的输出将是：

Example example: some input some input  something

可以看到，”something” 被输出到了下一行，这不是我们期望的结果。

立即学习“go语言免费学习笔记（深入）”；

解决方案：移除换行符

解决这个问题的关键在于移除input字符串末尾的换行符。 Go语言的strings包提供了strings.TrimRight函数，可以方便地移除字符串末尾指定的字符。

以下是修改后的代码：

package main  import (     "bufio"     "fmt"     "os"     "strings" )  func main() {     fmt.Println("Example")     print("example: ")     in := bufio.NewReader(os.Stdin)     input, err := in.ReadString('n')     if err != nil {         fmt.Println("Error: ", err)         return     }     fmt.Println(strings.TrimRight(input, "n"), "something") }

在这个修改后的代码中，我们使用strings.TrimRight(input, “n”)移除了input字符串末尾的换行符。 现在，如果用户输入 “some input” 并按下回车，程序的输出将是：

Example example: some input some input something

现在，”something” 已经和用户输入在同一行输出了，达到了预期的效果。

总结

在使用bufio.NewReader和ReadString(‘n’)读取用户输入时，务必注意移除字符串末尾的换行符，以避免输出格式问题。 strings.TrimRight函数是一个简单有效的解决方案。 在实际开发中，应根据具体需求选择合适的字符串处理方法，确保程序的输出符合预期。 除了n，可能还需要考虑移除r（回车符），尤其是在处理来自不同操作系统的输入时。可以使用strings.TrimSuffix函数，它比TrimRight更精确，只移除字符串末尾完全匹配的后缀，避免误删其他字符。例如：strings.TrimSuffix(input, “rn”)或者strings.TrimSuffix(input, “n”)。