0

0

Go 并发模式:使用 WaitGroup 和通道避免死锁

霞舞

霞舞

发布时间:2025-11-09 11:20:26

|

854人浏览过

|

来源于php中文网

原创

go 并发模式:使用 waitgroup 和通道避免死锁

本文深入探讨了Go语言中N个worker goroutine与一个监控goroutine协调时常见的死锁问题。通过分析`sync.WaitGroup`和通道(channel)的不当使用,文章提供了两种有效的解决方案:一是通过在所有worker完成后关闭通道,使接收方优雅退出;二是在打印逻辑也由单独goroutine处理时,引入额外的同步通道来确保主程序正确终止,从而避免`all goroutines are asleep - deadlock`。

在Go语言的并发编程中,goroutine、channel和sync.WaitGroup是实现高效并发模式的核心工具。然而,不恰当的使用方式,尤其是通道的生命周期管理,很容易导致程序进入死锁状态,并抛出all goroutines are asleep - deadlock的错误。本教程将通过一个经典的“生产者-消费者”场景,深入分析这类死锁的成因,并提供两种健壮的解决方案。

理解死锁问题:监控 Goroutine 的无限等待

考虑一个常见的并发场景:有N个工作(worker)goroutine负责生产数据并发送到一个共享通道,一个监控(monitor)goroutine负责从该通道接收并处理数据,而主程序需要等待所有工作和监控任务完成后才退出。

以下是一个可能导致死锁的初始代码示例:

package main

import (
    "fmt"
    "strconv"
    "sync"
)

func worker(wg *sync.WaitGroup, cs chan string, i int) {
    defer wg.Done()
    cs <- "worker" + strconv.Itoa(i)
}

func monitorWorker(wg *sync.WaitGroup, cs chan string) {
    defer wg.Done()
    for i := range cs { // 此处会无限等待
        fmt.Println(i)
    }
}

func main() {
    wg := &sync.WaitGroup{}
    cs := make(chan string)

    // 启动10个worker goroutine
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go worker(wg, cs, i)
    }

    // 启动一个monitorWorker goroutine
    wg.Add(1)
    go monitorWorker(wg, cs)

    // 等待所有goroutine完成
    wg.Wait()
}

问题分析:

上述代码中,main函数启动了10个worker goroutine和一个monitorWorker goroutine。worker goroutine将数据发送到cs通道,并在完成后调用wg.Done()。monitorWorker goroutine通过for i := range cs循环从通道cs接收数据。

死锁发生的原因在于:

  1. 所有worker goroutine完成任务后,都已将数据发送到cs通道并调用了wg.Done()。
  2. 此时,monitorWorker goroutine的for i := range cs循环会继续尝试从cs通道接收数据。由于没有其他goroutine会再向cs发送数据,并且cs通道从未被关闭,monitorWorker将无限期地阻塞在此处。
  3. main函数调用wg.Wait(),它在等待monitorWorker goroutine调用wg.Done()。但monitorWorker被阻塞,无法执行defer wg.Done()。
  4. 最终,程序中所有可运行的goroutine(除了main函数,monitorWorker被阻塞)都已完成或处于休眠状态,导致Go运行时检测到死锁并报错all goroutines are asleep - deadlock。

解决方案一:通过专用 Goroutine 关闭通道

解决此问题的关键在于,当所有发送方完成任务后,必须关闭通道cs,以通知接收方monitorWorker(或main函数中的for range循环)不再有数据会到来,从而使其优雅地退出循环。

我们可以引入一个专门的monitorWorker goroutine来负责等待所有worker完成,然后关闭cs通道。而数据接收和打印的逻辑则可以放在main函数中。

XPaper Ai
XPaper Ai

AI撰写论文、开题报告生成、AI论文生成器尽在XPaper Ai论文写作辅助指导平台

下载
package main

import (
    "fmt"
    "strconv"
    "sync"
)

