文章目录

  • Golang Context 详细原理和使用技巧
    • Context 背景 和 适用场景
      • Context 的背景
      • Context 的功能和目的
      • Context 的基本使用
      • Context 的同步控制设计
    • Context 的定义和实现
      • Context interface 接口定义
      • parent Context 的具体实现
      • Context 的继承和各种 With 系列函数
    • Context 的常用方法实例
      • 1. 调用 Context Done方法取消
      • 2. 通过 context.WithValue 来传值
      • 3. 超时取消 context.WithTimeout
      • 4. 截止时间取消 context.WithDeadline
    • Context 使用原则 和 技巧
    • 最后

Golang Context 详细原理和使用技巧

Context 背景 和 适用场景

Context 的背景

Golang 在 1.6.2 的时候还没有自己的 context,在1.7的版本中就把 golang.org/x/net/context包被加入到了官方的库中。Golang 的 Context 包,中文可以称之为“上下文”,是用来在 goroutine 协程之间进行上下文信息传递的,这些上下文信息包括 kv 数据、取消信号、超时时间、截止时间等。

Context 的功能和目的

虽然我们知道了 context 上下文的基本信息,但是想想,为何 Go 里面把 Context 单独拧出来设计呢?这就和 Go 的并发有比较大的关系,因为 Go 里面创建并发协程非常容易,但是,如果没有相关的机制去控制这些这些协程的生命周期,那么可能导致协程泛滥,也可能导致请求大量超时,协程无法退出导致协程泄漏、协程泄漏导致协程占用的资源无法释放,从而导致资源被占满等各种问题。所以,context 出现的目的就是为了解决并发协程之间父子进程的退出控制。

一个常见例子,有一个 web 服务器,来一个请求,开多个协程去处理这个请求的业务逻辑,比如,查询登录状态、获取用户信息、获取业务信息等,那么如果请求的下游协程的生命周期无法控制,那么我们的业务请求就可能会一直超时,业务服务可能会因为协程没有释放导致协程泄漏。因此,协程之间能够进行事件通知并且能控制协程的生命周期非常重要,怎么实现呢? context 就是来干这些事的。

另外,既然有大量并发协程,那么各个协程之间的一些基础数据如果想要共享,比如把每个请求链路的 tarceID 都进行传递,这样把整个链路串起来,要怎么做呢? 还是要依靠 context。

总体来说,context 的目的主要包括两个:

  1. 协程之间的事件通知(超时、取消)
  2. 协程之间的数据传递键值对的数据(kv 数据)

Context 的基本使用

Go 语言中的 Context 直接使用官方的 "context"包就可以开始使用了,一般是在我们所有要传递的地方(函数的第一个参数)把 context.Context 类型的变量传递,并对其进行相关 API 的使用。context 常用的使用姿势包括但不限于:

  1. 通过 context 进行数据传递,但是这里只能传递一些通用或者基础的元数据,不要传递业务层面的数据,不是说不可以传递,是在 Go 的编码规范或者惯用法中不提倡
  2. 通过 context 进行协程的超时控制
  3. 通过 context 进行并发控制

Context 的同步控制设计

Go 里面控制并发有两种经典的方式,一种是 WaitGroup,另外一种就是 Context。

在 Go 里面,当需要进行多批次的计算任务同步,或者需要一对多的协作流程的时候;通过 Context 的关联关系(go 的 context 被设计为包含了父子关系),我们就可以控制子协程的生命周期,而其他的同步方式是无法控制其生命周期的,只能是被动阻塞等待完成或者结束。context 控制子协程的生命周期,是通过 context 的 context.WithTimeout 机制来实现的,这个是一般系统中或者底层各种框架、库的普适用法。context 对并发做一些控制包括 Context Done 取消、截止时间取消 context.WithDeadline、超时取消 context.WithTimeout 等。

比如有一个网络请求 Request,每个 Request 都需要开启一个 goroutine 做一些业务逻辑,这些 goroutine 又可能会开启其他的 goroutine。那么这样的话,我们就可以通过 Context 来跟踪并控制这些 goroutine。

另外一个实际例子是,在 Go 实现的 web server 中,每个请求都会开一个 goroutine 去处理。但是我们的这个 goroutine 请求逻辑里面, 还需继续创建goroutine 去访问后端其他资源,比如数据库、RPC 服务等。由于这些 goroutine 都是在处理同一个请求,因此,如果请求超时或者被取消后,所有的 goroutine 都应该马上退出并且释放相关的资源,这种情况也需要用 Context 来为我们取消掉所有 goroutine。

请允许我打个小广告:这篇文章首发在我微信公众号【后端系统和架构】中,点击这里可以去往公众号查看原文链接,如果对你有帮助,欢迎前往关注,更加方便快捷的接收最新优质文章

Context 的定义和实现

Context interface 接口定义

在 golang 里面,interface 是一个使用非常广泛的结构,它可以接纳任何类型。而 context 就是通过 interface 来定义的,定义很简单,一共4个方法,这也是 Go 的设计理念,接口尽量简单、小巧,通过组合来实现丰富的功能。

定义如下:

type Context interface {//  返回 context 是否会被取消以及自动取消的截止时间(即 deadline)Deadline() (deadline time.Time, ok bool)// 当 context 被取消或者到了 deadline,返回一个被关闭的 channelDone() <-chan struct{}// 返回取消的错误原因,因为什么 Context 被取消Err() error// 获取 key 对应的 valueValue(key interface{}) interface{}
}
  1. Deadline 返回 context 是否会被取消以及自动取消的截止时间,第一个返回值是截止时间,到了这个时间点,Context 会自动发起取消请求;第二个返回值 ok==false 时表示没有设置截止时间,如果需要取消的话,需要调用取消函数进行取消。

  2. Done 方法返回一个只读的 chan,类型为 struct{},如果该方法返回的 chan 可以读取,那么就说明 parent context 已经发起了取消请求,当我们通过 Done 方法收到这个信号后,就应该做清理操作,然后退出 goroutine,释放资源。之后,Err 方法会返回一个错误,告知为什么 Context 被取消。

  3. Err 方法返回取消的错误原因,因为什么 Context 被取消。

  4. Value 方法获取该 Context 上保存的键值对,所以要通过一个 Key 才可以获取对应的值,这个值一般是线程安全(并发安全)的。虽然 context 是一个并发安全的类型,但是如果 context 中保存着 value,则这些 value 通常不是并发安全的,并发读写这些 value 可能会造成数据错乱,严重的情况下可能发生 panic,所以在并发时,如果我们的业务代码需要读写 context 中的 value,那么最好建议我们 clone 一份原来的 context 中的 value,并塞到新的 ctx 传递给各个gorouinte。当然, 如果已经明确不会有并发读取,那么可以直接使用,或者使用的时候加锁。

parent Context 的具体实现

Context 虽然是个接口,但是并不需要使用方实现,golang 内置的 context 包,已经帮我们实现了,查看 Go 的源码可以看到如下定义:

var (background = new(emptyCtx)todo = new(emptyCtx)
)func Background() Context {return background
}func TODO() Context {return todo
}

Background 和 TODO 两个其实都是基于 emptyCtx 来实现的,emptyCtx 类型实现了 context 接口定义的 4 个方法,它本身是一个不可取消,没有设置截止时间,没有携带任何值的 Context,查看官方源码如下:

type emptyCtx intfunc (*emptyCtx) Deadline() (deadline time.Time, ok bool) {return
}func (*emptyCtx) Done() <-chan struct{} {return nil
}func (*emptyCtx) Err() error {return nil
}func (*emptyCtx) Value(key interface{}) interface{} {return nil
}

Background 方法,一般是在 main 函数的入口处(或者请求最初的根 context)就定义并使用,然后一直往下传递,接下来所有的子协程里面都是基于 main 的 context 来衍生的。TODO 这个一般不建议业务上使用,一般没有实际意义,在单元测试里面可以使用。

Context 的继承和各种 With 系列函数

查看官方文档 https://pkg.go.dev/golang.org/x/net/context,看到有如下函数:

