在 Go 應用程式中,協調多個 Goroutine 的關閉對於確保它們乾淨、同步終止至關重要。本文探討了同步兩個 goroutine 以在任一例程發生錯誤時一起回傳的問題。
考慮以下程式碼片段:
func main() { go func() { ... if err != nil { return } }() go func() { ... if err != nil { return } }() }
此場景需要同步 goroutine這樣當其中一個遇到錯誤並返回時,另一個也應該終止。挑戰在於如何在不引入對封閉通道的寫入的情況下實現這一目標,否則可能會導致恐慌。
一個高效的解決方案涉及利用 Go 的上下文套件在 goroutine 之間進行通訊。透過建立上下文並將其傳遞給每個 goroutine,您可以提供一種機制來發出終止訊號。以下是示範此方法的更新程式碼片段:
package main import ( "context" "sync" ) func main() { ctx, cancel := context.WithCancel(context.Background()) wg := sync.WaitGroup{} wg.Add(3) go func() { defer wg.Done() for { select { // msg from other goroutine finish case <-ctx.Done(): // end } } }() go func() { defer wg.Done() for { select { // msg from other goroutine finish case <-ctx.Done(): // end } } }() go func() { defer wg.Done() // your operation // call cancel when this goroutine ends cancel() }() wg.Wait() }
在此程式碼中,上下文變數 ctx 充當 goroutine 之間的通訊通道。當任何一個 goroutine 中發生錯誤時,它會在上下文中呼叫 cancel(),這會向其他 goroutine 發出終止訊號。這個優雅的解決方案確保所有 goroutine 正常關閉,而不會出現恐慌的風險。
以上是Go中發生錯誤時如何優雅地關閉多個Goroutine?的詳細內容。更多資訊請關注PHP中文網其他相關文章!