在 os/exec.Cmd.Wait() 之后保留标准输出

Question

我正在使用 os/exec，在命令运行时发送输入和接收输出。我需要在命令完成时存储命令的返回码，所以我有一个带有 的 goroutine err := cmd.Wait()，并且我从错误中获取任何失败返回码。但是 Wait() 似乎扔掉了我还需要的剩余标准输出。那么如何在 Cmd.Wait() 之后保留 os/exec.Cmd 的剩余标准输出？示例代码，使用 Unix bc 计算器命令：package mainimport (    "fmt"    "os/exec"    "bufio"    "io"    "time")func main() {    cmd := exec.Command("sh", "-c", "bc")    stdin, _ := cmd.StdinPipe()    stdout, _ := cmd.StdoutPipe()    scanner := bufio.NewScanner(stdout)    cmd.Start()    go func() {        cmd.Wait()        fmt.Println("finished")    }()    io.WriteString(stdin, "1 + 2\n")    fmt.Println(scanner.Scan(), scanner.Text())    io.WriteString(stdin, "3 + 4\n")    fmt.Println(scanner.Scan(), scanner.Text())    io.WriteString(stdin, "5 + 6\n")    io.WriteString(stdin, "quit\n")  // cmd.Wait() runs    time.Sleep(time.Second)    // Prints false :(    fmt.Println(scanner.Scan(), scanner.Text())}这打印： true 3 true 7 finished false我想：真 3 真 7 完成 真 11我还尝试将 cmd.Stdout 设置为 bytes.Buffer，例如：    var buf bytes.Buffer    cmd.Stdout = &buf    scanner := bufio.NewScanner(&buf)但那是不可靠的。除非我用 time.Sleep() 添加延迟，否则它打印的都是假的。

Answer 1

读取到 stdout 末尾后调用 cmd.Wait()。

选项 1：在 scanner.Scan() 返回 false 后从主 goroutine 调用 cmd.Wait。

cmd := exec.Command("sh", "-c", "bc")

stdin, _ := cmd.StdinPipe()

stdout, _ := cmd.StdoutPipe()

scanner := bufio.NewScanner(stdout)

cmd.Start()

io.WriteString(stdin, "1 + 2\n")

fmt.Println(scanner.Scan(), scanner.Text())

io.WriteString(stdin, "3 + 4\n")

fmt.Println(scanner.Scan(), scanner.Text())

io.WriteString(stdin, "5 + 6\n")

io.WriteString(stdin, "quit\n") // cmd.Wait() runs

fmt.Println(scanner.Scan(), scanner.Text())

fmt.Println(scanner.Scan(), scanner.Text()) // prints false

cmd.Wait()

选项 2：从等待的 goroutine 中读取：

cmd := exec.Command("sh", "-c", "bc")

stdin, _ := cmd.StdinPipe()

stdout, _ := cmd.StdoutPipe()

scanner := bufio.NewScanner(stdout)

cmd.Start()

var wg sync.WaitGroup

wg.Add(1)

go func() {

    for scanner.Scan() {

        fmt.Println(scanner.Text())

    }

    cmd.Wait()

    defer wg.Done()

}()

io.WriteString(stdin, "1 + 2\n")

io.WriteString(stdin, "3 + 4\n")

io.WriteString(stdin, "5 + 6\n")

io.WriteString(stdin, "quit\n") // cmd.Wait() runs

wg.Wait()