Context 在 Go 语言中主要用于控制 goroutine 的生命周期、传递请求的上下文信息等。
- 创建 Context
Go 语言标准库提供了 context 包，其中包含了创建 Context 的函数。
context.Background(): 创建一个空的 Context，通常作为处理树的根节点。context.TODO(): 目前和 context.Background() 一样，但是未来可能会有不同的实现。context.WithCancel(parent Context) (ctx Context, cancel CancelFunc): 创建一个新的 Context，它通过调用 cancel 函数来取消它和它的子 Context。context.WithDeadline(parent Context, deadline time.Time) (Context, CancelFunc): 创建一个新的 Context，当指定的截止日期到达或者父 Context 被关闭时，它会自动取消。context.WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc): 创建一个新的 Context，当指定的超时时间到达或者父 Context 被关闭时，它会自动取消。context.WithValue(parent Context, key interface{}, val interface{}) Context: 创建一个新的 Context，它可以存储一些额外的数据。
例如：
ctx := context.Background()
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
- 使用 Context
Context 可以用于控制 goroutine 的生命周期，并且可以传递信息。
例如，在 HTTP 服务中，可以将 Context 传递给处理请求的所有函数，以便在任何地方都可以访问请求的信息或者控制请求的生命周期。
func handler(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
// 使用 ctx 进行其他操作
}
- 取消 Context
当 Context 被取消时，它会通知所有等待此 Context 的 goroutine。取消 Context 的方法有：
- 调用返回的
cancel 函数。 - 当父 Context 被取消时，它的所有子 Context 也会被取消。
- 当一个使用 Context 的操作在操作完成之前就返回了错误，表明操作已经不能继续进行。
例如：
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
select {
case <-time.After(15 * time.Second):
fmt.Println("timeout")
case <-ctx.Done():
fmt.Println(ctx.Err())
}
- 存储额外数据
context.WithValue 可以用来在 Context 中存储一些额外的数据。
例如：
type key string
var UserID key = "userID"
ctx := context.WithValue(context.Background(), UserID, "user-123")
userID := ctx.Value(UserID).(string)
fmt.Println(userID) // 输出: user-123
以上就是 Go 语言中 Context 的基本使用方法。