将流中的字节发送到两个地方

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《将流中的字节发送到两个地方》，聊聊，我们一起来看看吧！

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

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 用于使用从 respo.body 读取的内容来填充 &buf。

但是，函数（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 新手，不确定上述方向是否正确。

解决方案

在您的场景中，最好为 2 个网络调用提供 2 个独立的字节流。如果它们依赖于一个源流，则当 sendtoclient 停止时，uploadetoserver 将挂起。通道不能解决上述问题，但会引入锁定开销。

你可以尝试io.multiwriter制作2个独立的字节流

var buf1, buf2 bytes.Buffer
mw := io.MultiWriter(&buf1, &buf2)

if _, err := io.Copy(mw, r.Body); err != nil {
    ...
}
go sendToClient(buf1)
go uploadeToServer(buf2)
...

理论要掌握，实操不能落！以上关于《将流中的字节发送到两个地方》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！