34

我想阅读stdinGo 程序的原件。例如，如果我这样做了echo test stdin | go run test.go，我希望能够访问“测试标准输入”。我已经尝试从 中读取os.Stdin，但如果其中没有任何内容，那么它将等待输入。我也尝试先检查大小，但os.Stdin.Stat().Size()即使传入输入也是 0。

我能做些什么？

4

3 回答 3

73

使用 stdin 读取os.Stdin应该按预期工作：

package main

import "os"
import "log"
import "io/ioutil"

func main() {
    bytes, err := ioutil.ReadAll(os.Stdin)

    log.Println(err, string(bytes))
}

执行echo test stdin | go run stdin.go应该打印'test stdin'就好了。

如果您附上用于识别您遇到的问题的代码，将会有所帮助。

对于基于行的阅读，您可以使用bufio.Scanner：

import "os"
import "log"
import "bufio"

func main() {
    s := bufio.NewScanner(os.Stdin)
    for s.Scan() {
        log.Println("line", s.Text())
    }
}
于 2012-09-11T12:26:25.417 回答
20

我认为您的问题本身没有明智的答案，因为没有“初始标准输入”之类的东西。类 Unix 操作系统和 Windows 实现了“标准流”的概念，其工作原理如下（简化）：当创建一个进程时，它会自动打开三个文件描述符（Windows 中的句柄）——stdin、stdout 和 stderr。毫无疑问，你熟悉这个概念，但我想强调“流”这个词的含义——在你的例子中，当你打电话时

$ echo 'test stdin' | ./stdin

shell 创建一个pipe，生成两个进程（一个用于echo二进制文件，一个用于二进制文件）并使用它创建的管道：管道的写入 FD 连接到echo的标准输出，管道的读取 FD 连接到二进制的标准输入。然后，无论echo进程喜欢写入其标准输出的任何内容，都会通过管道（原文如此！）传输到您进程的标准输入。（实际上，当今的大多数 shellecho都是作为内置原语实现的，但这不会以任何方式改变语义；您也可以尝试一下/bin/echo，这是一个真正的程序。还要注意，我只是用来./stdin指代您的程序 -这是为了清楚起见，go run stdin.go最终也会这样做。）

请注意这里的几个关键事项：

  • 写入过程（echo在您的情况下）没有义务向其标准输出写入任何内容（例如，echo -n不会向其标准输出写入任何内容并成功退出）。
  • 它还能够任意延迟写入其数据（或者因为它想要做出这样的延迟，或者因为它已被操作系统抢占或在某些系统调用中休眠，等待某些繁忙的系统资源等）。
  • 操作系统缓冲区通过管道传输。这意味着写入过程发送到管道的内容可能会在读取端以任意块的形式出现。1
  • 只有两种方法可以知道写入端没有更多数据要通过管道发送：
    • 以某种方式将其编码在数据本身中（这意味着在编写器和读取器之间使用商定的数据传输协议）。
    • 写入器可能会关闭管道的一侧，这将导致读取器侧出现“文件结束”情况（但只有在缓冲区耗尽并read尝试再次调用之后，才会失败）。

让我们总结一下：您观察到的行为是正确且正常的。如果您希望从标准输入中获取任何数据，则不能期望它随时可用。如果您也不想阻塞标准输入，那么创建一个 goroutine 将在无限循环中阻止从标准输入读取（但检查 EOF 条件）并通过通道向上传递收集的数据（可能在某些处理之后，如果需要）。

1这就是为什么通常出现在管道中的两个管道之间的某些工具（例如grep）可能具有特殊选项以使它们在写入每一行后刷新其标准输出 - 阅读手册页中的一个示例--line-buffered中的选项。不了解这种“默认情况下完全缓冲”语义的人会感到困惑，为什么当受监控的文件明显更新时似乎停止并且不显示任何内容。greptail -f /path/to/some/file.log | grep whatever | sed ...

附带说明：如果您要“按原样”运行二进制文件，例如

$ ./stdin

这并不意味着生成的进程没有标准输入（或“初始标准输入”或 whaveter），相反，它的标准输入将连接到您的 shell 接收键盘导入的同一流（因此您可以直接在进程的标准输入中键入一些内容）。

让进程的标准输入无处连接的唯一可靠方法是使用

$ ./stdin </dev/null

在类 Unix 操作系统和

C:\> stdin <NUL

在 Windows 上。这个“空设备”使进程首先read从其标准输入中看到 EOF。

于 2012-09-12T08:37:25.013 回答
5

您无法检查标准输入的内容，但您可以检查标准输入是否与终端或管道相关联。IsTerminal 只接受标准的 unix fd 数字 (0,1,2)。syscall 包分配了变量，因此如果您喜欢命名它们，您可以执行 syscall.Stdin。

package main

import (
    "code.google.com/p/go.crypto/ssh/terminal"
    "fmt"
    "io/ioutil"
    "os"
)

func main() {
    if ! terminal.IsTerminal(0) {
        b, _ := ioutil.ReadAll(os.Stdin)
        fmt.Print(string(b))
    } else {
        fmt.Println("no piped data")
    }
}
于 2013-05-25T20:54:52.710 回答