确定 Go 中的管道输入

了解命令是否通过管道传输在 Go 应用程序中至关重要，尤其是在处理来自各种来源的数据时。本文探讨了如何确定命令是否通过管道传输，使开发人员能够相应地调整他们的代码。

解决方案

Go 提供了 os.Stdin.Stat( ) 函数检索与标准输入关联的文件信息。此信息包括文件模式，该模式指示输入是来自终端还是管道。以下代码片段演示了如何使用 os.Stdin.Stat() 来实现此目的：

package main

import (
    "fmt"
    "os"
)

func main() {
    fi, _ := os.Stdin.Stat()

    if (fi.Mode() & os.ModeCharDevice) == 0 {
        fmt.Println("data is from pipe")
    } else {
        fmt.Println("data is from terminal")
    }
}

当命令通过管道传输时，fi.Mode() & os.ModeCharDevice 的计算结果为 0，表示输入不是来自字符设备（例如终端）。相反，非零值表示输入来自字符设备。

此方法提供了一种区分管道输入和非管道输入的可靠方法，允许开发人员相应地定制应用程序的行为。