Question

有没有办法检查输入流（os.Stdin）是否有数据？

帖子Read from initial stdin in GO?显示了如何读取数据，但不幸的是，如果没有数据通过管道传输到标准输入，则会阻塞。

Answer 1

os.Stdin就像任何其他“文件”一样，所以你可以查看它的大小：

package main

import (
    "fmt"
    "os"
)

func main() {
    file := os.Stdin
    fi, err := file.Stat()
    if err != nil {
        fmt.Println("file.Stat()", err)
    }
    size := fi.Size()
    if size > 0 {
        fmt.Printf("%v bytes available in Stdin\n", size)
    } else {
        fmt.Println("Stdin is empty")
    }
}

我将它构建为“管道”可执行文件，以下是它的工作原理：

$ ./pipe
Stdin is empty
$ echo test | ./pipe
5 bytes available in Stdin

Answer 2

这似乎是可靠的解决方案，甚至可以通过管道使用睡眠/延迟数据。 https://coderwall.com/p/zyxyeg/golang-having-fun-with-os-stdin-and-shell-pipes

package main

import (
  "os"
  "fmt"
)

func main() {
  fi, err := os.Stdin.Stat()
  if err != nil {
    panic(err)
  }
  if fi.Mode() & os.ModeNamedPipe == 0 {
    fmt.Println("no pipe :(")
  } else {
    fmt.Println("hi pipe!")
  }
}

确定Stdin是否包含Go数据

2 个答案: