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内容请搜索www.887551.com以前的文章或继续浏览下面的相关文章希望大家以后多多支持www.887551.com！
