Context: Go中的上下文信息处理利器

后端

2023-02-24 13:10:45

Context：轻松掌控并发编程的上下文

什么是Context？

Context 是 Go 语言中定义上下文信息的接口，它提供四种主要方法：

Deadline()： 设置或获取请求的截止时间。
Done()： 返回一个 channel，当 Context 被取消时关闭。
Err()： 如果 Context 被取消，返回错误信息，否则返回 nil。
Value(key interface{}) interface{}： 从 Context 中获取指定键的值。

Context 的妙用

Context 可以在多种场景下发挥作用：

函数参数： 传递 Context 作为函数参数，使函数可以访问上下文信息。
Goroutine： 在创建 Goroutine 时传递 Context，让 Goroutine 能够感知上下文的取消。
Channel： 将 Context 发送到 channel 中，以便其他 Goroutine 接收和使用。

示例代码

以下是使用 Context 的示例代码：

package main

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

func main() {
    // 设置截止时间为 1 秒
    ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
    defer cancel()

    // 创建一个等待组，等待所有 Goroutine 完成
    var wg sync.WaitGroup

    // 启动 10 个 Goroutine，每个 Goroutine 都使用 Context 来计时
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            select {
            case <-ctx.Done():
                fmt.Printf("Goroutine %d canceled.\n", i)
            case <-time.After(2 * time.Second):
                fmt.Printf("Goroutine %d finished.\n", i)
            }
        }(i)
    }

    // 等待所有 Goroutine 完成
    wg.Wait()
}