在 Go 中实时打印来自 exec 命令的标准输出

Question

我有一个小的 Go 工具，它基本上允许用户定义一个命令，然后使用os/exec.

我的问题是我想向用户显示命令的输出（stdout/stderr）。

一个例子可能是这样的：用户定义了一个命令，最后是sh test.sh. test.sh 的内容：

echo "Start"
sleep 7s
echo "Done"

使用我当前的实现，用户只能在完整命令完成后才能看到输出。在上面的示例中，用户在命令和第二个完成之前不会看到输出Start。sleepecho

我目前检索命令的输出是这样的：

cmd := exec.Command(command, args...)
cmd.Dir = dir
// Attach to the standard out to read what the command might print
stdout, err := cmd.StdoutPipe()
if err != nil {
    log.Panic(err)
}
// Execute the command
if err := cmd.Start(); err != nil {
    log.Panic(err)
}

buf := new(bytes.Buffer)
buf.ReadFrom(stdout)
log.Print(buf.String())

是否有可能以某种方式实时读取 stdout/stderr 。这意味着一旦用户定义的命令创建并输出它就会被打印出来？

Answer 1

谢谢 mh-cbon。这将我推向了正确的方向。

代码现在看起来像这样，并且完全符合我的要求。我还发现当我使用Run()而不是Start()在命令完成后程序的执行才继续。

cmd := exec.Command(command, args...)
cmd.Dir = dir

var stdBuffer bytes.Buffer
mw := io.MultiWriter(os.Stdout, &stdBuffer)

cmd.Stdout = mw
cmd.Stderr = mw

// Execute the command
if err := cmd.Run(); err != nil {
    log.Panic(err)
}

log.Println(stdBuffer.String())