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

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

人声去除
人声去除

用强大的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 并发编程中管理协同任务生命周期的标准实践。

相关专题

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

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

715

2023.08.22

Golang channel原理
Golang channel原理

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

242

2025.11.14

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

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

321

2025.11.17

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

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

16

2025.12.22

php源码安装教程大全
php源码安装教程大全

本专题整合了php源码安装教程,阅读专题下面的文章了解更多详细内容。

203

2025.12.31

php网站源码教程大全
php网站源码教程大全

本专题整合了php网站源码相关教程,阅读专题下面的文章了解更多详细内容。

111

2025.12.31

视频文件格式
视频文件格式

本专题整合了视频文件格式相关内容,阅读专题下面的文章了解更多详细内容。

124

2025.12.31

不受国内限制的浏览器大全
不受国内限制的浏览器大全

想找真正自由、无限制的上网体验?本合集精选2025年最开放、隐私强、访问无阻的浏览器App,涵盖Tor、Brave、Via、X浏览器、Mullvad等高自由度工具。支持自定义搜索引擎、广告拦截、隐身模式及全球网站无障碍访问,部分更具备防追踪、去谷歌化、双内核切换等高级功能。无论日常浏览、隐私保护还是突破地域限制,总有一款适合你!

93

2025.12.31

出现404解决方法大全
出现404解决方法大全

本专题整合了404错误解决方法大全,阅读专题下面的文章了解更多详细内容。

663

2025.12.31

热门下载

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

精品课程

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

共101课时 | 8.1万人学习

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

共39课时 | 3.1万人学习

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

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