避免 goroutine 中的 io.Copy 阻塞

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《避免 goroutine 中的 io.Copy 阻塞》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我正在读《black hat go》这本书。我发现了一个简单的 tcp 代理示例。它基本上只是转发请求并发回回复。代理代码的要点如下（已修改）：

func handle(src net.Conn) {
    dst, err := net.Dial("tcp", "example.com:80")
    if err != nil {
        log.Fatalln("Unable to connect to remote")
    }
    defer dst.Close()

    // Run in goroutine to prevent io.Copy from blocking
    go func() {
        if _, err := io.Copy(dst, src); err != nil {
            log.Fatalln("Something wrong src -> dst")
        }
    }()

    if _, err := io.Copy(src, dst); err != nil {
        log.Fatalln("Something wrong dst -> src")
    }
}

我不明白的部分是注释“在 goroutine 中运行以防止 io.copy 阻塞”。我已经在将其包装在 goroutine 中和没有将其包装在 goroutine 中的情况下运行它，并且它仅适用于 goroutine，但我不明白为什么。难道我们不能针对请求运行一个阻塞 io.copy ，然后再运行另一个阻塞响应吗？我想我很难理解如果第一个 io.copy 在 goroutine 中运行，我们如何保证排序。

解决方案

当您处理 TCP 流时，您必须处理双向流量，从 src 到 dest 以及从 dest 到 src。您可以通过两次复制操作来完成此操作。但是，复制将阻塞，直到收到错误或 EOF（即套接字关闭），因此您必须将至少其中一个复制操作放入 goroutine 中。按照写入方式，一个复制操作将从一个套接字读取数据并写入另一个套接字，另一个复制操作将执行相反的操作，直到发生错误或 EOF。当接收到错误或EOF时，两个复制操作都会停止。

io.Copy函数在读取器或写入器返回错误（包括io.EOF）后返回。如果 Go 例程中未运行 io.Copy 调用，则在将任何数据从 dst 复制到 src 之前，所有数据都会从 src 复制到 dst。

到这里，我们也就讲完了《避免 goroutine 中的 io.Copy 阻塞》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！