编写HTTP Handler时是否需监听请求上下文取消？如何终止后台任务？

一、Handler在请求上下文取消时的退出方式 #

不需要“必须”设置监听器，但监听r.Context().Done()是最可靠的常规方式，能让你在请求取消（比如客户端主动断开、请求超时）时立即终止Handler逻辑，避免不必要的资源消耗。

除此之外，如果你调用的第三方库或标准库函数本身支持Context（比如http.Do、db.QueryContext），这些函数会自动监听Context取消并返回错误，Handler可以通过判断这些错误自然退出，不需要额外写select监听。但如果是你自己实现的阻塞逻辑，还是得手动监听Context。

二、为什么doSomething在上下文取消后还在后台运行？ #

你的代码存在两个核心问题：

1. time.Sleep是不可中断的阻塞操作，它不会感知Context的取消信号，哪怕Context已经Done，sleep还是会执行完整个时长。
2. runInContext里虽然监听了Context，但它只是启动goroutine执行fn()，然后在自己的select里等待结果——但这个goroutine里的time.Sleep并不会被中断，所以即使runInContext返回了Context错误，执行sleep的goroutine依然会在后台跑完。

三、修复方案 #

要让后台逻辑响应Context取消，需要把不可中断的阻塞操作替换为能感知Context的逻辑：

1. 实现支持Context的可中断sleep

替换原生time.Sleep：

func sleepWithContext(ctx context.Context, d time.Duration) error {
    select {
    case <-time.After(d):
        return nil
    case <-ctx.Done():
        return ctx.Err()
    }
}

2. 修改业务逻辑，让操作响应Context取消

调整doSomething和相关函数：

func doSomething(ctx context.Context) error {
    log.Println("doing something")
    defer log.Println("done doing something")

    // 用可中断sleep替代原生time.Sleep
    return sleepWithContext(ctx, 10*time.Second)
}

3. 优化Handler逻辑，确保后台goroutine及时终止

func handleSomething(w http.ResponseWriter, r *http.Request) {
    done := make(chan error)

    go func() {
        // 先检查Context是否已取消，再执行任务
        select {
        case <-r.Context().Done():
            done <- r.Context().Err()
            return
        default:
            done <- doSomething(r.Context())
        }
    }()

    select {
    case <-r.Context().Done():
        log.Println("context is done!")
        http.Error(w, r.Context().Err().Error(), http.StatusRequestTimeout)
        return
    case err := <-done:
        if err != nil {
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }
        w.WriteHeader(http.StatusOK)
        w.Write([]byte("ok"))
    }
}

关键要点总结 #

• Handler退出不代表后台goroutine会自动停止，必须让后台逻辑主动监听Context取消信号。
• 避免使用不可中断的阻塞操作（如time.Sleep），改用能结合Context的替代方案。
• 对于长时间运行的任务，要在任务的关键节点检查Context是否已取消，及时终止操作。

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