在 Go 中如何优雅地同时杀死多个 Goroutine？

Go 中如何一次性终止多个 Goroutine

在 Go 中，你可能会遇到需要同时终止多个 Goroutine 的场景。当一个 goroutine 的完成应该触发其他 goroutine 的终止时，这尤其有用。

考虑以下代码片段：

func main() {

  go func() {
    ...
    if err != nil {
      return
    }
  }()

  go func() {
    ...
    if err != nil {
      return
    }
  }()

}

在这种情况下，您要确保当其中一个 goroutine 完成时返回时，对方也应该退出。实现这一点的常见方法是使用信道进行信令。然而，这可能会导致写入关闭通道恐慌。

解决此问题的可靠方法是利用 Go 的上下文机制。上下文允许您在例程之间建立通信，并在操作完成或取消时发出信号。下面是一个示例：

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()
}

在这个示例中，我们创建一个可以传递给 goroutine 来检查终止信号的上下文。当第三个 goroutine 完成其操作时，它会调用 cancel()，这会向等待其完成的两个 goroutine 发送信号。结果，所有 goroutine 都会正常终止。

以上是在 Go 中如何优雅地同时杀死多个 Goroutine？的详细内容。更多信息请关注PHP中文网其他相关文章！