Golang | 一文带你深入context

最新推荐文章于 2025-05-10 15:00:00 发布

灭BUG ing

最新推荐文章于 2025-05-10 15:00:00 发布

阅读量828

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/RoddeHope/article/details/118660460

本文详细介绍了Golang的Context接口及其重要结构体cancelCtx、timerCtx和valueCtx，强调了主动调用cancelFunc以避免上下文泄露的重要性。还探讨了Context中通道懒加载的原因、多次调用cancelFunc的影响、上下文截止时间的限制以及background和todo的使用区别。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言

首先解答上一篇文章一文带你快速入门context中留下的疑惑，为什么要defer cancelFunc()？

func main() {
    parent := context.Background()
    for i := 0; i < 100; i++ {
        go doRequest(parent)
    }
    time.Sleep(time.Second * 10)
}

// doRequest 模拟网络请求
func doRequest(parent context.Context) {
    ctx, _ := context.WithTimeout(parent, time.Second*5)
    time.Sleep(time.Millisecond * 200)
    go func() {
        <-ctx.Done()
        fmt.Println("ctx done!")
    }()
}

看上面的代码，在main函数中异步调用doRequest函数，doRequest函数中新建一个5s超时的上下文，doRequest函数的调用时长为200ms

可以看到，doRequest的上下文时间范围远大于函数调用花费的时间，在函数结束后没有主动取消上下文，这会造成上下文泄露

所以，defer cancelFunc()的目的是避免上下文泄露！！

主动调用cancelFunc是一个好习惯！

了解一下Context接口

type Context interface {
  // [1] 返回上下文的截止时间
    Deadline() (deadline time.Time, ok bool)
  // 返回一个通道，当上下文结束时，会关闭该通道，此时 <-ctx.Done() 结束阻塞
  Done() <-chan struct{}
  // [2] 该方法会在上下文结束时返回一个not nil err，该err用于表示上下文结束的原因
  Err() error
  // 返回与key关联的上下文的value
  Value(key interface{}) interface{}
}

[1]处，当上下文没有设置截止时间时，调用Deadline，返回结果值中，ok = false

func main() {
    ctx, cancelFunc := context.WithCancel(context.Background())
    defer cancelFunc()
    deadline, ok := ctx.Deadline()
    fmt.Printf("ok = %v, deadline = %v\n", ok, deadline)
    //