在golang中使用WaitGroup時需要注意哪些事項

在golang中使用WaitGroup時需要注意哪些事項？相信很多沒有經(jīng)驗的人對此束手無策，為此本文總結了問題出現(xiàn)的原因和解決方法，通過這篇文章希望你能解決這個問題。

前言

WaitGroup在go語言中，用于線程同步，單從字面意思理解，wait等待的意思，group組、團隊的意思，WaitGroup就是指等待一組，等待一個系列執(zhí)行完成后才會繼續(xù)向下執(zhí)行。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) 已經(jīng)執(zhí)行 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 的副本執(zhí)行的。因此 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")
}

看完上述內(nèi)容，你們掌握在golang中使用WaitGroup時需要注意哪些事項的方法了嗎？如果還想學到更多技能或想了解更多相關內(nèi)容，歡迎關注億速云行業(yè)資訊頻道，感謝各位的閱讀！