如何將 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中文網其他相關文章！