0

0

如何优雅地同步终止两个 Goroutine

碧海醫心

碧海醫心

发布时间:2026-01-03 17:05:02

|

171人浏览过

|

来源于php中文网

原创

如何优雅地同步终止两个 Goroutine

本文介绍在 go 中通过共享退出通道(quit channel)协调多个 goroutine 的生命周期,确保任一 goroutine 异常或正常退出时,其他 goroutine 能立即响应并安全退出,避免资源泄漏和 goroutine 泄露。

在构建 WebSocket 服务(如基于 gorilla/websocket)时,常见模式是为每个连接启动两个长期运行的 Goroutine:一个负责从客户端读取消息(readFromSocket),另一个负责向客户端发送消息(writeToSocket)。理想情况下,二者应“同生共死”——任一 Goroutine 因连接断开、错误或主动关闭而退出时,另一个也应立即停止,而非继续阻塞在通道读取或等待中。

原代码的问题在于:writeToSocket 依赖 p.writeChan 的关闭来退出循环,但 close(p.writeChan) 发生在 Cleanup() 中,而 Cleanup() 又依赖 p.closeEventChan 的信号——该信号仅由 readFromSocket 单方面触发。这导致典型的竞态:若 readFromSocket 先退出并触发 Cleanup(),writeToSocket 才会收到通道关闭信号;但若 writeToSocket 因写失败先退出,则 readFromSocket 仍无限循环,造成 Goroutine 泄漏。

✅ 正确解法是引入统一的退出信号源——一个只读的 quit

以下是重构后的核心实现:

Facetune
Facetune

一款在线照片和视频编辑工具,允许用户创建AI头像

下载
func (p *Player) EventLoop() {
    l4g.Info("Starting player %s event loop", p)
    quit := make(chan struct{}) // 共享退出通道
    go p.readFromSocket(quit)
    go p.writeToSocket(quit)

    // 等待第一个 Goroutine 通知退出
    <-p.closeEventChan

    // 广播退出信号给所有协作 Goroutine
    close(quit)

    // 等待剩余 Goroutine 完成清理(此处共 2 个,已收 1 个,还需收 1 个)
    <-p.closeEventChan

    p.cleanup()
}

func (p *Player) writeToSocket(quit <-chan struct{}) {
    defer func() { p.closeEventChan <- true }() // 统一通知主协程
    for {
        select {
        case <-quit:
            return // 退出信号优先级最高
        case m, ok := <-p.writeChan:
            if !ok {
                return // writeChan 已关闭
            }
            if p.conn == nil || reflect.DeepEqual(network.Packet{}, m) {
                return
            }
            if err := p.conn.WriteJSON(m); err != nil {
                return
            }
        }
    }
}

func (p *Player) readFromSocket(quit <-chan struct{}) {
    defer func() { p.closeEventChan <- true }()
    for {
        select {
        case <-quit:
            return
        default:
            if p.conn == nil {
                return
            }
            var m network.Packet
            if err := p.conn.ReadJSON(&m); err != nil {
                return
            }
            // 处理消息逻辑(如转发到 writeChan 等)
        }
    }
}

⚠️ 关键注意事项:

  • quit 通道使用 struct{} 类型,零内存开销,语义清晰(仅作信号传递);
  • select +
  • defer 确保无论何种路径退出,都能向 p.closeEventChan 发送完成信号;
  • 主协程需按预期数量接收 closeEventChan 信号(本例为 2 次),避免因漏收导致阻塞;
  • p.writeChan 和 p.closeEventChan 本身不应在 EventLoop 中提前关闭,而应由 Cleanup() 在所有子 Goroutine 退出后统一关闭(或由 defer 隐式处理)。

这种模式可轻松扩展至 N 个协作 Goroutine(如增加 ping/pong 心跳协程),只需共享同一 quit 通道并计数等待即可,是 Go 并发编程中管理协同任务生命周期的标准实践。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
if什么意思
if什么意思

if的意思是“如果”的条件。它是一个用于引导条件语句的关键词,用于根据特定条件的真假情况来执行不同的代码块。本专题提供if什么意思的相关文章,供大家免费阅读。

772

2023.08.22

Golang channel原理
Golang channel原理

本专题整合了Golang channel通信相关介绍,阅读专题下面的文章了解更多详细内容。

247

2025.11.14

golang channel相关教程
golang channel相关教程

本专题整合了golang处理channel相关教程,阅读专题下面的文章了解更多详细内容。

344

2025.11.17

Golang WebSocket与实时通信开发
Golang WebSocket与实时通信开发

本专题系统讲解 Golang 在 WebSocket 开发中的应用,涵盖 WebSocket 协议、连接管理、消息推送、心跳机制、群聊功能与广播系统的实现。通过构建实际的聊天应用或实时数据推送系统,帮助开发者掌握 如何使用 Golang 构建高效、可靠的实时通信系统,提高并发处理与系统的可扩展性。

20

2025.12.22

PHP WebSocket 实时通信开发
PHP WebSocket 实时通信开发

本专题系统讲解 PHP 在实时通信与长连接场景中的应用实践,涵盖 WebSocket 协议原理、服务端连接管理、消息推送机制、心跳检测、断线重连以及与前端的实时交互实现。通过聊天系统、实时通知等案例,帮助开发者掌握 使用 PHP 构建实时通信与推送服务的完整开发流程,适用于即时消息与高互动性应用场景。

59

2026.01.19

个人所得税税率表2026 个人所得税率最新税率表
个人所得税税率表2026 个人所得税率最新税率表

以工资薪金所得为例,应纳税额 = 应纳税所得额 × 税率 - 速算扣除数。应纳税所得额 = 月度收入 - 5000 元 - 专项扣除 - 专项附加扣除 - 依法确定的其他扣除。假设某员工月工资 10000 元,专项扣除 1000 元,专项附加扣除 2000 元,当月应纳税所得额为 10000 - 5000 - 1000 - 2000 = 2000 元,对应税率为 3%,速算扣除数为 0,则当月应纳税额为 2000×3% = 60 元。

0

2026.01.26

oppo云服务官网登录入口 oppo云服务登录手机版
oppo云服务官网登录入口 oppo云服务登录手机版

oppo云服务https://cloud.oppo.com/可以在云端安全存储您的照片、视频、联系人、便签等重要数据。当您的手机数据意外丢失或者需要更换手机时,可以随时将这些存储在云端的数据快速恢复到手机中。

1

2026.01.26

抖币充值官方网站 抖币性价比充值链接地址
抖币充值官方网站 抖币性价比充值链接地址

网页端充值步骤:打开浏览器,输入https://www.douyin.com,登录账号;点击右上角头像,选择“钱包”;进入“充值中心”,操作和APP端一致。注意:切勿通过第三方链接、二维码充值,谨防受骗

3

2026.01.26

Java Spring Security 与认证授权
Java Spring Security 与认证授权

本专题系统讲解 Java Spring Security 框架在认证与授权中的应用,涵盖用户身份验证、权限控制、JWT与OAuth2实现、跨站请求伪造(CSRF)防护、会话管理与安全漏洞防范。通过实际项目案例,帮助学习者掌握如何 使用 Spring Security 实现高安全性认证与授权机制,提升 Web 应用的安全性与用户数据保护。

24

2026.01.26

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
WEB前端教程【HTML5+CSS3+JS】
WEB前端教程【HTML5+CSS3+JS】

共101课时 | 8.5万人学习

JS进阶与BootStrap学习
JS进阶与BootStrap学习

共39课时 | 3.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号