go语言-context的基本使用

1. 什么是 Context？

Go 1.7 标准库引入 context，中文译作“上下文”，准确说它是 goroutine 的上下文，包含 goroutine 的运行状态、环境、现场等信息。

context 主要用来在 goroutine 之间传递上下文信息，包括：取消信号、超时时间、截止时间、k-v 等。

Context，也叫上下文，它的接口定义如下

type Context interface {Deadline() (deadline time.Time, ok bool)Done() <-chan struct{}Err() errorValue(key interface{}) interface{}
}

可以看到 Context 接口共有 4 个方法

• Deadline：返回的第一个值是截止时间，到了这个时间点，Context 会自动触发 Cancel 动作。返回的第二个值是 一个布尔值，true 表示设置了截止时间，false 表示没有设置截止时间，如果没有设置截止时间，就要手动调用 cancel 函数取消 Context。

• Done：返回一个只读的通道（只有在被cancel后才会返回），类型为 struct{}。当这个通道可读时，意味着parent context已经发起了取消请求，根据这个信号，开发者就可以做一些清理动作，退出goroutine。

• Err：返回 context 被 cancel 的原因。

• Value：返回被绑定到 Context 的值，是一个键值对，所以要通过一个Key才可以获取对应的值，这个值一般是线程安全的。

2. 为什么要用Context

• 用于控制goroutine的结束，但它解决的并不是 能不能 的问题，而是解决 更好用 的问题。

2.1. 当不用Context时，利用channel+select来主动让goroutine停止

示例1

package mainimport ("fmt""time"
)/*
1. 利用channel控制goroutine的停止
*/func main() {stopChan := make(chan bool)go func() {for {select {case <-stopChan:fmt.Println("goroutin1 exit.")returndefault:fmt.Println("goroutin1 sleep 1s, keep going.")time.Sleep(time.Second * 2)}}}()go func() {for {select {case <-stopChan:fmt.Println("goroutin2 exit.")returndefault:fmt.Println("goroutin2 sleep 1s, keep going.")time.Sleep(time.Second * 3)}}}()time.Sleep(10 * time.Second)fmt.Println("10s 时间到了，主进程需要退出了.")// 发送信号让goroute1结束stopChan <- true// 发送信号让goroute2结束stopChan <- truetime.Sleep(5 * time.Second)
}

示例2

package mainimport ("fmt""time"
)/*1. 利用关闭channel的方法，让2个goroutine同时结束
*/func main() {stopChan := make(chan bool)go func() {for {select {case <-stopChan:fmt.Println("goroutin1 exit.")returndefault:fmt.Println("goroutin1 sleep 1s, keep going.")time.Sleep(time.Second * 2)}}}()go func() {for {select {case <-stopChan:fmt.Println("goroutin2 exit.")returndefault:fmt.Println("goroutin2 sleep 1s, keep going.")time.Sleep(time.Second * 3)}}}()time.Sleep(10 * time.Second)fmt.Println("10s 时间到了，主进程需要退出了.")// 利用关闭channel的方法，让2个goroutine同时结束close(stopChan)time.Sleep(5 * time.Second)
}

2.2 使用context来主动让goroutine停止

先ctx, cancel := context.WithCancel(context.Background()) 创建一个ctx实例
再利用cancel()函数执行控制goroutine的停止

package mainimport ("context""fmt""time"
)/*
// 利用context，手动让2个goroutine同时结束[是不是更简单?]
*/
func main() {ctx, cancel := context.WithCancel(context.Background())go func() {for {select {case <-ctx.Done():fmt.Println("goroutin1 exit.")returndefault:fmt.Println("goroutin1 sleep 1s, keep going.")time.Sleep(time.Second * 1)}}}()go func() {for {select {case <-ctx.Done():fmt.Println("goroutin2 exit.")returndefault:fmt.Println("goroutin2 sleep 1s, keep going.")time.Sleep(time.Second * 1)}}}()time.Sleep(10 * time.Second)fmt.Println("10s 时间到了，goroutine需要退出了.")// 利用context的方法，手动让2个goroutine同时结束cancel()time.Sleep(5 * time.Second)
}

2.3 使用context实现goroutine的超时控制

• 使用场景：让goroutine执行一个任务，如果在指定时间内没有完成，这利用context的WithTimeout（）主动让goroutine退出

package mainimport ("fmt""time""context"
)// 场景： 如果你需要对一个用协程启动的函数做超时控制，可以用context来完成goroutine的控制func main()  {// 设置一个用于超时控制的context ctx, ctx作为参数可以用来作为协程的超时控制ctx,cancel := context.WithTimeout(context.Background(),10 * time.Second)defer cancel()// ctx作为参数传递给需要做超时控制的函数go Monitor(ctx)time.Sleep(20 * time.Second)
}func Monitor(ctx context.Context)  {for {select {// 如果context 超时，ctx.Done()就会返回一个空接口 struct{}case <- ctx.Done():// 如果超时时间到了，就退出循环fmt.Println(ctx.Err())return// 如果没有超时，打印输出后继续循环default:time.Sleep(1*time.Second)fmt.Println("monitor")}}
}

2.4 利用context向goroutine传递参数

• 除了超时控制与主动停止goroutine，还有可以通过Context传递上下文变量给其他协程。这样可以避免在协程之间传递大量的变量，代码更整洁可维护。下面的例子通过WithValue传递给协程一个变量，并且通过channel在协程之间通信。

package mainimport ("context""fmt""time"
)func main() {// 为ctx设置一个key-valuectx := context.Background()ctx = context.WithValue(ctx, "hello", "world")x := ctx.Value("hello")fmt.Println("x=", x) // world// 将key-vluae值传递到goroutinego work(ctx)time.Sleep(3 * time.Second)}
func work(ctx context.Context) {fmt.Println("do worker.")fmt.Println("hello=", ctx.Value("hello")) // world，利用context传递key-value// 继续传递到下层goroutinego subwork(ctx)
}func subwork(ctx context.Context) {fmt.Println("do subwork.")fmt.Println("hello=", ctx.Value("hello")) // world，利用context传递key-value到更进一层
}

程序输出：

x= world
do worker.
world
do subwork.
world