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

在使用Go语言的bufio包读取标准输入时，经常会遇到读取的字符串包含换行符的问题。这会导致在后续的输出中，文本被显示在新的行上，影响用户体验。本文将介绍如何有效地去除bufio.Reader读取的字符串末尾的换行符，从而实现将后续文本与用户输入显示在同一行的目的。

在使用bufio.NewReader读取输入时，ReadString('\n') 方法会一直读取到换行符为止，并将换行符包含在返回的字符串中。因此，我们需要手动去除这个换行符。strings.TrimRight 函数可以方便地从字符串的末尾去除指定的字符。

以下是一个示例代码，展示了如何使用 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")
}

代码解释：

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

1. 引入必要的包: bufio 用于读取输入，fmt 用于格式化输出，os 用于访问操作系统功能，strings 用于字符串操作。
2. 创建 bufio.Reader: in := bufio.NewReader(os.Stdin) 创建一个新的 bufio.Reader，它从标准输入读取数据。
3. 读取输入: input, err := in.ReadString('\n') 从标准输入读取一行，直到遇到换行符。如果发生错误，则打印错误信息并退出。
4. 去除换行符: strings.TrimRight(input, "\n") 从 input 字符串的末尾删除所有换行符。
5. 输出结果: fmt.Println(strings.TrimRight(input, "\n"), "something") 将去除换行符后的输入字符串与 "something" 连接起来，并打印到标准输出。

运行示例：

Krikey AI

68

查看详情

如果用户输入 "some input"，则输出如下：

Example
example: some input
some input something

注意事项：

• strings.TrimRight 函数只会去除字符串末尾的换行符。如果字符串中包含其他位置的换行符，则不会被去除。
• 除了 \n 之外，不同的操作系统可能使用不同的换行符，例如 Windows 使用 \r\n。如果需要处理多种换行符，可以使用 strings.TrimRight(input, "\r\n")。
• 在实际应用中，可能还需要处理输入字符串前后的空格。可以使用 strings.TrimSpace 函数去除字符串前后的空格。

总结：

通过使用 strings.TrimRight 函数，可以方便地去除 bufio.Reader 读取的字符串末尾的换行符，从而实现更灵活的输入输出控制。在处理用户输入时，务必注意去除不必要的空格和换行符，以保证程序的正确性和用户体验。

以上就是Go语言使用bufio读取输入并避免换行的详细内容，更多请关注php中文网其它相关文章！