// 最基础的实现,也可以叫做父 context
func Background() Context
func TODO() Context// 在 Background() 根 context 基础上派生的各种  With 系列函数
func WithCancel(parent Context) (ctx Context, cancel CancelFunc)
func WithDeadline(parent Context, deadline time.Time) (Context, CancelFunc)
func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)
func WithValue(parent Context, key interface{}, val interface{}) Context
  • WithCancel 函数,传递一个 parent Context 作为参数,返回子 Context,以及一个取消函数用来取消 Context。我们前面说到控制父子协程的生命周期,那么就可以通过这个函数来实现

  • WithDeadline 函数,和 WithCancel 差不多,但是它会多传递一个截止时间参数,这样的话,当到了截止的时间点,就会自动取消 Context,当然我们也可以不等到这个时候,然后可以通过取消函数提前进行取消。

  • WithTimeout 函数,和 WithDeadline 基本上一样,会传入一个 timeout 超时时间,也就是是从现在开始,直到过来 timeout 时间后,就进行超时取消,注意,这个是超时取消,不是截止时间取消。

  • WithValue 函数,这个和 WithCancel 就没有关系了,它不是用来控制父子协程生命周期的,这个是我们说到的,在 context 中传递基础元数据用的,这个可以在 context 中存储键值对的数据,然后这个键值对的数据可以通过 Context.Value 方法获取到,这是我们实际用经常要用到的技巧,一般我们想要通过上下文来传递数据时,可以通过这个方法,如我们需要 tarceID 追踪系统调用栈的时候。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-YscUL2va-1669634015262)(media/16346485078421.jpg)]

Context 的常用方法实例

1. 调用 Context Done方法取消

func ContextDone(ctx context.Context, out chan<- Value) error {for {v, err := AllenHandler(ctx)if err != nil {return err}select {case <-ctx.Done():log.Infof("context has done")return ctx.Err()case out <- v:}}
}

2. 通过 context.WithValue 来传值

