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 exceeded
timeout, _ := 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 exceeded
println(ctx.Err().Error())
return
default:
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 canceled
fmt.Println(ctx.Err().Error())
return
default:
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())
return
default:
time.Sleep(time.Millisecond * 100)
fmt.Println("me在持续监控中...")
}
}
}
Tips:
异步任务使用context注意点:
使用go func进行异步任务传入context时,要考虑context的过期时间,如果异步任务不考虑过期时间,则最好传入一个无过期时间的context,负责异步任务多或执行时间长context会过期,影响异步任务的执行。
场景:mongo批量插入数据后,将返回的Id集合传入异步任务(在异步任务中通过id查询数据),如果共用了接口设置过期时间的context,可能会导致一部分数据查询不到(context过期导致)