Go语言：优雅判断标准输入（STDIN）是否来自管道或终端

package main

import (
    "fmt"
    "io/ioutil"
    "os"
)

func main() {
    bytes, _ := ioutil.ReadAll(os.Stdin) // 潜在的阻塞点

    if len(bytes) > 0 {
        fmt.Println("标准输入有数据: " + string(bytes))
    } else {
        fmt.Println("标准输入无数据。")
    }
}

// os.FileMode 定义
// const (
//     // 文件类型位
//     ModeDir        FileMode = 1 << (32 - 1 - iota) // d: 目录
//     ModeAppend                                     // a: 只能追加
//     ModeExclusive                                  // l: 独占文件
//     ModeTemporary                                  // T: 临时文件 (在Unix上删除)
//     ModeSymlink                                    // L: 符号链接
//     ModeDevice                                     // D: 设备文件
//     ModeNamedPipe                                  // p: 命名管道 (FIFO)
//     ModeSocket                                     // S: Unix域套接字
//     ModeSetuid                                     // u: setuid
//     ModeSetgid                                     // g: setgid
//     ModeCharDevice                                 // c: 字符设备
//     ModeSticky                                     // t: 粘滞位
//     ModeIrregular                                  // ?: 不规则文件；并非普通文件、目录、符号链接、设备、命名管道或套接字。
//
//     // 权限位
//     ModePerm FileMode = 0777 // 所有权限位
// )

package main

import (
    "fmt"
    "io/ioutil"
    "os"
)

func main() {
    // 1. 获取标准输入的文件信息
    // os.Stdin 是一个指向 os.File 类型的指针，代表标准输入。
    // Stat() 方法返回一个 FileInfo 接口，其中包含文件的元数据。
    stat, err := os.Stdin.Stat()
    if err != nil {
        // 如果获取状态失败，通常是权限问题或文件描述符无效。
        fmt.Fprintf(os.Stderr, "获取标准输入状态失败: %v\n", err)
        os.Exit(1) // 退出程序并返回错误码
    }

    // 2. 判断标准输入是否为字符设备 (通常是终端)
    // stat.Mode() 返回文件的模式和权限位。
    // os.ModeCharDevice 是一个常量，表示字符设备的文件模式位。
    // 通过位与操作 `&` 来检查 ModeCharDevice 位是否被设置。
    // 如果 ModeCharDevice 位未设置 (即结果为 0)，则表示它不是字符设备。
    // 这意味着数据可能来自管道、文件重定向或非交互式源。
    if (stat.Mode() & os.ModeCharDevice) == 0 {
        fmt.Println("检测到标准输入来自管道、文件重定向或非交互式源。")
        fmt.Println("正在尝试读取所有可用数据...")

        // 在这种情况下，ioutil.ReadAll 会读取所有可用数据直到EOF，
        // 但不会无限期阻塞，因为管道/文件重定向会提供明确的EOF。
        bytes, err := ioutil.ReadAll(os.Stdin)
        if err != nil {
            fmt.Fprintf(os.Stderr, "从标准输入读取数据失败: %v\n", err)
            os.Exit(1)
        }

        if len(bytes) > 0 {
            fmt.Println("成功读取到数据:")
            fmt.Println(string(bytes))
        } else {
            fmt.Println("标准输入无数据 (管道/重定向为空)。")
        }
    } else {
        // 标准输入是字符设备 (终端)
        fmt.Println("检测到标准输入来自终端。")
        // 在终端模式下，如果直接使用 ioutil.ReadAll，程序会等待用户输入直到EOF (Ctrl+D)。
        // 通常，对于终端输入，我们会使用 bufio.NewScanner 或其他交互式读取方式，
        // 或者等待用户输入特定命令。
        fmt.Println("程序将不会尝试阻塞读取终端输入。")
        fmt.Println("如果需要读取终端输入，请使用 bufio.NewScanner 等方法进行交互式处理。")
    }
}