在Go语言开发过程中,经常需要程序等待用户从命令行输入内容,再根据输入执行对应的逻辑,比如实现交互式命令行工具、读取用户配置参数等场景都需要用到这个功能。Go语言的标准库提供了多个包来支持读取命令行输入,不同的方案适用于不同的使用场景。

使用fmt包读取输入
fmt包提供了基础的输入读取方法,适合读取简单的单行输入,使用起来非常便捷。最常用的就是fmt.Scan和fmt.Scanln两个函数,二者都是用来读取标准输入的内容,但是行为上有一些区别。
fmt.Scan会读取输入直到遇到空白符(空格、换行、制表符等)就停止,不会读取换行符,而fmt.Scanln会读取到换行符为止,并且会把换行符从输入缓冲区中消费掉。
基础读取示例
package main
import "fmt"
func main() {
var name string
var age int
// 提示用户输入内容
fmt.Print("请输入你的姓名和年龄,用空格分隔:")
// 使用fmt.Scan读取输入,会自动按空白符分割赋值给对应变量
_, err := fmt.Scan(&name, &age)
if err != nil {
fmt.Println("读取输入失败:", err)
return
}
fmt.Printf("你输入的姓名是:%s,年龄是:%dn", name, age)
}
如果需要读取一行完整的字符串,包括中间的空格,可以使用fmt.Scanln配合字符串变量,示例如下:
package main
import "fmt"
func main() {
var address string
fmt.Print("请输入你的家庭地址:")
// Scanln会读取到换行符为止,适合读取带空格的单行字符串
_, err := fmt.Scanln(&address)
if err != nil {
fmt.Println("读取地址失败:", err)
return
}
fmt.Println("你输入的地址是:", address)
}
使用bufio包读取输入
如果需要读取更复杂的输入,比如包含换行符的多行内容,或者需要更灵活的输入控制,就可以使用bufio包。bufio包提供了带缓冲的读取器,可以逐行或者逐个字符读取输入内容。
逐行读取示例
最常用的就是创建bufio.Reader实例,然后调用它的ReadString方法读取到指定的分隔符为止,通常我们用换行符作为分隔符来读取单行内容。
package main
import (
"bufio"
"fmt"
"os"
)
func main() {
// 创建读取标准输入的带缓冲读取器
reader := bufio.NewReader(os.Stdin)
fmt.Print("请输入一段描述信息:")
// 读取到换行符为止,返回的字符串会包含换行符
input, err := reader.ReadString('n')
if err != nil {
fmt.Println("读取输入失败:", err)
return
}
// 可以按需处理掉末尾的换行符
fmt.Printf("你输入的内容是:%s", input)
}
多行读取示例
如果需要读取多行输入,直到用户输入特定的结束标记才停止,可以循环调用ReadString方法,示例如下:
package main
import (
"bufio"
"fmt"
"os"
"strings"
)
func main() {
reader := bufio.NewReader(os.Stdin)
var lines []string
fmt.Println("请输入多行内容,输入end单独一行结束输入:")
for {
line, err := reader.ReadString('n')
if err != nil {
fmt.Println("读取输入失败:", err)
return
}
// 去掉换行符后判断是否为结束标记
trimLine := strings.TrimSuffix(line, "n")
if trimLine == "end" {
break
}
lines = append(lines, line)
}
fmt.Println("你输入的多行内容是:")
for _, l := range lines {
fmt.Print(l)
}
}
两种方案对比
我们可以把两种常用的读取方案做个对比,方便在不同场景下选择:
| 方案 | 适用场景 | 优点 | 缺点 |
|---|---|---|---|
| fmt包函数 | 简单的单行输入,按空白符分割的多个参数输入 | 使用简单,代码量少 | 无法读取带空格的完整单行字符串,无法处理多行输入 |
| bufio包读取器 | 带空格的单行输入,多行输入,需要灵活控制输入的场景 | 功能灵活,支持多行读取,可自定义分隔符 | 使用相对复杂,需要额外创建读取器实例 |
注意事项
- 使用
fmt.Scan系列函数时,如果输入的格式和变量类型不匹配,会返回错误,需要做错误处理。 - 使用
bufio.Reader的ReadString方法时,返回的字符串会包含指定的分隔符,如果需要纯净的内容,要记得做字符串处理。 - 读取输入时如果程序需要长时间等待,要考虑到用户可能会中断输入的情况,做好异常捕获。