你好,我是四哥。
前一篇文章从源码的角度详细介绍了 Context 的实现原理,但是还没有提到 Context 的使用场景,今天我们一起来看下:
1.请求链路传值。
传值使用方式如下:
func func1 ctx = context.WithValue func2func func2 fmt.Println.) ctx = context.WithValue(ctx, "k2", "v2") func3func func3 fmt.Println("func3:",ctx.Value("k1").) fmt.Println("func3:",ctx.Value("k2").)func main() ctx := context.Background() func1
我们在 func1 通过函数 WithValue 设置了一个键值对 k1—v1,在 func2 可以获取到 func1 设置的键值对,如果调用 func3 时把这个 ctx 继续传入的话,在 func3 中依然还是可以获取到 k1—v1。
但是在 func1 中获取不到 func2 设置的键值对 k2—v2,因为 context 只能自上而下携带值,这点需要注意。。
2.取消耗时操作,及时释放资源。
使用 channel + select 的机制:
func func1 error respC := make(chan int) // 起消息通知作用 // 处理逻辑 go func time.Sleep(time.Second * 3) // 模拟处理业务逻辑 respC close(respC) // 判断是否超时 select case r := lt,—respC: fmt.Printf("Resp: %d ", r) return nil case lt,—time.After(time.Second * 2): // 超过设置的时间就报错 fmt.Println("catch timeout") return errors.New("timeout") func main err := func1 fmt.Printf("func1 error: %v ", err)
上面的方式平时也会用到,通过 context 怎么实现呢。
下面来看下如何使用 context 进行主动取消,超时取消。
主动取消:
func func1 error defer wg.Done() respC := make(chan int) go func() time.Sleep(time.Second * 5) // 模拟业务逻辑处理 respC () // 取消机制 select case lt,—ctx.Done(): fmt.Println("cancel") return errors.New("cancel") case r := lt,—respC: fmt.Println(r) return nil func main() wg := amp,sync.WaitGroup ctx, cancel := context.WithCancel(context.Background()) wg.Add(1) go func1(ctx, wg) time.Sleep(time.Second * 2) cancel() // 主动取消 wg.Wait() // 等待 goroutine 退出
超时取消:
func func1 resp := make(chan int) go func() time.Sleep(time.Second * 5) // 模拟处理逻辑 resp () // 超时机制 select case lt,—ctx.Done(): fmt.Println("ctx timeout") fmt.Println(ctx.Err()) case lt,—resp: fmt.Println("done") returnfunc main() ctx, cancel := context.WithTimeout(context.Background(), time.Second*2) defer cancel() func13.防止 goroutine 泄露。iOS中的“快捷方式”应用程序现在可以在Mac上使用,而“焦点”通过减少干扰来帮助人们保持工作。
引自)
func gen ch := make(chan int) go func var n int for ch n++ time.Sleep(time.Second) return ch
这是一个可以生成无限整数的协程,但如果我只需要它产生的前 5 个数,那么就会发生 goroutine 泄漏:
func main for n := range gen fmt.Println(n) if n 5 break // hellip,hellip,
当 n 5 的时候,直接 break 掉那么 gen 函数的协程就会执行无限循环,永远不会停下来发生了 goroutine 泄漏
用 context 改进这个例子:
func gen ch := make(chan int) go func() var n int for select case lt,—ctx.Done(): return case ch n++ time.Sleep(time.Second) () return chfunc main() ctx, cancel := context.WithCancel(context.Background()) defer cancel() // 避免其他地方忘记 cancel,且重复调用不影响 for n := range gen fmt.Println(n) if n 5 cancel() break // hellip,hellip,
增加一个 context,在 break 前调用 cancel 函数,取消 goroutinegen 函数在接收到取消信号后,直接退出,系统回收资源
总结
这篇文章列出的几个例子是 context 最基本的使用场景,其他框架,第三包基本上都是从这几种用法扩展的,所以非常有必要掌握基础用法。Mac现在可以检测照片中的文字,或者在图像中提供动物,艺术,地标和植物的细节。
另外希望这篇文章能给你带来帮助,如果文中有理解错误之处或者你还想到其他用法,可以在留言区留言,一定回复!抱团学习不孤单!
参考资料
。