Trong Go, làm thế nào để tôi chụp được hàm stdout của một hàm?

Question

Trong Python, ví dụ, tôi có thể làm như sau:

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 

Bạn có thể làm điều này trong Go?

Answer 1

Tôi đồng ý bạn nên sử dụng các chức năng fmt.Fprint nếu bạn có thể quản lý. Tuy nhiên, nếu bạn không kiểm soát mã có đầu ra bạn đang chụp, bạn có thể không có tùy chọn đó.

Câu trả lời của Mostafa hoạt động, nhưng nếu bạn muốn làm điều đó mà không có tệp tạm thời, bạn có thể sử dụng os.Pipe. Dưới đây là một ví dụ tương đương với Mostafa với một số mã lấy cảm hứng từ gói thử nghiệm của 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) 
} 

Answer 2

Tôi nghĩ rằng toàn bộ ý tưởng không được khuyến khích (điều kiện chủng tộc) ở tất cả, nhưng tôi đoán người ta có thể gây rối với os.Stdout theo cách tương tự/tương tự với ví dụ của bạn.

Answer 3

Tôi không đề xuất điều này, nhưng bạn có thể đạt được điều này bằng cách thay đổi os.Stdout. Vì biến này thuộc loại os.File, đầu ra tạm thời của bạn cũng phải là tệp.

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)) 
} 

Tôi không khuyên bạn nên vì đây là quá nhiều hack, và không phải là rất thành ngữ trong Go. Tôi khuyên bạn nên chuyển một số io.Writer đến các hàm và viết kết quả đầu ra cho điều đó. Đây là cách tốt hơn để làm gần như cùng một điều.

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) 
} 

Answer 4

Câu trả lời này tương tự như câu trả lời trước nhưng trông sạch hơn bằng cách sử dụng 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 
}