Go 语言中的 Context：掌控并发，优雅退出

2025-01-08 18:00

登链社区

2025-01-08 18:00

来源链接

订阅此专栏

收藏此文章

在 Go 语言的并发编程中，context 包扮演着至关重要的角色。它为 Goroutine 之间传递请求相关的数据、取消信号以及超时信息提供了标准化的方式，帮助我们更好地控制并发行为，构建健壮、可维护的应用程序。

Context 是什么？

Context 本质上是一个接口类型，定义了四个方法：

type Context interface {
    Deadline() (deadline time.Time, ok bool)
    Done() <-chan struct{}
    Err() error
    Value(key interface{}) interface{}
}

Deadline(): 返回 Context 的截止时间，如果没有设置截止时间，则 ok 为 false。
Done(): 返回一个只读的 channel，当 Context 被取消或超时时，该 channel 会被关闭。
Err(): 返回 Context 结束的原因，如果 Context 还未结束，则返回 nil。
Value(key): 返回与 key 关联的值，如果 key 不存在，则返回 nil。

为什么需要 Context？

在并发编程中，我们经常需要处理以下场景：

取消操作: 当用户取消请求或发生错误时，我们需要及时取消所有相关的 Goroutine，释放资源。
超时控制: 为操作设置时间限制，避免长时间阻塞。
传递数据: 在 Goroutine 之间传递请求相关的数据，例如用户信息、请求 ID 等。

Context 提供了一种统一的方式来处理这些场景，避免了在代码中手动传递取消信号、超时信息等，使得代码更加简洁、易读。

如何使用 Context？

Go 语言提供了几个函数来创建和使用 Context：

context.Background(): 返回一个空的 Context，通常作为根 Context 使用。
context.TODO(): 类似于 Background()，用于不确定使用哪个 Context 的场景。
context.WithCancel(parent): 基于父 Context 创建一个新的 Context，并返回一个取消函数，用于取消该 Context。
context.WithDeadline(parent, deadline): 基于父 Context 创建一个新的 Context，并设置一个截止时间。
context.WithTimeout(parent, timeout): 基于父 Context 创建一个新的 Context，并设置一个超时时间。
context.WithValue(parent, key, val): 基于父 Context 创建一个新的 Context，并存储一个键值对。

使用示例

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    // 创建一个带有超时的 Context
    ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
    defer cancel()

    // 启动一个 Goroutine 执行任务
    go func(ctx context.Context) {
        select {
        case <-time.After(time.Second * 3):
            fmt.Println("任务完成")
        case <-ctx.Done():
            fmt.Println("任务取消:", ctx.Err())
        }
    }(ctx)

    // 等待 Goroutine 结束
    time.Sleep(time.Second * 4)
}