Go context.AfterFunc 适合怎样触发资源回收

我在给一个会被请求取消的后台连接加清理逻辑时，最容易误判的是：context.AfterFunc 看起来像“取消后执行 defer”，其实它会把回调放到独立 goroutine 里运行。它很适合做取消驱动的兜底回收，但不能替代正常路径上的显式关闭，也不能把返回的 stop 当成“清理已经完成”的证明。

把 AfterFunc 用在“上下文取消就要触发”的资源回收上；把 stop() 用来解除尚未启动的回调；当它返回 false 时，再通过完成信号等待回调真正结束。

先判断 AfterFunc 是否适合资源回收

这个 API 的边界很清楚：它接收一个上下文和无参数函数，返回一个停止关联的函数。取消发生后，回调在自己的 goroutine 中执行；多次注册也互不替换。这样的语义适合“请求取消时关闭读写、唤醒等待、撤销临时占用”一类动作，尤其适合作为漏网路径的兜底。

但正常完成仍应由主流程主动收尾。例如读取连接成功后，不要只等 context 取消才关闭连接；否则资源会一直活到超时。更稳的分工是：正常路径先调用 stop() 尝试撤销回调，取消路径则让回调负责关闭；两条路径共享同一个幂等清理函数。

用 stop 函数处理正常完成与取消竞态

关键竞态发生在 stop() 返回 false 的瞬间：回调可能已经开始，也可能早就被停止。官方语义并不保证它等待回调完成，所以如果主流程接下来要复用或销毁同一资源，必须让回调主动发出完成信号。

package main

import (
    "context"
    "fmt"
    "sync"
)

// closeResource 代表关闭连接、删除临时文件等一次性资源动作。
func closeResource() { fmt.Println("resource closed") }

func useResource(ctx context.Context) {
    var once sync.Once
    done := make(chan struct{})

    cleanup := func() {
        // Once 防止正常路径和取消回调同时触发两次真实关闭。
        once.Do(func() { closeResource() })
        // 关闭信号只发送一次，等待方才能确认回调已返回。
        close(done)
    }

    stop := context.AfterFunc(ctx, cleanup)

    // 这里模拟业务完成；真实代码应在读写结束后进入该分支。
    if stop() {
        // true 说明回调尚未启动，正常路径可以直接完成资源收尾。
        cleanup()
        return
    }

    // false 不代表 cleanup 已经结束，必须等待回调发出完成信号。
    

上例把“是否触发”与“是否完成”拆成两个问题。正常完成时 stop() 返回 true，先解除回调再执行一次清理；如果返回 false，说明取消与正常完成发生了竞争，等待 done 才能安全使用后续资源。实际工程里，如果资源的 Close 本身可能阻塞，还应把超时和错误记录放进清理函数，而不是静默丢掉。

让清理动作具备幂等性

sync.Once 适合“只允许一次实际副作用”的关闭动作。如果资源库明确保证并发调用 Close 安全，也可以直接复用它的约束；如果关闭失败还需要重试，就不要用 Once 把失败永久吞掉，而应把状态机和错误传递设计清楚。

这些误用最容易让资源回收失效

• 把 AfterFunc 当成同步回调，取消后立刻关闭或复用资源，却没有等待回调结束。
• 只写 defer stop()，却没有在正常路径显式释放资源，导致回收依赖取消或超时。
• 回调里直接修改主 goroutine 共享状态，没有锁、原子变量或明确的完成协议。
• 用 stop(false) 判断“清理失败”，忽略了它只表达启动状态。

我现在会先画出资源的两个出口：正常完成和 context 取消，再决定是否需要 AfterFunc。只要两条出口都调用同一个幂等清理函数，并且取消竞争时有完成信号，这个 API 就能承担好兜底回收的角色。

常见问题

context 已经取消，再调用 AfterFunc 会怎样？

回调会在自己的 goroutine 中尽快执行，不会变成当前调用栈里的同步执行。

stop 返回 false 后还需要等待吗？

如果后续操作依赖清理完成，就需要用 channel、WaitGroup 或其他协议等待；stop 本身不等待。

AfterFunc 能替代 defer Close 吗？

不能。defer 负责函数正常返回时的确定性收尾，AfterFunc 适合补上取消、超时等异步退出路径。