如何在 Go 程序中正确区分标准输入是否来自终端（TTY）以支持交互式输入

当 Go 程序通过管道（如 echo “x” | ./binary）执行时，fmt.Scanln 会立即返回 EOF，导致递归调用无限循环；而直接运行时可正常读取用户输入。本文介绍如何通过检测 os.Stdin 是否为字符设备（即是否连接终端），安全地切换输入模式。

当 go 程序通过管道（如 `echo “x” | ./binary`）执行时，`fmt.scanln` 会立即返回 eof，导致递归调用无限循环；而直接运行时可正常读取用户输入。本文介绍如何通过检测 `os.stdin` 是否为字符设备（即是否连接终端），安全地切换输入模式。

在 Shell 脚本自动化部署场景中（例如 wget -qO- url | sh -s “param”），Go 程序常被嵌入执行，但又需向真实用户发起交互式确认（如 [y/n]）。此时若盲目调用 fmt.Scanln，因管道使 stdin 不再关联终端（TTY），Scanln 将持续失败并返回 io.EOF —— 若逻辑未妥善处理该错误，极易陷入无限递归或死循环。

根本解法是：主动判断标准输入是否连接到交互式终端。Go 的 os.Stdin.Stat() 可获取文件状态，其中 os.ModeCharDevice 标志位仅在 stdin 是终端（如 /dev/tty）时被置位。据此可安全分支：

package main

import (
    "fmt"
    "io"
    "os"
)

func main() {
    if scanlnTest() {
        fmt.Println("Success!")
    } else {
        fmt.Println("Cancelled.")
    }
}

func scanlnTest() bool {
    stat, err := os.Stdin.Stat()
    if err != nil {
        fmt.Fprintf(os.Stderr, "Failed to stat stdin: %v\n", err)
        return false
    }

    // 检查是否为字符设备（即是否连接到终端）
    if (stat.Mode() & os.ModeCharDevice) == 0 {
        // 非交互环境：避免阻塞或无限循环，直接退出或降级处理
        fmt.Fprintln(os.Stderr, "Warning: No TTY detected. Interactive input disabled.")
        return false // 或根据需求返回默认值、读取环境变量等
    }

    // 交互环境：安全执行 Scanln
    for {
        fmt.Print("Please type yes or no and then press enter [y/n]: ")
        var response string
        _, err := fmt.Scanln(&response)
        if err != nil {
            if err == io.EOF || err == io.ErrUnexpectedEOF {
                fmt.Fprintln(os.Stderr, "\nInput stream closed unexpectedly.")
                continue
            }
            fmt.Fprintf(os.Stderr, "\nRead error: %v\n", err)
            continue
        }
        response = trimSpace(response)
        switch response {
        case "y", "Y", "yes", "YES":
            return true
        case "n", "N", "no", "NO":
            return false
        default:
            fmt.Println("Invalid input. Please enter 'y' or 'n'.")
        }
    }
}

func trimSpace(s string) string {
    // 简单去首尾空格（不依赖 strings 包，保持示例轻量）
    start, end := 0, len(s)
    for start < end && (s[start] == ' ' || s[start] == '\t' || s[start] == '\n' || s[start] == '\r') {
        start++
    }
    for end > start && (s[end-1] == ' ' || s[end-1] == '\t' || s[end-1] == '\n' || s[end-1] == '\r') {
        end--
    }
    return s[start:end]
}

✅ 关键要点总结：

• 永远不要在未校验 TTY 的情况下对 stdin 执行阻塞式读取（如 Scanln/ReadString）；
• 使用 os.Stdin.Stat().Mode() & os.ModeCharDevice != 0 是跨平台检测终端的可靠方式（Linux/macOS/Windows 均支持）；
• 在非 TTY 环境下，应明确降级策略：返回默认值、报错退出、或从 os.Args/环境变量获取配置；
• 实际生产代码中建议添加超时控制（如 time.AfterFunc）和更健壮的错误处理，避免因用户长时间无输入导致卡死；
• 若需强制获取终端（绕过管道限制），可尝试打开 /dev/tty（Linux/macOS）或 CONIN$（Windows），但需注意权限与可移植性。

此方案既保持了交互体验，又确保了脚本化部署的健壮性，是 Go CLI 工具处理混合输入场景的标准实践。