Golang 语言控制并发 Goroutine的方法


Posted in Golang onJune 30, 2021

goroutine 是 Go语言中的轻量级线程实现,由 Go 运行时(runtime)管理。Go 程序会智能地将 goroutine 中的任务合理地分配给每个 CPU。

01介绍

Golang 语言的优势之一是天生支持并发,我们在 Golang 语言开发中,通常使用的并发控制方式主要有 Channel,WaitGroup 和 Context,本文我们主要介绍一下 Golang 语言中并发控制的这三种方式怎么使用?关于它们各自的详细介绍在之前的文章已经介绍过,感兴趣的读者朋友们可以按需翻阅。

02Channel

在 Golang 语言中,Channel 不仅可以用于协程之间通信,还可以使用 Channel 控制子协程,而且使用 Channel 实现并发控制比较简单,比如以下示例,我们在 Golang 应用程序中启动两个协程,分别是主协程和子协程,主协程需要等待子协程运行结束后再退出程序。

示例代码:

func main () { 
 done := make(chan struct{}) 
 go func() { 
  fmt.Println("goroutine run over") 
  done <- struct{}{} 
 }() 
 <- done 
 fmt.Println("main goroutine run over") 
}

阅读上面这段代码,我们在子 goroutine 运行结束后,通过 Channel 通知主 goroutine 退出程序,实际上也可以反过来处理,主 goroutine 通知子 goroutine 退出程序,主 goroutine 向 channel 中发送数据,子 goroutine 等待接收 channel 中的数据。

03sync.WaitGroup

如果在 Golang 应用程序中,需要让主 goroutine 等待多个 goroutine 都运行结束后再退出程序,我们应该怎么实现呢?是的,同样可以使用 Channel 实现,但是,有一个更优雅的实现方式,那就是 WaitGroup,顾名思义,WaitGroup 就是等待一组 goroutine 运行结束。

示例代码:

func main () { 
 wg := sync.WaitGroup{} 
 wg.Add(10) 
 for i := 0; i < 10; i++ { 
  go func(id int) { 
   fmt.Println(id, "运行结束") 
   wg.Done() 
  }(i) 
 } 
 wg.Wait() 
 fmt.Println("main goroutine run over") 
}

阅读上面这段代码,我们启动 10 个子 goroutine,主 goroutine 需要等待 10 个子 goroutine 都运行结束后再退出程序,我们使用的是 WaitGroup,它有三个方法,分别是 Add、Done 和 Wait,实际上 WaitGroup 维护了一个计数器,这三个方法都是围绕这个计数器工作,Add 用于设置计数器的数值,Done 用于扣减计数器的数值,Wait 在计数器数值为 0 之前一直阻塞。关于 WaitGroup 的源码解读,在之前的文章中已介绍过,限于篇幅,这里就不再赘述。

04Context

Channel 和 WaitGroup 通常用于父子两个层级的 goroutine 的应用程序的并发控制中,如果在 Golang 应用程序中,子协程继续派生出协程,我们应该怎么控制呢?这种多级 goroutine 的应用程序,我们可以使用 Context 实现并发控制。

示例代码:

func main() { 
 ctx, cancel := context.WithCancel(context.Background()) 
 go firstCtx(ctx) 
 time.Sleep(5 * time.Second) 
 fmt.Println("stop all sub goroutine") 
 cancel() 
 time.Sleep(5 * time.Second) 
} 
 
func firstCtx(ctx context.Context) { 
 go secondCtx(ctx) 
 for { 
  select { 
  case <-ctx.Done(): 
   fmt.Println("first done") 
   return 
  default: 
   fmt.Println("first running") 
   time.Sleep(2 * time.Second) 
  } 
 } 
} 
 
func secondCtx(ctx context.Context) { 
 for { 
  select { 
  case <-ctx.Done(): 
   fmt.Println("second done") 
   return 
  default: 
   fmt.Println("second running") 
   time.Sleep(2 * time.Second) 
  } 
 } 
}

阅读上面这段代码,在子协程 firstCtx 启动子协程 secondCtx,主 goroutine 创建 context,并把 context 传递到所有子协程,然后主 goroutine 通过调用 cancle 停掉所有子协程。

05总结

