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过期导致)