首頁 > 後端開發 > Golang > 主體

如何將 WaitGroup 傳遞給順序函數呼叫?

王林
發布: 2024-02-05 22:57:19
轉載
1008 人瀏覽過

如何将 WaitGroup 传递给顺序函数调用?

問題內容

我有一個可以在單獨的 goroutine 中順序或並發呼叫的函數。

我想確保函數在主 goroutine 完成之前完全執行,因此我將 *sync.waitgroup 參數傳遞給函數。現在,在某些地方該函數將被順序呼叫。

我可以將 nil waitgroup 傳遞給函數,如下所示:

func my_func(wg *sync.WaitGroup){
   if wg != nil{
   defer wg.Done()
   }
   // do the task
}

func main(){
my_func(nil) // sequential call

wg := sync.WaitGroup{}
wg.Add(1)
go my_func(&wg)  // concurrent call
wg.Wait()
}
登入後複製

有沒有更好的方法來達成這個目標?


正確答案


你的my_func() 不應該知道/不應該關心它是如何執行的(無論是否在新的goroutine中)。因此,為此您不應該傳遞 wg。不要強制並發或非並發使用您的 api,讓您的套件的用戶決定他們希望如何呼叫它。

如果有人希望同時運行它,在一個新的goroutine 中,可以在my_func()外部處理my_func(),如下所示:

wg.add(1)
go func() {
    defer wg.done()
    my_func()
}()
登入後複製

這也使得可以在將在 wg.done() 呼叫之前執行的函數呼叫之前/之後放置更多程式碼:

wg.add(1)
go func() {
    defer wg.done()

    // other code before
    my_func()
    // other code after
}()
登入後複製

另請注意,如果您在許多地方都有此功能,則可以建立一個輔助函數來負責 goroutine 啟動和 waitgroup 處理:

func launchmyfunc(wg *sync.waitgroup) {
    go func() {
        defer wg.done()

        my_func()
    }()
}
登入後複製

您也可以建立一個接受任意無參數無返回函數的幫助器:

func launchfunc(wg *sync.waitgroup, f func()) {
    go func() {
        defer wg.done()

        f()
    }()
}
登入後複製

使用上面的幫助器,您可以這樣做:

wg.Add(1)
launchMyFunc(wg)

// or
wg.Add(1)
launchFunc(wg, my_func)
登入後複製

以上是如何將 WaitGroup 傳遞給順序函數呼叫?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:stackoverflow.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板
關於我們 免責聲明 Sitemap
PHP中文網:公益線上PHP培訓,幫助PHP學習者快速成長!