Go中如何优雅处理已取消的Context？

问题背景 #

现有Go程序骨架如下，核心需求：

• 当从completionChannel读取到false（任务失败）时，调用cancel()终止所有运行中的goroutine
• func1和func2会重复执行任务（比如指数退避重试），需在Context被取消时立即停止迭代，避免内存泄漏

程序骨架代码：

ctx, cancel := context.WithCancel(context.Background())
defer cancel()

// 跟踪任务执行结果（成功/失败）
completionChannel := make(chan bool)

go func1(ctx)
go func2(ctx)

numberOfFunctions := 2
for i := 0; i < numberOfFunctions; i++ {
    // 异步接收任务执行结果
    result := <-completionChannel
    if !result {
        // 取消所有goroutine的后续处理
        cancel()
        return
    }
}

func1实现骨架（func2与它几乎完全相同）：

func func1(ctx context.Context) {
    retry := true
    for retry {
        // 此处需添加逻辑：若Context已取消，提前退出循环

        err := doThing()
        if err.Status == 404 {
            retry = false
        }
    }
}

补充提问 #

能否直接在注释位置添加如下非阻塞select来处理Context取消？

select {
case <-ctx.Done():
    fmt.Println("任务终止！")
    return
default:
}

解决方案 #

核心实现思路 #

实现goroutine优雅终止的核心是在循环的每次迭代中检查Context的取消状态，避免goroutine陷入无限制的重试循环。

推荐实现方式 #

你补充的非阻塞select方案完全可行，这也是Go中处理Context取消的标准做法之一。如果重试逻辑包含指数退避（失败后休眠再重试），还需要在休眠阶段也监听Context的取消信号，避免休眠期间无法响应取消。

修改后的func1示例：

import "time"

func func1(ctx context.Context) {
    retry := true
    backoff := 1 * time.Second // 初始退避时长
    for retry {
        // 非阻塞检查Context是否已取消
        select {
        case <-ctx.Done():
            fmt.Println("func1收到取消信号，终止执行")
            completionChannel <- false
            return
        default:
            // Context未取消，继续执行任务
        }

        err := doThing()
        if err != nil {
            if err.Status == 404 {
                retry = false
                completionChannel <- true
            } else {
                // 指数退避重试，同时监听取消信号
                select {
                case <-ctx.Done():
                    fmt.Println("func1在退避时收到取消信号，终止执行")
                    completionChannel <- false
                    return
                case <-time.After(backoff):
                    backoff *= 2
                    // 设置退避时长上限，避免过长
                    if backoff > 30*time.Second {
                        backoff = 30*time.Second
                    }
                }
            }
        } else {
            // 任务执行成功，终止重试
            retry = false
            completionChannel <- true
        }
    }
}

关键细节说明 #

1. 非阻塞检查：在每次任务执行前通过select+default非阻塞检查Context状态，确保任务执行前能响应取消信号
2. 退避阶段监听：如果有退避逻辑，必须在休眠时同时监听ctx.Done()，避免goroutine在休眠期间无法被取消
3. 结果通知：无论任务正常完成还是被取消，都要向completionChannel发送结果，避免主goroutine长期阻塞在通道读取上

内容的提问来源于stack exchange，提问作者user38643