前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >EasyDSS开发中Go语言在for循环中使用协程的注意事项

EasyDSS开发中Go语言在for循环中使用协程的注意事项

原创
作者头像
TSINGSEE青犀视频
修改2021-11-05 14:39:58
1.6K0
修改2021-11-05 14:39:58
举报
文章被收录于专栏:TSINGSEE青犀视频

之前我们介绍过EasyDSS开发中对野协程的管理,有兴趣的朋友可以了解一下:EasyDSS协程出现panic并导致程序退出,如何对野协程进行管理?在 EasyDSS 的程序开发中,有时为了加快速度,会在 for 循环中采用协程的方式进行代码编写,类似代码如下:

代码语言:javascript
复制
wg := sync.WaitGroup{}
wg.Add(length)
for s := range group.rtcSubSessionSet {
   //s.WriteRtcPacket(pkt)
   go func() {
      defer wg.Done()
      s.WriteRtcPacket(pkt)
   }()
}
wg.Wait()

以上代码 s 为从集合中取出的指针变量,因为采用协程的方式, go func(){} 代码会新启动一个协程进行运行。for 循环进入下一次循环,此时 s 的值被改变,因此会导致发的包会乱序。因此修改代码如下:

代码语言:javascript
复制
wg := sync.WaitGroup{}
wg.Add(length)
for s := range group.rtcSubSessionSet {
   //s.WriteRtcPacket(pkt)
   go func(sender *rtc.SubSession) {
      defer wg.Done()

      sender.WriteRtcPacket(pkt)
   }(s)
}
wg.Wait()

以上代码将前一个指针变量以传递参数的方式传递到协程中,即使下一次循环中 s 改变,也不会出现问题。

以上代码可以正常运行,但是如果协程中出现 panic 错误,就会导致整个程序退出,导致系统不稳定,因此我们进一步优化代码如下:

代码语言:javascript
复制
wg := sync.WaitGroup{}
wg.Add(length)
for s := range group.rtcSubSessionSet {
   //s.WriteRtcPacket(pkt)
   go func(sender *rtc.SubSession) {
      defer func() {
         if err := recover(); err != nil {
            log.Error(fmt.Sprintf("panic %s\n", err))
            log.Error(fmt.Sprintf(string(debug.Stack())))
         }
      }()
      defer wg.Done()

      sender.WriteRtcPacket(pkt)
   }(s)
}
wg.Wait()

在代码的最开始加上 recover() 函数捕获所有的 panic 信息,这样既可以知道哪里出错,也能保证程序稳定运行。

总结以下在写协程的时候主要注意两点:

1.保证捕获协程中的 panic 异常; 2.在协程中使用外部的变量时,应该以传参的方式传递到协程中。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档