如何在 Go 中通过管道将 yes 命令输出传递给 Shell 脚本执行-Golang-PHP中文网

如何在 Go 中通过管道将 yes 命令输出传递给 Shell 脚本执行

本文介绍在 Go 程序中安全、可靠地模拟 yes | ./install.sh 行为的方法，核心是使用 exec.Command 构建管道链，将 yes 的标准输出连接到目标脚本的标准输入，并正确处理启动顺序与 I/O 重定向。

本文介绍在 go 程序中安全、可靠地模拟 `yes | ./install.sh` 行为的方法，核心是使用 `exec.command` 构建管道链，将 `yes` 的标准输出连接到目标脚本的标准输入，并正确处理启动顺序与 i/o 重定向。

在 Go 中直接用 exec.Command 执行含 shell 管道（如 yes | ./install.sh）的命令会失败，因为 exec.Command 不解析 shell 语法（如 |、>、&），它仅执行单个二进制程序。你之前尝试的两种写法：

exec.Command("/bin/bash", "path/to/install.sh", "| yes") // ❌ 参数被当作脚本参数，非管道
exec.Command("/bin/bash", "yes | path/to/install.sh")     // ❌ bash 未启用 -c，字符串不会被解释

均因误解 exec.Command 的行为而无效。

✅ 正确做法是：*显式创建两个 `exec.Cmd` 实例，手动建立管道连接**，并控制执行时序。以下是推荐实现（已优化健壮性）：

package main

import (
    "os"
    "os/exec"
)

func runYesPipeToScript(scriptPath string) error {
    // 启动 `yes` 命令（无参数，默认持续输出 "y\n"）
    cmdYes := exec.Command("yes")
    yesOut, err := cmdYes.StdoutPipe()
    if err != nil {
        return err
    }

    // 启动安装脚本，将其 stdin 指向 yes 的 stdout
    cmdInstall := exec.Command(scriptPath)
    cmdInstall.Stdin = yesOut
    cmdInstall.Stdout = os.Stdout // 可选：透传脚本输出到终端
    cmdInstall.Stderr = os.Stderr // 强烈建议：不忽略错误流

    // 注意：必须先 Start() 安装脚本，再 Run() yes（否则管道阻塞）
    if err := cmdInstall.Start(); err != nil {
        return err
    }

    // 启动 yes（它会开始写入管道）
    if err := cmdYes.Run(); err != nil {
        return err
    }

    // 等待安装脚本结束
    return cmdInstall.Wait()
}

func main() {
    if err := runYesPipeToScript("./install.sh"); err != nil {
        panic(err)
    }
}

? 关键要点说明：

cmdYes.StdoutPipe() 创建一个可读管道，cmdInstall.Stdin 直接复用该管道——这是 Go 中构建进程间管道的标准方式；
必须先 cmdInstall.Start() 再 cmdYes.Run()：确保接收方（脚本）已就绪读取，避免 yes 因管道满而阻塞；
不推荐使用 /bin/bash -c "yes | ./install.sh" 形式，因其依赖 shell 解析，且难以捕获子进程错误、无法精细控制 I/O；
若脚本需静默运行（不打印 yes 输出），可将 cmdInstall.Stdout 设为 io.Discard；若需捕获输出用于日志，可设为 &bytes.Buffer{}；
生产环境应添加超时控制（cmdInstall.Wait() 前设置 cmdInstall.Process.Signal(os.Interrupt) 或使用 context.WithTimeout）。