目錄
考一考
陷阱分析
正確的做法
首頁 後端開發 Golang 分析Golang的WaitGroup陷阱並解決問題

分析Golang的WaitGroup陷阱並解決問題

Sep 14, 2021 pm 03:57 PM
golang goroutine wait

本文由go語言教學專欄為大家介紹關於Golang的WaitGroup陷阱,希望對需要的朋友有幫助!

sync.WaitGroup是並發環境中,相當常用的資料結構,用來等待所有協程的結束,寫程式碼的時候都是照著例子的樣子寫的,也沒用深究過它的使用。前幾日想著能不能在協程中執行Add()函數,答案是不能,這裡介紹下。

陷阱在WaitGroup的3個函數的呼叫順序上。先回顧下3個函數的功能:

  1. Add(delta int):增加計數器delta,例如啟動1個協程就增加1。
  2. Done():協程退出前執行,把計數器減1。
  3. Wait():阻塞等待計數器為0。

考一考

下面的程式是建立了協程father,然後father協程建立了10個子協程,main函數等待所有協程結束後退出,看看下面程式碼有沒有什麼問題?

package main

import (
    "fmt"
    "sync"
)

func father(wg *sync.WaitGroup) {
    wg.Add(1)
    defer wg.Done()

    fmt.Printf("father\n")
    for i := 0; i < 10; i++ {
        go child(wg, i)
    }
}

func child(wg *sync.WaitGroup, id int) {
    wg.Add(1)
    defer wg.Done()

    fmt.Printf("child [%d]\n", id)
}

func main() {
    var wg sync.WaitGroup
    go father(&wg)

    wg.Wait()
    log.Printf("main: father and all chindren exit")
}
登入後複製

發現問題了嗎?如果沒有看下面的運行結果:main函數在子協程結束前就開始結束了。

father
main: father and all chindren exit
child [9]
child [0]
child [4]
child [7]
child [8]
登入後複製

陷阱分析

產生上述問題的原因在於,建立協程後在協程內才執行Add()函數,而此時Wait ()函數可能已經在執行,甚至Wait()函數在所有Add()執行前就執行了,Wait( )執行時立刻就滿足了WaitGroup的計數器為0,Wait結束,主程式退出,導致所有子協程還沒完全退出,main函數就結束了。

正確的做法

Add函數一定要在Wait函數執行前執行,這在Add函數的文件中就提示了: Note that calls with a positive delta that occur when the counter is zero must happen before a Wait.

如何確保Add函數一定在Wait函數前執行?在協程情況下,我們無法預測協程中程式碼執行的時間是否早於Wait函數的執行時間,但是,我們可以在分配協程前就執行Add函數,然後再執行Wait函數,以此確保。

以下是修改後的程序,以及輸出結果。

package main

import (
    "fmt"
    "sync"
)

func father(wg *sync.WaitGroup) {
    defer wg.Done()

    fmt.Printf("father\n")
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go child(wg, i)
    }
}

func child(wg *sync.WaitGroup, id int) {
    defer wg.Done()
    fmt.Printf("child [%d]\n", id)
}

func main() {
    var wg sync.WaitGroup
    wg.Add(1)
    go father(&wg)

    wg.Wait()
    fmt.Println("main: father and all chindren exit")
}
登入後複製
father
child [9]
child [7]
child [8]
child [1]
child [4]
child [5]
child [2]
child [6]
child [0]
child [3]
main: father and all chindren exit
登入後複製

以上是分析Golang的WaitGroup陷阱並解決問題的詳細內容。更多資訊請關注PHP中文網其他相關文章!

本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn

熱AI工具

Undresser.AI Undress

Undresser.AI Undress

人工智慧驅動的應用程序,用於創建逼真的裸體照片

AI Clothes Remover

AI Clothes Remover

用於從照片中去除衣服的線上人工智慧工具。

Undress AI Tool

Undress AI Tool

免費脫衣圖片

Clothoff.io

Clothoff.io

AI脫衣器

AI Hentai Generator

AI Hentai Generator

免費產生 AI 無盡。

熱門文章

R.E.P.O.能量晶體解釋及其做什麼(黃色晶體)
2 週前 By 尊渡假赌尊渡假赌尊渡假赌
倉庫:如何復興隊友
4 週前 By 尊渡假赌尊渡假赌尊渡假赌
Hello Kitty Island冒險:如何獲得巨型種子
3 週前 By 尊渡假赌尊渡假赌尊渡假赌

熱工具

記事本++7.3.1

記事本++7.3.1

好用且免費的程式碼編輯器

SublimeText3漢化版

SublimeText3漢化版

中文版,非常好用

禪工作室 13.0.1

禪工作室 13.0.1

強大的PHP整合開發環境

Dreamweaver CS6

Dreamweaver CS6

視覺化網頁開發工具

SublimeText3 Mac版

SublimeText3 Mac版

神級程式碼編輯軟體(SublimeText3)

如何使用 Golang 安全地讀取和寫入檔案? 如何使用 Golang 安全地讀取和寫入檔案? Jun 06, 2024 pm 05:14 PM

如何使用 Golang 安全地讀取和寫入檔案?

如何為 Golang 資料庫連線配置連線池? 如何為 Golang 資料庫連線配置連線池? Jun 06, 2024 am 11:21 AM

如何為 Golang 資料庫連線配置連線池?

Golang 與 C++ 的異同 Golang 與 C++ 的異同 Jun 05, 2024 pm 06:12 PM

Golang 與 C++ 的異同

golang框架架構的學習曲線有多陡峭? golang框架架構的學習曲線有多陡峭? Jun 05, 2024 pm 06:59 PM

golang框架架構的學習曲線有多陡峭?

如何在 Golang 中從列表中產生隨機元素? 如何在 Golang 中從列表中產生隨機元素? Jun 05, 2024 pm 04:28 PM

如何在 Golang 中從列表中產生隨機元素?

golang框架的優缺點比較 golang框架的優缺點比較 Jun 05, 2024 pm 09:32 PM

golang框架的優缺點比較

Golang 框架中的錯誤處理最佳實務有哪些? Golang 框架中的錯誤處理最佳實務有哪些? Jun 05, 2024 pm 10:39 PM

Golang 框架中的錯誤處理最佳實務有哪些?

golang框架文件使用說明 golang框架文件使用說明 Jun 05, 2024 pm 06:04 PM

golang框架文件使用說明

See all articles