Go标准库Context

Context 在 Go 语言中主要用于控制 goroutine 的生命周期、传递请求的上下文信息等。

1. 创建 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()

2. 使用 Context

Context 可以用于控制 goroutine 的生命周期，并且可以传递信息。

例如，在 HTTP 服务中，可以将 Context 传递给处理请求的所有函数，以便在任何地方都可以访问请求的信息或者控制请求的生命周期。

func handler(w http.ResponseWriter, r *http.Request) {
    ctx := r.Context()
    // 使用 ctx 进行其他操作
}

3. 取消 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())
}

4. 存储额外数据

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 的基本使用方法。