例如，在 Python 中，我可以执行以下操作：
realout = sys.stdout
sys.stdout = StringIO.StringIO()
some_function() # prints to stdout get captured in the StringIO object
result = sys.stdout.getvalue()
sys.stdout = realout
你可以在 Go 中做到这一点吗？
【问题讨论】：
我同意你应该使用fmt.Fprint 功能，如果你可以管理它。但是，如果您不控制要捕获其输出的代码，则可能没有该选项。
Mostafa 的回答有效，但如果你想在没有临时文件的情况下这样做，你可以使用os.Pipe。这是一个与 Mostafa 等效的示例，其中包含一些受 Go 的测试包启发的代码。
package main
import (
"bytes"
"fmt"
"io"
"os"
)
func print() {
fmt.Println("output")
}
func main() {
old := os.Stdout // keep backup of the real stdout
r, w, _ := os.Pipe()
os.Stdout = w
print()
outC := make(chan string)
// copy the output in a separate goroutine so printing can't block indefinitely
go func() {
var buf bytes.Buffer
io.Copy(&buf, r)
outC <- buf.String()
}()
// back to normal state
w.Close()
os.Stdout = old // restoring the real stdout
out := <-outC
// reading our temp stdout
fmt.Println("previous output:")
fmt.Print(out)
}
【讨论】：
io.Copy，因为管道的缓冲能力存在限制。移动 goroutine 的启动是微不足道的，并且完全绕过了这个问题。例如：play.golang.org/p/PNqa5M8zo7
os.Pipe 的错误，并且可能使用defer 来恢复os.Stdout，这样任何可能退出函数的添加代码都不需要担心它。这对于并发使用是非常不安全的。
这个答案与之前的答案相似，但使用 io/ioutil 看起来更简洁。
http://play.golang.org/p/fXpK0ZhXXf
package main
import (
"fmt"
"io/ioutil"
"os"
)
func main() {
rescueStdout := os.Stdout
r, w, _ := os.Pipe()
os.Stdout = w
fmt.Println("Hello, playground") // this gets captured
w.Close()
out, _ := ioutil.ReadAll(r)
os.Stdout = rescueStdout
fmt.Printf("Captured: %s", out) // prints: Captured: Hello, playground
}
【讨论】：
我不建议这样做，但您可以通过更改 os.Stdout 来实现。由于此变量的类型为os.File，因此您的临时输出也应该是一个文件。
package main
import (
"fmt"
"io/ioutil"
"os"
"path/filepath"
)
func print() {
fmt.Println("output")
}
func main() {
// setting stdout to a file
fname := filepath.Join(os.TempDir(), "stdout")
fmt.Println("stdout is now set to", fname)
old := os.Stdout // keep backup of the real stdout
temp, _ := os.Create(fname) // create temp file
os.Stdout = temp
print()
// back to normal state
temp.Close()
os.Stdout = old // restoring the real stdout
// reading our temp stdout
fmt.Println("previous output:")
out, _ := ioutil.ReadFile(fname)
fmt.Print(string(out))
}
我不建议这样做，因为这太过分了，而且在 Go 中不是很惯用。我建议将io.Writer 传递给函数并将输出写入其中。这是做几乎相同事情的更好方法。
package main
import (
"bytes"
"fmt"
"io"
"os"
)
func print(w io.Writer) {
fmt.Fprintln(w, "output")
}
func main() {
fmt.Println("print with byes.Buffer:")
var b bytes.Buffer
print(&b)
fmt.Print(b.String())
fmt.Println("print with os.Stdout:")
print(os.Stdout)
}
【讨论】：
我认为整个想法根本是不可取的（竞争条件），但我想人们可能会以与您的示例类似/类比的方式混淆 os.Stdout。
【讨论】：
os.Stdout 是一个全局变量。无条件/无保护设置和恢复是竞争条件的典型示例。