go context之WithCancel的使用

简言

1. WithCancel()函数接受一个 Context 并返回其子Context和取消函数cancel

2. 新创建协程中传入子Context做参数，且需监控子Context的Done通道，若收到消息，则退出

3. 需要新协程结束时，在外面调用 cancel 函数，即会往子Context的Done通道发送消息

4. 注意：当 父Context的 Done() 关闭的时候，子 ctx 的 Done() 也会被关闭

实验步骤

1. 利用根Context创建一个父Context，使用父Context创建一个协程，

2. 利用上面的父Context再创建一个子Context，使用该子Context创建一个协程

3. 一段时间后，调用父Context的cancel函数，会发现父Context的协程和子Context的协程都收到了信号，被结束了

实验结果如下图

代码如下

package main

import (
	"context"
	"fmt"
	"time"
)

func main() {
	// 父context(利用根context得到)
	ctx, cancel := context.WithCancel(context.Background())

	// 父context的子协程
	go watch1(ctx)

	// 子context，注意：这里虽然也返回了cancel的函数对象，但是未使用
	valueCtx, _ := context.WithCancel(ctx)
	// 子context的子协程
	go watch2(valueCtx)

	fmt.Println("现在开始等待3秒,time=", time.Now().Unix())
	time.Sleep(3 * time.Second)

	// 调用cancel()
	fmt.Println("等待3秒结束,调用cancel()函数")
	cancel()

	// 再等待5秒看输出，可以发现父context的子协程和子context的子协程都会被结束掉
	time.Sleep(5 * time.Second)
	fmt.Println("最终结束,time=", time.Now().Unix())
}

// 父context的协程
func watch1(ctx context.Context) {
	for {
		select {
		case <-ctx.Done(): //取出值即说明是结束信号
			fmt.Println("收到信号，父context的协程退出,time=", time.Now().Unix())
			return
		default:
			fmt.Println("父context的协程监控中,time=", time.Now().Unix())
			time.Sleep(1 * time.Second)
		}
	}
}

// 子context的协程
func watch2(ctx context.Context) {
	for {
		select {
		case <-ctx.Done(): //取出值即说明是结束信号
			fmt.Println("收到信号，子context的协程退出,time=", time.Now().Unix())
			return
		default:
			fmt.Println("子context的协程监控中,time=", time.Now().Unix())
			time.Sleep(1 * time.Second)
		}
	}
}