func main() {ctx, cancel := context.WithCancel(context.Background())valueCtx := context.WithValue(ctx, key, "add value from allen")go watchAndGetValue(valueCtx)time.Sleep(10 * time.Second)cancel()time.Sleep(5 * time.Second)
}func watchAndGetValue(ctx context.Context) {for {select {case <-ctx.Done()://get valuelog.Infof(ctx.Value(key), "is cancel")returndefault://get valuelog.Infof(ctx.Value(key), "int goroutine")time.Sleep(2 * time.Second)}}
}

3. 超时取消 context.WithTimeout

 package mainimport ("fmt""sync""time""golang.org/x/net/context")var (wg sync.WaitGroup)func work(ctx context.Context) error {defer wg.Done()for i := 0; i < 1000; i++ {select {case <-time.After(2 * time.Second):fmt.Println("Doing some work ", i)// we received the signal of cancelation in this channelcase <-ctx.Done():fmt.Println("Cancel the context ", i)return ctx.Err()}}return nil}func main() {ctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)defer cancel()fmt.Println("Hey, I'm going to do some work")wg.Add(1)go work(ctx)wg.Wait()fmt.Println("Finished. I'm going home")}

4. 截止时间取消 context.WithDeadline

 package mainimport ("context""fmt""time")func main() {d := time.Now().Add(1 * time.Second)ctx, cancel := context.WithDeadline(context.Background(), d)// Even though ctx will be expired, it is good practice to call its// cancelation function in any case. Failure to do so may keep the// context and its parent alive longer than necessary.defer cancel()select {case <-time.After(2 * time.Second):fmt.Println("oversleep")case <-ctx.Done():fmt.Println(ctx.Err())}}

Context 使用原则 和 技巧

  • Context 是线程安全的,可以放心的在多个 goroutine 协程中传递

  • 可以把一个 Context 对象传递给任意个数的 gorotuine,对它执行 取消 操作时,所有 goroutine 都会接收到取消信号。

  • 不要把 Context 放在结构体中,要以参数的方式传递,parent Context 一般为Background,并且一般要在 main 函数的入口处创建然后传递下去

  • Context 的变量名建议都统一为 ctx,并且要把 Context 作为第一个参数传递给入口请求和出口请求链路上的每一个函数

  • 往下游给一个函数方法传递 Context 的时候,千万不要传递 nil,否则在 tarce 追踪的时候,就会中断链路,并且如果函数里面有获取值的逻辑,可能导致 panic。

  • Context 的 Value 只能传递一些通用或者基础的元数据,不要传递业务层面的数据,不是说不可以传递,是在 Go 的编码规范或者惯用法中不提倡不要什么数据都使用这个传递。由于 context 存储 key-value 是链式的,因此查询复杂度为O(n),所以,尽量不要随意存储不必要的数据

最后

  • 请允许我打个小广告:这篇文章首发在我微信公众号【后端系统和架构】中,点击这里可以去往公众号查看原文链接,如果对你有帮助,欢迎前往关注,更加方便快捷的接收最新优质文章

Golang Context 详细原理和使用技巧相关推荐

  1. Golang interface 接口详细原理和使用技巧

    文章目录 Golang interface 接口详细原理和使用技巧 一.Go interface 介绍 interface 在 Go 中的重要性说明 interface 的特性 interface 接 ...

  2. Golang中context实现原理剖析

    转载: Go 并发控制context实现原理剖析 1. 前言 Golang context是Golang应用开发常用的并发控制技术,它与WaitGroup最大的不同点是context对于派生gorou ...

  3. Android在Context详细解释 ---- 你不知道Context

                                                                                                         ...

  4. Linux 企业级安全原理和防范技巧

    Linux 企业级安全原理和防范技巧 1. 企业级Linux系统防护概述 1.1 企业级Linux系统安全威胁 1.2 企业级Linux系统安全立体式防范体系 1.2.1 Linux文件系统访问安全 ...

  5. 支持向量机SVM详细原理,Libsvm工具箱详解,svm参数说明,svm应用实例,神经网络1000案例之15

    目录 支持向量机SVM的详细原理 SVM的定义 SVM理论 Libsvm工具箱详解 简介 参数说明 易错及常见问题 SVM应用实例,基于SVM的股票价格预测 支持向量机SVM的详细原理 SVM的定义 ...

  6. java异常详细讲解_Java异常处理机制的详细讲解和使用技巧

    一起学习 1. 异常机制 1.1 异常机制是指当程序出现错误后,程序如何处理.具体来说,异常机制提供了程序退出的安全通道.当出现错误后,程序执行的流程发生改变,程序的控制权转移到异常处理器. 1.2 ...

  7. Qt 实现钢笔画线效果详细原理

    前言 上一篇文章:Qt 实现画线笔锋效果详细原理,根据这篇介绍的实现笔锋效果的原理,我们很容易实现另外一种笔效:钢笔. 所谓的钢笔笔效,就是真实还原钢笔书写出来的线条效果,其特征就是:根据笔的绘制速度 ...

  8. 很详细的“追女生技巧”

    很详细的"追女生技巧"恋人们感到最困难的是不知怎样去开始这个程序.男子会担心自己行为唐突,女子会担心男子会因而视自己虚浮.说到底,应由男方来开始这个程序,不妨参照下面所列之具体技巧 ...

  9. P2P技术详解(一):NAT详解——详细原理、P2P简介(转)

    这是一篇介绍NAT技术要点的精华文章,来自华3通信官方资料库,文中对NAT技术原理的介绍很全面也很权威,对网络应用的应用层开发人员而言有很高的参考价值. <P2P技术详解>系列文章 ➊ 本 ...

最新文章

  1. 这个新方法,竟然能检测 Python 代码的好坏!
  2. VB.Net实现Web Service的基础
  3. 演示:混合配置基于Linux winows cisco环境动态路由
  4. 在 MySQL 中使用码农很忙 IP 地址数据库
  5. [密码学基础][每个信息安全博士生应该知道的52件事][Bristol Cryptography][第32篇]基于博弈的证明和基于模拟的证明
  6. 算法与数据结构(三) 二叉树的遍历及其线索化(Swift版)
  7. (九)linux中断编程
  8. CSS3 多列布局列的填充方式column-fill属性
  9. 通向云帝国的铁王座:卖书的贝佐斯和卖软件的纳德拉
  10. php 调用mp3,使用PHP合并MP3文件的类,兼容php4、php5(2)
  11. 树莓派+android+盒子,最强电视盒子诞生记-树莓派4电视盒子
  12. 单招计算机专业的自我介绍,单招面试三分钟自我介绍范文
  13. 微信公众号 隐藏菜单
  14. 软件智能:aaas系统中AI的任务能力和工作
  15. RoadRunner中自建地图并作为Carla Map笔记
  16. javaweb,img问题scr路径
  17. 怎么安装使用pcsx2的方法(用pc玩ps2游戏 )
  18. VB亲身开发一个Windows软件(三)界面设计
  19. cmd中发送http请求_curl命令与HTTP请求
  20. 如何上传专用密码和登录iCloud教程

热门文章

  1. Shiro 授权(权限)
  2. Linux协议栈--NAPI机制
  3. python psutil 进程cpu_python 模块psutil获取进程信息
  4. element-ui 使用自定义复选框
  5. Python——创建对象
  6. misra c编码规范个人整理总结/misra c 2012中文版-个人总结-【方便查询】
  7. IO子系统(一) — 块设备驱动程序
  8. 动态IP与静态ip的区别是什么
  9. 【Unity】【Wwise】在Unity中获取某个Wwise事件的持续时间
  10. TESTTESTTESTTESTTESTTEST