同时将来自流的字节发送到两个目的地

Question

将值从流同时广播到两个网络目标的最佳方法是什么？这是简化的代码：

func main() {

    resp, _ := http.Get("http://origin.com/image.jpeg")

    var buf bytes.Buffer

    // tee, when read, writes to &buf

    respBodyTee := io.TeeReader(resp.Body, &buf)

    sendToClient(respBodyTee)

    uploadeToServer(&buf)

}

流不能读取两次，因此TeeReader用于填充&buf从中读取的任何内容respo.Body。

但是，函数 (sendToClient和uploadToServer) 将同步运行，而我希望它们同时工作。

我想到的解决方案是传递一个通道，sendToClient该通道将使用已发送到客户端的字节填充通道。后来uploadToServer从同一频道阅读。沿着这些思路：

func main() {

    resp, _ := http.Get("http://origin.com/image.jpeg")

    ch := make(chan byte)

    go sendToClient(respBodyTee, ch) // pass 'ch' for writing and run in a goroutine

    uploadeToServer(ch) // will read from 'ch' (synchronous)

}

我是 Go 新手，不确定上述方向是否正确。

Answer 1

在您的场景中，最好为 2 个网络调用提供 2 个独立的字节流。如果它们依赖于一个源流，那么sendToClient停顿uploadeToServer就会挂起。通道不能解决上述问题，但会引入锁定开销。你可以尝试io.MultiWriter制作2个独立的字节流var buf1, buf2 bytes.Buffermw := io.MultiWriter(&buf1, &buf2)if _, err := io.Copy(mw, r.Body); err != nil {    ...}go sendToClient(buf1)go uploadeToServer(buf2)...