我有一個可以在單獨的 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中文網其他相關文章!