Go语言读取标准输入到文件方法

作者：归人云淡风轻时间：2026-01-06 来源：正软商城浏览：0

本文旨在帮助初学者理解Go语言中如何处理文件输入，特别是当程序需要从标准输入读取数据，并在没有输入时提供默认行为（例如显示帮助信息）的场景。我们将探讨如何检测标准输入是否为空，并提供灵活的解决方案，允许程序从文件或标准输入读取数据。

本文目录

Go语言中处理文件输入：从标准输入到文件读取

在Go语言中，从标准输入（stdin）读取数据是一种常见的操作。然而，当没有数据通过管道或重定向传递给程序时，程序可能会无限期地等待输入，这可能不是我们期望的行为。本文将介绍如何处理这种情况，并提供一种更灵活的方法，允许程序从文件或标准输入读取数据。

从标准输入读取数据

使用 bufio.Scanner 是从标准输入读取数据的常用方法。以下是一个简单的示例：

package main

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

func main() {
    scanner := bufio.NewScanner(os.Stdin)
    for scanner.Scan() {
        str := scanner.Text()
        fmt.Println(str)
    }
    if err := scanner.Err(); err != nil {
        fmt.Fprintln(os.Stderr, "reading standard input:", err)
    }
}

这段代码会逐行读取标准输入，并将每行打印到标准输出。但是，如果没有数据输入，程序将一直等待。

检测标准输入是否为空

要解决程序无限等待的问题，我们需要一种方法来检测标准输入是否为空。不幸的是，Go标准库没有直接提供这样的功能。一种常见的做法是结合使用 flag 包来处理命令行参数，并根据参数决定从文件或标准输入读取数据。

使用 flag 包处理文件输入

以下是一个示例，展示了如何使用 flag 包来指定输入文件，并在没有指定文件时默认从标准输入读取数据：

package main

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

func main() {
    var in *os.File
    var err error

    flag.Parse()

    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)
    for scanner.Scan() {
        str := scanner.Text()
        fmt.Println(str)
    }
    if err := scanner.Err(); err != nil {
        fmt.Fprintln(os.Stderr, "reading input:", err)
    }
}

在这个示例中，我们使用 flag.Arg(0) 获取第一个命令行参数，如果参数为空，则从标准输入读取数据，否则尝试打开指定的文件。

运行示例

从文件读取：
```
go run main.go input.txt
```
假设 input.txt 文件包含一些文本行，程序会将这些行打印到标准输出。
从标准输入读取：
```
go run main.go
```
在这种情况下，程序会等待用户从键盘输入数据，每输入一行并按下回车键，程序会将该行打印到标准输出。如果没有输入，程序会一直等待，直到用户输入数据或使用 Ctrl+C 终止程序。如果需要程序在没有输入时退出，则需要更复杂的逻辑来判断标准输入是否为空。可以考虑使用 os.Stdin.Stat() 获取文件信息，并检查文件大小是否为0。
使用管道：
```
cat input.txt | go run main.go
```
程序将从管道接收数据，并将每一行打印到标准输出。