Go context 包的作用是什么？常用场景有哪些？

知识点图片

Go 语言中的 context 包是并发编程中非常核心的一个组件。简单来说，它的主要作用是在Goroutine 之间传递上下文信息，用于控制 Goroutine 的生命周期和在调用链中传递数据。

以下是关于 context 包的详细解析：

一、Context 的核心作用

context 主要解决了三个问题：

取消信号（Cancellation）： 当一个任务被取消时（例如用户关闭了浏览器），通知所有为该任务服务的 Goroutine 停止工作，释放资源。
超时控制（Timeout/Deadline）： 设定一个任务的最长执行时间，如果超时，自动发送取消信号。
数据传递（Request-scoped values）： 在调用链中传递与请求相关的元数据（如 UserID、TraceID、认证 Token 等），而不需要修改函数签名来逐层传递。

二、核心机制：树状结构

Context 是层级化的。

通常从 context.Background()（根节点）开始。
通过 WithCancel、WithTimeout、WithValue 等函数派生出子 Context。
关键特性： 当父 Context 被取消或超时，所有基于它派生的子 Context 也会自动被取消。这就像砍断树枝，上面的叶子都会掉下来。

三、常用场景

1. HTTP 请求的超时与取消控制（最常见）

在 Web 开发中，一个 HTTP 请求通常会触发后续的一系列操作（查数据库、调用 RPC、计算等）。

场景： 用户请求一个 API，但后端处理太慢，用户不耐烦关闭了页面，或者客户端设置了 2 秒超时。
作用： http.Request 自带 Context。当连接断开，Context 会变为 Done 状态。后端的所有操作（如 SQL 查询）如果监听了这个 Context，就会立即停止，避免浪费服务器资源计算一个没人接收的结果。

func handler(w http.ResponseWriter, r *http.Request) {
    // 获取请求的 context
    ctx := r.Context()

    select {
    case <-time.After(5 * time.Second): // 模拟耗时操作
        fmt.Fprintln(w, "Hello")
    case <-ctx.Done(): // 监听取消信号
        err := ctx.Err()
        fmt.Println("Request canceled:", err)
        return
    }
}

2. 数据库查询与 RPC 调用

现代的 Go 数据库驱动（如 database/sql）和 RPC 框架（如 gRPC）都原生支持 Context。

场景： 执行一个复杂的 SQL 查询，要求必须在 500ms 内返回，否则视为失败。
代码示例：

ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
defer cancel() // 这是一个好习惯，确保资源释放

// 如果 500ms 没查完，db.QueryContext 会自动中断并返回 error
rows, err := db.QueryContext(ctx, "SELECT * FROM huge_table")
if err != nil {
    // 处理超时错误
}

3. 防止 Goroutine 泄漏

在 Go 中，启动 Goroutine 很容易，但确保它们退出很难。如果父 Goroutine 退出了，但子 Goroutine 还在阻塞等待（例如读一个永远没数据的 channel），就会造成内存泄漏。

场景： 启动一个后台 worker 监控某些指标。
作用： 通过 Context 通知后台 worker 退出循环。

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            return // 收到取消信号，安全退出
        default:
            // 执行工作...
        }
    }
}

4. 全链路追踪（Distributed Tracing）

在微服务架构中，一个请求可能经过 A -> B -> C 三个服务。

场景： 需要知道这三个服务处理同一个请求分别花了多少时间，或者需要统一记录日志。
作用： 使用 context.WithValue 将 TraceID 或 RequestID 注入 Context，并在服务间透传。

// 中间件中注入
ctx := context.WithValue(r.Context(), "trace_id", "12345")
next.ServeHTTP(w, r.WithContext(ctx))

// 后续函数中获取
traceID := ctx.Value("trace_id").(string)

四、Context 的最佳实践（避坑指南）

作为第一个参数： 按照 Go 的惯例，Context 应该作为函数的第一个参数，通常命名为 ctx。
- func DoSomething(ctx context.Context, arg string) error
不要放在结构体中： Context 应该是随函数调用传递的，而不是作为 struct 的成员保存（除非是极特殊的底层框架设计）。
不要传递 nil： 如果你不确定用什么 Context，请使用 context.TODO()，而不是 nil。
WithValue 仅用于元数据： 不要用 WithValue 传递核心业务参数（如函数的可选参数）。它应该只用于“请求域”的数据（如 LogID、Token）。
Context 是不可变的： 每次修改（如添加超时、添加值）都会返回一个新的 Context 对象，原对象不变。
及时 Cancel： 使用 WithTimeout 或 WithCancel 时，通常配合 defer cancel() 使用，确保函数退出时释放 Context 相关的资源（计时器等）。

总结

Go context 包是 Go 并发模式的基石。 只要涉及到网络请求、数据库操作或多 Goroutine 协作，几乎必然会用到它。它让开发者能够优雅地控制“停止”这件事，并让请求上下文数据在复杂的系统中流动。

右滑查看面试常问

播面