本文我们介绍了不同场景中分别适合哪种控制并发 goroutine 的方式,其中,channel 适合控制少量 并发 goroutine,WaitGroup 适合控制一组并发 goroutine,而 context 适合控制多级并发 goroutine。

到此这篇关于Golang 语言控制并发 Goroutine的方法的文章就介绍到这了,更多相关Golang并发控制Goroutine内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Golang 相关文章推荐
go原生库的中bytes.Buffer用法
Apr 25 Golang
Go语言中的UTF-8实现
Apr 26 Golang
golang中切片copy复制和等号复制的区别介绍
Apr 27 Golang
golang 如何通过反射创建新对象
Apr 28 Golang
解决Go gorm踩过的坑
Apr 30 Golang
go语言中fallthrough的用法说明
May 06 Golang
浅谈Golang 切片(slice)扩容机制的原理
Jun 09 Golang
Go 语言结构实例分析
Jul 04 Golang
golang fmt格式“占位符”的实例用法详解
Jul 04 Golang
Golang表示枚举类型的详细讲解
Sep 04 Golang
golang操作rocketmq的示例代码
Apr 06 Golang
Golang流模式之grpc的四种数据流
Apr 13 Golang
Golang的继承模拟实例
Jun 30 #Golang
go select编译期的优化处理逻辑使用场景分析
Go 语言下基于Redis分布式锁的实现方式
Jun 28 #Golang
go语言使用Casbin实现角色的权限控制
Go语言设计模式之结构型模式
浅谈Go语言多态的实现与interface使用
Jun 16 #Golang
再次探讨go实现无限 buffer 的 channel方法
Jun 13 #Golang
You might like
星际争霸 Starcraft 游戏介绍
2020/03/14 星际争霸
php打印一个边长为N的实心和空心菱型的方法
2015/03/02 PHP
php判断访问IP的方法
2015/06/19 PHP
PHP实现二维数组按照指定的字段进行排序算法示例
2019/04/23 PHP
JavaScript使用focus()设置焦点失败的解决方法
2014/09/03 Javascript
jQuery判断一个元素是否可见的方法
2015/06/05 Javascript
javascript数组对象常用api函数小结(连接,插入,删除,反转,排序等)
2016/09/20 Javascript
Windows系统下安装Node.js的步骤图文详解
2016/11/15 Javascript
Vue2.0利用vue-resource上传文件到七牛的实例代码
2017/07/28 Javascript
微信小程序实现轮播图效果
2017/09/07 Javascript
把vue-router和express项目部署到服务器的方法
2018/02/21 Javascript
[56:58]VP vs Optic 2018国际邀请赛小组赛BO2 第一场 8.16
2018/08/17 DOTA
使用Protocol Buffers的C语言拓展提速Python程序的示例
2015/04/16 Python
Python字符串、元组、列表、字典互相转换的方法
2016/01/23 Python
TensorFlow搭建神经网络最佳实践
2018/03/09 Python
Django中反向生成models.py的实例讲解
2018/05/30 Python
Python wxpython模块响应鼠标拖动事件操作示例
2018/08/23 Python
Pycharm设置utf-8自动显示方法
2019/01/17 Python
Python3实现的判断回文链表算法示例
2019/03/08 Python
20行python代码的入门级小游戏的详解
2019/05/05 Python
python实现美团订单推送到测试环境,提供便利操作示例
2019/08/09 Python
Python socket模块ftp传输文件过程解析
2019/11/05 Python
关于ResNeXt网络的pytorch实现
2020/01/14 Python
Python unittest基本使用方法代码实例
2020/06/29 Python
python绘制趋势图的示例
2020/09/17 Python
任意存:BOXFUL
2018/05/21 全球购物
理肤泉俄罗斯官网:La Roche-Posay俄罗斯
2018/07/24 全球购物
网络程序员自荐信
2014/01/25 职场文书
一年级小学生评语
2014/04/22 职场文书
党员干部群众路线个人整改措施
2014/09/18 职场文书
舌尖上的中国观后感
2015/06/02 职场文书
辣妈辣妹观后感
2015/06/10 职场文书
在校大学生才艺比赛策划书怎么写?
2019/08/26 职场文书
尝试使用Python爬取城市租房信息
2022/04/12 Python
使用vue判断当前环境是安卓还是IOS
2022/04/12 Vue.js
Go语言安装并操作redis的go-redis库
2022/04/14 Golang