Golang 中处理标准输入 (stdin) 的正确姿势-Golang-PHP中文网

Golang 中处理标准输入 (stdin) 的正确姿势

花韻仙語

发布： 2025-10-22 10:36:24

原创

154人浏览过

golang 中处理标准输入 (stdin) 的正确姿势

本文旨在帮助初学者理解如何在 Golang 程序中正确处理标准输入 (stdin)。我们将探讨如何检测标准输入是否为空，并提供一种优雅的解决方案，允许程序在没有输入时执行特定操作，例如显示帮助信息，而不是无限期地等待输入。同时，我们将介绍如何灵活地从文件或标准输入读取数据，以适应不同的使用场景。

在 Golang 中，使用 os.Stdin 可以访问标准输入。然而，直接读取 os.Stdin 可能会导致程序在没有输入时无限期地等待，这在某些情况下并不是期望的行为。例如，我们可能希望程序在没有通过管道或重定向提供输入时，显示帮助信息并退出。

以下是一个简单的示例，展示了如何从标准输入读取数据并打印每一行：

package main

import (
    "bufio"
    "fmt"
    "os"
)

func main() {
    scanner := bufio.NewScanner(os.Stdin)
    scanner.Split(bufio.ScanLines)
    for scanner.Scan() {
        str := scanner.Text()
        fmt.Println(str)
    }
}

登录后复制

上述代码在有输入时工作正常，但当没有输入时，程序会一直阻塞，等待输入。为了解决这个问题，我们可以结合使用 flag 包和 os.Stdin 来实现更灵活的输入处理。

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

解决方案：从文件或标准输入读取数据

一种常见的做法是允许程序接受一个文件名作为命令行参数。如果提供了文件名，则程序从该文件读取数据；否则，程序从标准输入读取数据。

以下是修改后的代码示例：

package main

import (
    "bufio"
    "flag"
    "fmt"
    "log"
    "os"
)

func main() {
    flag.Parse() // 解析命令行参数

    var in *os.File
    var err error

    switch name := flag.Arg(0); {
    case name == "":
        // 没有提供文件名，从标准输入读取
        in = os.Stdin
    default:
        // 提供了文件名，尝试打开文件
        if in, err = os.Open(name); err != nil {
            log.Fatal(err)
        }
        defer in.Close() // 确保文件在使用完毕后关闭
    }

    scanner := bufio.NewScanner(in)
    scanner.Split(bufio.ScanLines)
    for scanner.Scan() {
        str := scanner.Text()
        fmt.Println(str)
    }

    if err := scanner.Err(); err != nil {
        log.Fatal(err)
    }
}

登录后复制

代码解释：

flag.Parse(): 解析命令行参数。flag.Arg(0) 获取第一个非 flag 参数，这里我们用它来表示文件名。
switch name := flag.Arg(0); { ... }: 使用 switch 语句检查是否提供了文件名。
- case name == "":: 如果没有提供文件名，则将 in 设置为 os.Stdin，从标准输入读取。
- default:: 如果提供了文件名，则尝试打开该文件。如果打开失败，则使用 log.Fatal(err) 记录错误并退出程序。 defer in.Close() 确保在函数返回前关闭文件，这是一个良好的编程习惯。
scanner := bufio.NewScanner(in): 创建一个 bufio.Scanner，用于从 in（可能是 os.Stdin 或打开的文件）读取数据。
scanner.Split(bufio.ScanLines): 设置扫描器以按行读取数据。
for scanner.Scan() { ... }: 循环读取每一行数据，并使用 fmt.Println(str) 打印到标准输出。
if err := scanner.Err(); err != nil { log.Fatal(err) }: 检查扫描过程中是否发生错误。

使用方法：

慧中标AI标书

慧中标AI标书是一款AI智能辅助写标书工具。

120

查看详情

从标准输入读取：
```
go run your_program.go
```
登录后复制
程序将等待从键盘输入数据。
从文件读取：
```
go run your_program.go input.txt
```
登录后复制
程序将从 input.txt 文件读取数据。
通过管道读取：
```
cat input.txt | go run your_program.go
```
登录后复制
程序将从管道读取数据。