一个进程可以读取自己的“标准输出”流吗？

Question

进程如何读取自己的输出流？我正在编写自动化测试，它在与测试相同的过程中启动一些应用程序子流程（应用程序）。因此，标准输出是测试输出和应用输出的混合。

我想在运行时读取输出流，如果我看到应用程序出错，则无法通过测试。这可能/可行吗？如果是这样，我该怎么办？

注意：我知道我可以将应用程序作为自己独立的进程启动，然后读取它们的输出流。那是我现在所做的很多工作。

另外请注意，这不是How to test a function's output (stdout/stderr) in Go unit tests的骗局，虽然这张票很相似也很有帮助。另一张票是关于捕获单个函数调用的输出。这张票是关于连续阅读整个流。正确的答案也有一点不同 - 它需要一个管道。

Answer 1

是的，您可以使用os.Pipe()然后自己处理：

tmp := os.Stdout
r, w, err := os.Pipe()
if err != nil {
    panic(err)
}
os.Stdout = w

或者将os.Stdout转移到另一个文件或strings.Builder。 这是详细的答案： In Go, how do I capture stdout of a function into a string?

Answer 2

In Go, how do I capture stdout of a function into a string?使用os.Pipe（一种形式的IPC）给出的答案的略微修改版本：

Pipe返回一对连接的文件;从写入w的r返回字节读取。它返回文件和错误，如果有的话。

由于os.Stdout是*os.File，你可以用任何文件替换它。

 package main

 import (
    "bytes"
    "fmt"
    "io"
    "log"
    "os"
 )

 func main() {
    old := os.Stdout
    r, w, _ := os.Pipe() // TODO: handle error.
    os.Stdout = w

    // All stdout will be caputered from here on.

    fmt.Println("this will be caputered")

    // Access output and restore previous stdout.
    outc := make(chan string)

    go func() {
        var buf bytes.Buffer
        io.Copy(&buf, r) // TODO: handle error
        outc <- buf.String()
    }()

    w.Close()
    os.Stdout = old

    out := <-outc
    log.Printf("captured: %s", out)
 }