前言
WaitGroup在go語言中，用于線程同步，單從字面意思理解，wait等待的意思，group組、團隊的意思，WaitGroup就是指等待一組，等待一個系列執(zhí)行完成后才會繼續(xù)向下執(zhí)行。Golang 中的 WaitGroup 一直是同步 goroutine 的推薦實踐。自己用了兩年多也沒遇到過什么問題。
直到最近的一天同事扔過來一段奇怪的代碼：
第一個坑
復(fù)制代碼 代碼如下:
package main
import (
"log"
"sync"
)
func main() {
wg := sync.WaitGroup{}
for i := 0; i lt; 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")
}
撇了一眼，覺得沒什么問題。
然而，它的運行結(jié)果是這樣：
復(fù)制代碼 代碼如下:
2016/11/27 15:12:36 exit
[Finished in 0.7s]
或這樣：
復(fù)制代碼 代碼如下:
2016/11/27 15:21:51 i:2
2016/11/27 15:21:51 exit
[Finished in 0.8s]
或這樣：
復(fù)制代碼 代碼如下:
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 的設(shè)計目的就非常容易 fix 啦。因為 WaitGroup 同步的是 goroutine, 而上面的代碼卻在 goroutine 中進行 Add(1) 操作。因此，可能在這些 goroutine 還沒來得及 Add(1) 已經(jīng)執(zhí)行 Wait 操作了。
于是代碼改成了這樣：
第二個坑
復(fù)制代碼 代碼如下:
package main
import (
"log"
"sync"
)
func main() {
wg := sync.WaitGroup{}
for i := 0; i lt; 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 又睡了過去，而且是睡死了過去：
復(fù)制代碼 代碼如下:
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 中，導(dǎo)致只有 Add 操作，其實 Done操作是在 wg 的副本執(zhí)行的。因此 Wait 就死鎖了。
于是代碼改成了這樣：
填坑
復(fù)制代碼 代碼如下:
package main
import (
"log"
"sync"
)
func main() {
wg := sync.WaitGroup{}
for i := 0; i lt; 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")
}
總結(jié)
好了，到這里終于解決了，以上就是關(guān)于Go語言WaitGroup使用時需要注意的一些坑，希望本文中提到的這些問題對大家學(xué)習(xí)或者使用Go語言的時候能有所幫助，如果有疑問大家可以留言交流。
您可能感興趣的文章:- Golang 實現(xiàn)分片讀取http超大文件流和并發(fā)控制
- Go 并發(fā)控制context實現(xiàn)原理剖析(小結(jié))
- Go并發(fā):使用sync.WaitGroup實現(xiàn)協(xié)程同步方式
- 解決Golang 中使用WaitGroup的那點坑
- Golang中的sync.WaitGroup用法實例
- Go并發(fā)控制WaitGroup的使用場景分析
