如何使用上下文优雅地终止多个 Go 例程？-Golang-PHP中文网

如何使用上下文优雅地终止多个 Go 例程？

Susan Sarandon

发布： 2024-12-10 05:48:14

原创

173 人浏览过

How to Gracefully Terminate Multiple Go Routines Using Context?

如何使用 Context 同步多个 Go 例程

为了同步多个 goroutine，允许它们在其中一个返回时终止，context 提供了一个有效的

说明：

示例代码创建了两个 goroutine。为了同步它们，将启动 context.Context 并将其提供给两个 goroutine。每个 goroutine 都会进入一个 select {} 循环，监听来自上下文的消息。

当发生错误或满足特定条件时：

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

登录后复制

使用优点上下文：