Go语言WaitGroup使用时需要注意的坑
更新时间：2016年12月06日 09:58:24 投稿：daisy
Go语言中WaitGroup的用途是它能够一直等到所有的goroutine执行完成，并且阻塞主线程的执行，直到所有的goroutine执行完成。之前一直使用也没有问题，但最近通过同事的一段代码引起了关于WaitGroup的注意，下面这篇文章就介绍了WaitGroup使用时需要注意的坑及填坑。
前言
WaitGroup在go语言中，用于线程同步，单从字面意思理解，wait等待的意思，group组、团队的意思，WaitGroup就是指等待一组，等待一个系列执行完成后才会继续向下执行。Golang 中的 WaitGroup 一直是同步 goroutine 的推荐实践。自己用了两年多也没遇到过什么问题。
直到最近的一天同事扔过来一段奇怪的代码：
第一个坑
复制代码 代码如下:
package main
import (
"log"
"sync"
)
func main() {
wg := sync.WaitGroup{}
for i := 0; i < 5; i++ {
go func(wg sync.WaitGroup, i int) {
wg.Add(1)
log.Printf("i:%d", i)
wg.Done()
}(wg, i)
}
wg.Wait()
log.Println("exit")
}
撇了一眼，觉得没什么问题。
然而，它的运行结果是这样：
复制代码 代码如下:
2016/11/27 15:12:36 exit
[Finished in 0.7s]
或这样：
复制代码 代码如下:
2016/11/27 15:21:51 i:2
2016/11/27 15:21:51 exit
[Finished in 0.8s]
或这样：
复制代码 代码如下:
2016/11/27 15:22:51 i:3
2016/11/27 15:22:51 i:2
2016/11/27 15:22:51 exit
[Finished in 0.8s]
一度让我以为手上的 mac 也没睡醒……
这个问题如果理解了 WaitGroup 的设计目的就非常容易 fix 啦。因为 WaitGroup 同步的是 goroutine, 而上面的代码却在 goroutine 中进行 Add(1) 操作。因此，可能在这些 goroutine 还没来得及 Add(1) 已经执行 Wait 操作了。
于是代码改成了这样：
第二个坑
复制代码 代码如下:
package main
import (
"log"
"sync"
)
func main() {
wg := sync.WaitGroup{}
for i := 0; i < 5; i++ {
wg.Add(1)
go func(wg sync.WaitGroup, i int) {
log.Printf("i:%d", i)
wg.Done()
}(wg, i)
}
wg.Wait()
log.Println("exit")
}
然而，mac 又睡了过去，而且是睡死了过去：
复制代码 代码如下:
2016/11/27 15:25:16 i:1
2016/11/27 15:25:16 i:2
2016/11/27 15:25:16 i:4
2016/11/27 15:25:16 i:0
2016/11/27 15:25:16 i:3
fatal error: all goroutines are asleep - deadlock!
wg 给拷贝传递到了 goroutine 中，导致只有 Add 操作，其实 Done操作是在 wg 的副本执行的。因此 Wait 就死锁了。
于是代码改成了这样：
填坑
复制代码 代码如下:
package main
import (
"log"
"sync"
)
func main() {
wg := &sync.WaitGroup{}
for i := 0; i < 5; i++ {
wg.Add(1)
go func(wg *sync.WaitGroup, i int) {
log.Printf("i:%d", i)
wg.Done()
}(wg, i)
}
wg.Wait()
log.Println("exit")
}
总结
好了，到这里终于解决了，以上就是关于Go语言WaitGroup使用时需要注意的一些坑，希望本文中提到的这些问题对大家学习或者使用Go语言的时候能有所帮助，如果有疑问大家可以留言交流。
相关文章
之前的文章，我们给大家分享了不少汉诺塔算法的实现语言，包括C、c++、java、python等，今天我们就来使用go语言来实现一下，需要的小伙伴来参考下吧。2015-03-03
这篇文章主要介绍了Go语言实现多协程文件下载器的相关资料,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2024-08-08
在 Go 语言编程中,未使用的导入和变量是常见问题,Go 语言的设计理念强调代码的简洁性和可读性,因此它强制要求所有导入的包和声明的变量都必须被使用,本文将深入解读相关内容,并结合实际代码示例和项目场景,帮助开发者更好地理解和处理未使用的导入和变量2025-06-06
这篇文章主要为大家介绍了Go modules replace解决Go依赖引用问题，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-06-06
结构体嵌入是Go语言中的一种特性,允许在一个结构体中包含另一个结构体,从而实现代码复用和方法复用,嵌入结构体的字段和方法会被提升到外部结构体中,可以直接访问,本文给大家介绍go语言中的结构体嵌入的相关知识,感兴趣的朋友跟随小编一起看看吧2025-12-12
这篇文章主要给大家介绍了关于Golang读写Excel的方法教程，golang操作excel利用的是Excelize，Excelize是Golang编写的一个用来操作 Office Excel 文档类库，基于微软的Office OpenXML标准，文中介绍的非常详细，需要的朋友可以参考借鉴，下面来一起看看吧。2017-05-05
RSA 是一种非对称加密算法,广泛使用于数据的安全传输,crypto/rsa 是 Golang 中实现了 RSA 算法的一个标准库,提供了生成公私钥对、加解密数据、签名和验签等功能,本文给大家介绍了在Golang中实现RSA算法的加解密操作,需要的朋友可以参考下2023-12-12
最近在工作中,总是重复的做事,想着自己设置一个快捷键实现windows 剪贴板的功能,所以本文小编给大家分享了使用golang在windows上设置全局快捷键的操作,文中有相关的代码示例供大家参考,需要的朋友可以参考下2024-02-02
go logger不侵入业务代码使用slog替换zap并实现callerSkip详解
这篇文章主要为大家介绍了go logger不侵入业务代码使用slog替换zap并实现callerSkip详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-09-09
Go 语言的内存分配机制是理解和优化 Go 程序性能的关键,在 Go 中,内存管理是自动进行的,这得益于 Go 的垃圾回收机制,了解内存如何分配和回收,可以帮助我们写出更高性能的代码,本文将深入讲解下 Go 内存分配机制,需要的朋友可以参考下2023-12-12
最新评论