在Golang程序开发中,标准输入(stdin)是程序与外部交互的重要入口,很多命令行工具、批处理脚本都需要从标准输入读取用户或上游程序的输入内容。不同的输入场景对读取方式的要求不同,选择不恰当的处理方式很容易出现读取不全、程序阻塞或者资源浪费的问题。

Golang 标准输入的基础认知
Golang的标准输入对应的是os.Stdin,它是一个已经打开的文件对象,类型为*os.File,我们可以直接对这个对象进行读取操作。不过直接使用os.Stdin读取时需要注意,它本质是文件流,读取行为会受到缓冲区、输入结束标志的影响。
基础单次读取示例
最简单的读取方式是使用fmt.Scan系列函数,适合读取格式化的简单输入:
package main
import "fmt"
func main() {
var name string
var age int
// 按空格分隔读取两个值,遇到换行停止
fmt.Print("请输入姓名和年龄,用空格分隔:")
_, err := fmt.Scan(&name, &age)
if err != nil {
fmt.Println("读取输入失败:", err)
return
}
fmt.Printf("姓名:%s,年龄:%dn", name, age)
}
这种方式的局限性很明显,只能读取到第一个换行前的内容,而且要求输入格式严格匹配,不适合读取多行或者非格式化的输入。
不同场景下的正确处理方式
场景一:读取单行完整输入
如果需要读取一整行内容,包括行内的空格,推荐使用bufio.Reader的ReadString方法:
package main
import (
"bufio"
"fmt"
"os"
)
func main() {
reader := bufio.NewReader(os.Stdin)
fmt.Print("请输入一行内容:")
// 读取到换行符为止,返回的字符串包含换行符
line, err := reader.ReadString('n')
if err != nil {
fmt.Println("读取输入失败:", err)
return
}
// 去除末尾的换行符,兼容Windows的rn
line = line[:len(line)-1]
if len(line) > 0 && line[len(line)-1] == 'r' {
line = line[:len(line)-1]
}
fmt.Printf("你输入的内容是:%sn", line)
}
这种方式可以完整读取一行内容,不会因为空格中断读取,适合大多数单行输入的场景。
场景二:读取多行直到输入结束
如果需要读取所有输入内容,直到输入流结束(比如用户按Ctrl+D,或者管道输入结束),可以使用io.ReadAll:
package main
import (
"fmt"
"io"
"os"
)
func main() {
// 读取os.Stdin的所有内容直到流结束
data, err := io.ReadAll(os.Stdin)
if err != nil {
fmt.Println("读取输入失败:", err)
return
}
fmt.Printf("总共读取到%d字节内容:n%sn", len(data), string(data))
}
这种方式适合处理批量的多行输入,比如上游程序通过管道传入的多行数据,但是要注意如果标准输入是终端交互模式,程序会一直阻塞直到用户主动结束输入。
场景三:逐行读取大量输入
如果输入内容很大,一次性读取到内存会占用过多资源,这时候需要逐行处理:
package main
import (
"bufio"
"fmt"
"os"
)
func main() {
reader := bufio.NewReader(os.Stdin)
lineNum := 0
for {
line, err := reader.ReadString('n')
if err != nil {
// 遇到EOF说明输入结束
if err.Error() == "EOF" {
break
}
fmt.Println("读取失败:", err)
return
}
lineNum++
// 处理每一行,这里示例是打印行号和内容
fmt.Printf("第%d行:%s", lineNum, line)
}
fmt.Printf("总共读取了%d行n", lineNum)
}
常见错误和注意事项
- 不要多次关闭
os.Stdin,它是程序启动时自动打开的,关闭后后续无法再读取输入,而且可能导致其他问题。 - 使用
ReadString读取时要注意返回的字符串包含换行符,处理前需要根据需求决定是否去除。 - 交互式程序中不要使用
io.ReadAll读取标准输入,会导致程序一直阻塞等待输入结束。 - 如果输入内容包含非UTF8编码,需要先做编码转换再处理,避免出现乱码。
处理标准输入的核心是根据实际场景选择合适的读取方式,小量单行输入用bufio.Reader的ReadString,大量多行输入用逐行读取,批量非交互输入用io.ReadAll,避免盲目选择读取方式导致程序异常。