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

花韻仙語

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

167人浏览过

来源于php中文网

原创

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) }: 检查扫描过程中是否发生错误。

使用方法：

动态WEB网站中的PHP和MySQL：直观的QuickPro指南第2版

动态WEB网站中的PHP和MySQL详细反映实际程序的需求，仔细地探讨外部数据的验证(例如信用卡卡号的格式)、用户登录以及如何使用模板建立网页的标准外观。动态WEB网站中的PHP和MySQL的内容不仅仅是这些。书中还提到如何串联JavaScript与PHP让用户操作时更快、更方便。还有正确处理用户输入错误的方法，让网站看起来更专业。另外还引入大量来自PEAR外挂函数库的强大功能，对常用的、强大的包

下载