func worker(wg *sync.WaitGroup, cs chan string, i int) {
    defer wg.Done()
    cs <- "worker" + strconv.Itoa(i)
}

// 新的monitorWorker职责:等待所有worker完成,然后关闭通道
func monitorWorker(wg *sync.WaitGroup, cs chan string) {
    wg.Wait() // 等待所有worker goroutine完成
    close(cs) // 所有worker完成后,关闭通道cs
}

func main() {
    wg := &sync.WaitGroup{}
    cs := make(chan string)

    // 启动10个worker goroutine
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go worker(wg, cs, i)
    }

    // 启动一个专门的monitorWorker来关闭通道
    go monitorWorker(wg, cs) // 注意:这里不再将monitorWorker添加到wg中

    // main goroutine负责从通道接收并打印数据
    for i := range cs { // 当cs通道被关闭时,此循环将自动退出
        fmt.Println(i)
    }

    // main函数在此处退出,程序结束
}

方案分析:

  1. worker goroutine照常发送数据并调用wg.Done()。
  2. 新的monitorWorker goroutine在启动后,立即调用wg.Wait()。它会阻塞直到所有worker goroutine都调用了wg.Done()。
  3. 一旦wg.Wait()返回,意味着所有worker都已完成,monitorWorker随即调用close(cs)关闭通道。
  4. main函数中的for i := range cs循环会持续接收数据,直到cs通道被monitorWorker关闭。通道关闭后,for range循环会自动终止。
  5. main函数在循环结束后自然退出,程序正常终止,避免了死锁。

注意事项:

  • monitorWorker goroutine没有被添加到main函数的wg中,因为它不是一个需要main等待其完成的“工作”;它的职责是作为协调者,在worker完成后关闭通道。
  • 只有发送方才能关闭通道。 在本例中,虽然monitorWorker不是数据的直接发送方,但它通过wg.Wait()确保了所有发送方都已完成发送,因此它作为协调者来关闭通道是安全的。

解决方案二:多 Goroutine 协调与额外同步通道

如果业务需求坚持将数据接收和打印逻辑也放在一个独立的goroutine中(例如,printWorker),那么我们需要更复杂的同步机制来确保main函数在所有数据处理完毕后才退出。

package main

import (
    "fmt"
    "strconv"
    "sync"
)

func worker(wg *sync.WaitGroup, cs chan string, i int) {
    defer wg.Done()
    cs <- "worker" + strconv.Itoa(i)
}

// 职责同解决方案一:等待所有worker完成,然后关闭通道
func monitorWorker(wg *sync.WaitGroup, cs chan string) {
    wg.Wait()
    close(cs)
}

// 独立的打印goroutine,负责从cs接收并打印数据
func printWorker(cs <-chan string, done chan<- bool) {
    for i := range cs {
        fmt.Println(i)
    }
    // 当cs通道关闭且所有数据处理完毕后,向done通道发送信号
    done <- true
}

func main() {
    wg := &sync.WaitGroup{}
    cs := make(chan string)

    // 启动10个worker goroutine
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go worker(wg, cs, i)
    }

    // 启动monitorWorker来关闭cs通道
    go monitorWorker(wg, cs)

    // 创建一个用于printWorker通知main完成的通道
    done := make(chan bool, 1) // 使用带缓冲的通道,避免printWorker阻塞
    // 启动printWorker goroutine
    go printWorker(cs, done)

    // main goroutine等待printWorker完成的信号
    <-done // 阻塞直到从done通道接收到信号
}

方案分析:

  1. worker goroutine和monitorWorker goroutine的功能与解决方案一相同。monitorWorker在所有worker完成后关闭cs通道。
  2. printWorker goroutine负责从cs通道接收并打印数据。当cs通道被关闭,且所有已发送的数据都被printWorker消费完毕后,printWorker的for range循环会终止。
  3. printWorker在循环终止后,向done通道发送一个true值。
  4. main函数通过

