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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值