golang：context

context作用

goroutine的退出机制

多个goroutine都是平行的被调度的，多个goroutine如何协调工作涉及通信、同步、通知和退出

• 通信：goroutine之间的通信同步chan通道

• 同步：不带缓冲的chan提供了一个天然的同步等待机制。通过WaitGroup也可以为多个goroutine提供同步等待机制

• 通知：这个通知和上面通信的数据不一样，通知通常不是业务数据，而是管理、控制流数据。要处理这个也好办，在输入端绑定两个chan，一个用于业务流数据，另一个用于异常通知数据，然后通过 select 收敛进行处理。这个方案可以解决简的问题，但不是一个通用的解决方案。

• 退出：goroutine 之间没有父子关系，如何通知 goroutine 退出？可以通过增加一个个单独的通道，借助通道和 select 的广播机制（ close channel to broadcast）实现退出

通知退出机制：

• 读取已经关闭的通道不会引起阻塞，也不会导致panic，而是立即返回该通道存储类型的零值

• 关闭select监听的某个通道能使select立即感知这种通知，然后进行相应的处理，即所谓的退出通知机制

contet库的设计目的就是跟踪 goroutine调用树，并在这些goroutine 调用树中递通知和元数据

• 退出通知机制：通知可以传递给整个 goroutine调用树上的每一个goroutine

• 传递数据：数据可传递给整个goroutine 调用树上的每一个goroutine

使用

var wg sync.WaitGroup
func main() {background := context.Background()//通过cancel进行控制/*cancelctx, cancelFunc := context.WithCancel(background)wg.Add(2)go jiankong1(cancelctx)time.Sleep(time.Second * 2)println("并不是因为主协程退出，而导致另一个协程退出，而是通过cancel方法通知另一个协程进行退出！")cancelFunc()wg.Wait()println("完成退出")*///通过deadline进行控制，也可以通过返回的cancelFunc手动进行控制/*deadline, _ := context.WithDeadline(background, time.Now().Add(time.Second*2))go deadF(deadline)time.Sleep(time.Second * 4)println("结束")*///通过timeout控制，也可以通过返回的cancelFunc手动进行控制   超时以后报的错和deadline控制方式一样 context deadline exceededtimeout, _ := context.WithTimeout(background, time.Second*2)go deadF(timeout)time.Sleep(time.Second * 4)println("结束")
}func deadF(ctx context.Context) {for true {select {case <-ctx.Done():fmt.Println("只给了两秒的执行时间，时间到了，我得退出了。")//此处打印的错误为 context deadline exceededprintln(ctx.Err().Error())returndefault:time.Sleep(time.Millisecond * 1000)fmt.Println("我正在执行...")}}
}func jiankong1(ctx context.Context) {defer wg.Done()cancel, _ := context.WithCancel(ctx)go jiankong2(cancel)for true {select {case <-ctx.Done():fmt.Println("别人通知我主动退出了。在此处进行收尾工作，释放资源等")//此处报错为：context canceledfmt.Println(ctx.Err().Error())returndefault:time.Sleep(time.Millisecond * 100)fmt.Println("我在持续监控中...")}}
}func jiankong2(ctx context.Context) {defer wg.Done()for true {select {case <-ctx.Done():fmt.Println("上级通知我主动退出了。")fmt.Println(ctx.Err().Error())returndefault:time.Sleep(time.Millisecond * 100)fmt.Println("me在持续监控中...")}}
}

Tips:

异步任务使用context注意点：

使用go func进行异步任务传入context时，要考虑context的过期时间，如果异步任务不考虑过期时间，则最好传入一个无过期时间的context，负责异步任务多或执行时间长context会过期，影响异步任务的执行。

场景：mongo批量插入数据后，将返回的Id集合传入异步任务（在异步任务中通过id查询数据），如果共用了接口设置过期时间的context，可能会导致一部分数据查询不到（context过期导致）