首頁  >  文章  >  後端開發  >  如何將 WaitGroup 傳遞給順序函數呼叫?

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

王林
王林轉載
2024-02-05 22:57:191041瀏覽

如何将 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刪除