使用以编程方式编写的 stdin 执行命令并连接到 os.stdin

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《使用以编程方式编写的 stdin 执行命令并连接到 os.stdin》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我正在从我的 go 程序执行 docker-compose 可执行文件。为了配置它，我将 docker-compose 配置直接通过管道传输到 docker-compose 的标准输入。这在大多数情况下都可以正常工作，代码大致如下所示（为了简单起见，我删除了代码片段中的所有错误处理）：

args = append([]string{"-f", "-"}, args...)
cmd := exec.command("docker-compose", args...)
cmd.stdout = os.stdout
cmd.stderr = os.stderr
cmd.stdin = strings.newreader(input)
cmd.run()

args 是传递给 docker-compose 的参数，input 是作为 yaml 字符串的 docker-compose 配置。

虽然这在大多数情况下都有效，但对于交互式命令却会失败。 stdin 未连接到 os.stdin，命令失败。我可以获得交互式命令来使用以下代码：

cmd := exec.command("docker-compose", args...)
cmd.stdout = os.stdout
cmd.stderr = os.stderr
cmd.stdin = os.stdin
cmd.run()

但这没有通过管道传输配置，我最终需要这样做。我尝试以某种方式组合这些版本，但无法使其工作：

input, _ := getDockerComposeAsYaml(config)
args = append([]string{"-f", "-"}, args...)
cmd := exec.Command("docker-compose", args...)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr

stdin, _ := cmd.StdinPipe()

go func() {
    //defer stdin.Close()
    io.WriteString(stdin, input)
}()
cmd.Run()

我在显式关闭和不关闭 stdin 的情况下尝试了上述代码，但所需的行为应该是不关闭它，因为我想保持 stdin 打开以供交互使用。如果不关闭程序，程序就会永远挂起，关闭标准输入后，程序将无法按预期交互工作。

将执行程序的标准输入连接到 os.stdin，同时仍然能够在开始时通过管道输入一些数据的正确方法是什么？

解决方案

写入 stdin 的 goroutine（使用 cmd.StdinPipe 创建的管道）应该：

1. 写入 input 字符串
2. 然后循环真实的 os.Stdin，将其得到的所有内容写入 stdin
3. 当 os.Stdin 遇到 EOF 时，显式关闭 stdin。

但是，子进程可能会在 os.Stdin 关闭之前退出。在这种情况下，管道（stdin 将由 exec.Command 关闭）。您的循环应该检查是否有关闭的 stdin 并在发生这种情况时退出 goroutine。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。