golang context源码

2024-12-28 22:38:39 编程来源：ZONE.CI 全球网 0 阅读模式

在Go语言的标准库中，有一个非常重要的包——context。它提供了一种跨多个Goroutine传递请求相关数据、取消信号和超时的机制，是Go语言中用于处理并发的重要工具。

为什么需要Context？

在日常的开发中，我们经常会遇到需要在请求处理过程中传递参数、控制Goroutine超时或取消的情况。比如我们常见的HTTP请求、数据库连接等。如果我们希望对这些操作进行超时控制，或者在某些情况下主动停止这些操作，那么就需要一种机制来传递这些信息。

Context的基本使用

为了解决上述问题，Go语言的标准库中提供了context包。使用context可以从根节点开始向子节点传递请求的相关数据，并且在需要取消请求的情况下，可以及时通知所有相关的Goroutine停止处理。下面我们来看一个基本的例子：


func handleRequest(ctx context.Context) {
    // 模拟处理请求
    select {
    case <-ctx.done(): log.println("request="" canceled")="" return="" default:="" 执行具体的请求处理逻辑="" ...="" log.println("request="" handled")="" }="" }="" func="" main()="" {="" 创建一个可取消的context="" ctx,="" cancel="" :="context.WithCancel(context.Background())" defer="" cancel()="" 启动请求处理goroutine="" go="" handlerequest(ctx)="" 模拟请求取消="" time.sleep(1="" *="" time.second)="" cancel()="" time.sleep(1="" *="" time.second)="" }="">

在上述例子中，我们使用context包的WithCancel函数创建了一个可取消的Context，并且通过cancel函数通知所有使用这个Context的Goroutine停止处理。在handleRequest函数中，我们使用select语句来监听ctx.Done()的通道，一旦接收到停止信号，就停止当前的请求处理。

Context的嵌套使用

在真实的业务场景中，我们可能会遇到需要嵌套使用多个Context的情况。比如一个请求需要调用多个服务，每个服务都需要自己的Context来进行超时控制。下面我们来看一个嵌套使用Context的示例：


func serviceA(ctx context.Context) {
    // ...
}

func serviceB(ctx context.Context) {
    // ...
}

func serviceC(ctx context.Context) {
    // ...
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    ctxA, cancelA := context.WithTimeout(ctx, time.Second)
    defer cancelA()

    ctxB, cancelB := context.WithDeadline(ctx, time.Now().Add(2*time.Second))
    defer cancelB()

    ctxC, cancelC := context.WithCancel(ctx)
    defer cancelC()

    // 启动多个服务
    go serviceA(ctxA)
    go serviceB(ctxB)
    go serviceC(ctxC)

    // ...
}

在上述例子中，我们使用context包的WithTimeout、WithDeadline和WithCancel等函数创建了多个带有超时控制的Context。然后将这些Context传递给对应的服务进行处理。这样一来，每个服务都能够根据自己的需要进行超时控制，并且在需要停止时可以及时通知。