并发控制之channel
我们考虑一种场景,协程A在执行过程中需要创建子协程A1、A2、A3…An,协程A创建完子协程后就等待子协程退出。针对这种场景,Go提供了三种解决方案。
- Channel:使用channel控制子协程
- WaitGroup:使用信号量机制控制子协程
- Context:使用上下文控制子协程
三种方案各有优劣,比如Channel的优点是实现简单,清晰易懂,WaitGroup的优点是子协程的个数可以动态调整,Context的优点是对子协程派生出来的孙子协程的控制。各种解决方案的缺点是相对而言的,需要根据具体的场景进行选择。
channel
channel一般用于协程之间的通信,不过channel也可以用于控制并发。比如主协程启动N个子协程,主协程等待所有子协程退出后再继续后续流程,这种场景下channel也可以轻易实现并发控制。
场景示例
下面的程序展示了一个使用channel控制子协程的例子:
package main
import (
"fmt"
"time"
)
func Process(ch chan int) {
time.Sleep(time.Second * 2)
ch <- 1 //在管道中写入一个元素表示当前协程已经结束
}
func main() {
channels := make([]chan int, 10)
for i := 0; i < 10; i++ {
channels[i] = make(chan int)
go Process(channels[i])
}
for i, ch := range channels {
<-ch
fmt.Println("Routine ", i, " quit")
}
}
上面的程序通过创建N个channel来管理N个协程,每个协程都有一个channel用于跟父协程通信,父协程创建完所有协程后等待所有协程结束。
在这个例子中,父协程仅仅是等待子协程结束,其实父协程也可以向管道中写入数据通知子协程结束,这是子协程需要定期的探测管道中是否有消息出现。
小结
使用channel控制子协程的优点是实现简单,缺点是当需要大量创建协程时就需要有相同数量的channel,而且对于子协程继续派生出来的协程不方便进行控制。