Go：按顺序接收 os.cmd stdout 和 stderr

Question

我需要从 go 执行子命令并分别处理它的 stdout 和 stderr，同时保持 stdin/stdout 的输出顺序。我尝试了几种不同的方法，但无法实现正确的输出顺序；以下代码显示输出处理顺序是绝对随机的：

package main

import (
    "fmt"
    "log"
    "os/exec"
)

var (
    result = ""
)

type writer struct {
    result string
    write  func(bytes []byte)
}

func (writer *writer) Write(bytes []byte) (int, error) {
    writer.result += string(bytes) // process result later
    result += string(bytes)
    return len(bytes), nil
}

func main() {
    cmd := exec.Command("bash", "-c", "echo TEST1; echo TEST2 1>&2; echo TEST3")

    stderr := &writer{}
    cmd.Stderr = stderr

    stdout := &writer{}
    cmd.Stdout = stdout

    err := cmd.Start()
    if err != nil {
        log.Fatal(err)
    }

    err = cmd.Wait()
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println(result)
}

多次运行代码可以输出如下：

$ go run main.go
TEST1
TEST3
TEST2

我希望在所有情况下都有以下结果：

$ go run main.go
TEST1
TEST2
TEST3

我无法调用 cmd.CombinedOutput，因为我需要分别实时处理 stdout/stderr。

Answer 1

您正在执行的命令没有“顺序”。它们是并行管道，因此它们实际上是并发的，就像两个 goroutine 并发一样。您当然可以按照收到它们的顺序存储它们，并通过使用通道或互斥锁来标记它们的来源。为了使合成示例的输出不随机，您需要添加一点暂停。但是，我已经成功地将这种方法与实际命令一起使用：

package main

import (
    "fmt"
    "log"
    "os/exec"
    "sync"
)

var (
    result = ""
)

type write struct {
    source string
    data   string
}

type writer struct {
    source string

    mu     *sync.Mutex
    writes *[]write
}

func (w *writer) Write(bytes []byte) (int, error) {
    w.mu.Lock()
    defer w.mu.Unlock()
    *w.writes = append(*w.writes, write{
            source: w.source,
            data:   string(bytes),
    })
    return len(bytes), nil
}

func main() {
    cmd := exec.Command("bash", "-c", "echo TEST1; sleep .1; echo TEST2 1>&2; sleep .1; echo TEST3")

    var mu sync.Mutex
    var writes []write

    cmd.Stderr = &writer{
            source: "STDERR",
            mu:     &mu,
            writes: &writes,
    }
    cmd.Stdout = &writer{
            source: "STDOUT",
            mu:     &mu,
            writes: &writes,
    }

    err := cmd.Start()
    if err != nil {
            log.Fatal(err)
    }

    err = cmd.Wait()
    if err != nil {
            log.Fatal(err)
    }

    fmt.Printf("%q\n", writes)
}

会产生

[{"STDOUT" "TEST1\n"} {"STDERR" "TEST2\n"} {"STDOUT" "TEST3\n"}]