背景

在父子协程协作过程中, 父协程需要给子协程传递信息, 子协程依据父协程传递的信息来决定自己的操作.

这种需求下可以使用 context 包

简介

Context通常被称为上下文,在go中,上层的协程可以将context 传给下层的协程, 来实现协程之间的信息传递, 同时下层协程也可以将context 传给更下层的协程, 来形成一张树状图.

主要方法

获得顶级上下文

首先要获得最顶级协程使用的Context

func Background() Context

Background 方法一般创建根 Context 的时候。

func TODO() Context

TODO 方法,当不清楚使用哪个上下文时,可以使用 TODO 方法。

当前协程上下文的操作

针对 Context 可以执行如下操作.

Deadline() (deadline time.Time, ok bool)

Deadline 方法返回结果有两个,第一个是截止时间,到了这个截止时间,Context 会自动取消;第二个是一个 bool 类型的值,如果 Context 没有设置截止时间,第二个返回结果是 false,如果需要取消这个 Context,就需要调用取消函数。

Done() <-chan struct{}

Done 方法返回一个只读的 channel 对象,类型是 struct{},在 goroutine 中,如果 Done 方法返回的结果可以被读取,代表父 Context 调用了取消函数。

Err() error

Err 方法返回 Context 被取消的原因。

Value(key interface{}) interface{}

Value 方法返回此 Context 绑定的值。它是一个 kv 键值对,通过 key 获取对应 value 的值

创建下级协程的Context

我们要依据父级的协程的Context创建子级协程的Context

WithCancel(parent Context) (ctx Context, cancel CancelFunc)

WithCancel 方法,基于父 Context,接收一个父 Context 参数,生成一个新的子 Context,和一个 cancel 函数,用于取消 Context。

WithDeadline(parent Context, d time.Time) (Context, CancelFunc)

WithDeadline 方法,基于父 Context,接收一个父 Context 参数,和一个截止时间的参数,生成一个新的子 Context,和一个 cancel 函数,可以使用 cancel 函数取消 Context,也可以等到截止时间,自动取消 Context。

WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)

WithTimeout 方法,基于父 Context,接收一个父 Context 参数,和一个超时时间的参数,生成一个新的子 Context,和一个 cancel 函数,可以使用 cancel 函数取消 Context,也可以等到超时时间,自动取消 Context。

WithValue(parent Context, key, val interface{}) Context

WithValue 方法,基于父 Context,生成一个新的子 Context,携带了一个 kv 键值对,一般用于传递上下文信息。

场景示例

公司下班, 要领导下班后, 员工才下班, 但是员工忍耐也是有限度的, 领导老是不下班, 员工就自己走了.

up_context ctxup_context WithTimeoutcancel

可以看到, 领导就加班了3秒, 领导一下班, 员工就下班了.

如果领导加班7秒呢, 代码改成如下

执行结果如下

可以看到, 领导加班7秒, 员工并没有等着领导下班, 在第五秒的时候自己下班了.