时间戳

首页 > 解决方案 > 如何将 os.Stdout 输出复制到字符串变量

go - 如何将 os.Stdout 输出复制到字符串变量

问题描述

我有这样的功能：

package main

import (
    "fmt"
)

// PrintSomething prints some thing
func PrintSomething() {
    fmt.Println("print something")
}

func main() {
    PrintSomething()
}

如何将 PrintSomething 包装到另一个函数调用 CaptureSomething 以将字符串“print something”保存到变量并返回它？

标签： gooutput

解决方案

创建管道并将标准输出设置为管道编写器。启动一个 goroutine 将管道读取器复制到缓冲区。完成后，关闭管道编写器并等待 goroutine 完成读取。将缓冲区作为字符串返回。

// capture replaces os.Stdout with a writer that buffers any data written 
// to os.Stdout. Call the returned function to cleanup and get the data
// as a string.
func capture() func() (string, error) {
    r, w, err := os.Pipe()
    if err != nil {
        panic(err)
    }

    done := make(chan error, 1)

    save := os.Stdout
    os.Stdout = w

    var buf strings.Builder

    go func() {
        _, err := io.Copy(&buf, r)
        r.Close()
        done <- err
    }()

    return func() (string, error) {
        os.Stdout = save
        w.Close()
        err := <-done
        return buf.String(), err
    }

}

像这样使用它：

done := capture()
fmt.Println("Hello, playground")
capturedOutput, err := done()
if err != nil {
    // handle error
}

游乐场示例

推荐阅读