当前位置:首页 > 文章列表 > Golang > Go问答 > 避免 goroutine 中的 io.Copy 阻塞

避免 goroutine 中的 io.Copy 阻塞

来源:stackoverflow 2024-02-21 23:45:21 0浏览 收藏

来到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学习网公众号,带你了解更多关于的知识点!

版本声明
本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
处理延迟函数中的错误的正确方式处理延迟函数中的错误的正确方式
上一篇
处理延迟函数中的错误的正确方式
当创建简单卷时,磁盘空间不足的解决方法
下一篇
当创建简单卷时,磁盘空间不足的解决方法
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    540次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    504次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    497次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    484次学习
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码