关键点:

  • done := make(chan bool, 1):done通道被创建为带缓冲的(容量为1)。这确保了printWorker在发送true到done时不会因为main尚未准备好接收而阻塞,从而避免了潜在的死锁或竞态条件。
  • 这种模式在需要更细粒度的goroutine协调时非常有用,它允许不同的goroutine在完成各自任务后,通过通道向其他goroutine发出信号。

总结与最佳实践

解决Go并发编程中的死锁问题,特别是all goroutines are asleep - deadlock,核心在于对goroutine生命周期和通道状态的精确管理。

  1. 通道的关闭是关键: 当不再有数据发送到通道时,必须关闭该通道(close(channel)),以通知所有接收方for range循环可以安全退出。
  2. 谁来关闭通道: 只有发送方(或能确认所有发送方已完成的协调者)才能关闭通道。关闭一个已经关闭的通道会导致panic。从一个已关闭的通道接收数据会立即返回零值,并且第二个返回值(ok)为false。
  3. sync.WaitGroup 的作用: WaitGroup主要用于等待一组goroutine完成。在上述示例中,它被用来等待所有worker goroutine完成。
  4. 明确的同步信号: 当一个goroutine的完成需要通知另一个goroutine(尤其是main函数)时,使用一个额外的通道进行信号传递是一种健壮的模式。
  5. 避免无限等待: 仔细检查for range循环在通道上的使用。如果通道永远不会关闭,for range将导致无限期阻塞。

通过理解这些原则并应用上述解决方案,您可以有效地构建健壮、高效且无死锁的Go并发程序。

相关专题

更多
Go中Type关键字的用法
Go中Type关键字的用法

Go中Type关键字的用法有定义新的类型别名或者创建新的结构体类型。本专题为大家提供Go相关的文章、下载、课程内容,供大家免费下载体验。

233

2023.09.06

go怎么实现链表
go怎么实现链表

go通过定义一个节点结构体、定义一个链表结构体、定义一些方法来操作链表、实现一个方法来删除链表中的一个节点和实现一个方法来打印链表中的所有节点的方法实现链表。

442

2023.09.25

go语言编程软件有哪些
go语言编程软件有哪些

go语言编程软件有Go编译器、Go开发环境、Go包管理器、Go测试框架、Go文档生成器、Go代码质量工具和Go性能分析工具等。本专题为大家提供go语言相关的文章、下载、课程内容,供大家免费下载体验。

246

2023.10.13

0基础如何学go语言
0基础如何学go语言

0基础学习Go语言需要分阶段进行,从基础知识到实践项目,逐步深入。php中文网给大家带来了go语言相关的教程以及文章,欢迎大家前来学习。

691

2023.10.26

Go语言实现运算符重载有哪些方法
Go语言实现运算符重载有哪些方法

Go语言不支持运算符重载,但可以通过一些方法来模拟运算符重载的效果。使用函数重载来模拟运算符重载,可以为不同的类型定义不同的函数,以实现类似运算符重载的效果,通过函数重载,可以为不同的类型实现不同的操作。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

187

2024.02.23

Go语言中的运算符有哪些
Go语言中的运算符有哪些

Go语言中的运算符有:1、加法运算符;2、减法运算符;3、乘法运算符;4、除法运算符;5、取余运算符;6、比较运算符;7、位运算符;8、按位与运算符;9、按位或运算符;10、按位异或运算符等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

223

2024.02.23

go语言开发工具大全
go语言开发工具大全

本专题整合了go语言开发工具大全,想了解更多相关详细内容,请阅读下面的文章。

277

2025.06.11

go语言引用传递
go语言引用传递

本专题整合了go语言引用传递机制,想了解更多相关内容,请阅读专题下面的文章。

156

2025.06.26

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

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

7

2025.12.31

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Go 教程
Go 教程

共32课时 | 3.2万人学习

Go语言实战之 GraphQL
Go语言实战之 GraphQL

共10课时 | 0.8万人学习

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

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