golang并发控制之channel

并发控制之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，而且对于子协程继续派生出来的协程不方便进行控制。