Go中發生錯誤時如何優雅地關閉多個Goroutine？

Go 中多個 Goroutine 的優雅關閉

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