Go中如何使用channel实现并发控制？

[qloog]

1. 前言

channel一般用于协程之间的通信，channel也可以用于并发控制。比如主协程启动N个子协程，主协程等待所有子协程退出后再继续后续流程，这种场景下channel也可轻易实现。

2. 场景示例

下面程序展示一个使用channel控制子协程的例子：

package main

import (
    "time"
    "fmt"
)

func Process(ch chan int) {
    //Do some work...
    time.Sleep(time.Second)

    // 管道中写入一个元素表示当前协程已结束
    ch <- 1
}

func main() {
    // 创建一个10个元素的切片，元素类型为channel
    channels := make([]chan int, 10) 

    for i:= 0; i < 10; i++ {
        // 切片中放入一个channel
        channels[i] = make(chan int) 
        // 启动协程，传一个管道用于通信
        go Process(channels[i])      
    }
    // tips: 以上如果不是同一个函数执行，也可以不用for循环，而是单独调用不同的方法
    // 比如  
    // go func1(make(chan int) )
    // go func2(make(chan int) )
    // ...

    // 遍历切片，等待子协程结束
    for i, ch := range channels { 
        <-ch
        fmt.Println("Routine ", i, " quit!")
    }
}

上面程序通过创建N个channel来管理N个协程，每个协程都有一个channel用于跟父协程通信，父协程创建完所有协程中等待所有协程结束。

这个例子中，父协程仅仅是等待子协程结束.

3. 总结

使用channel来控制子协程的优点是实现简单，缺点是当需要大量创建协程时就需要有相同数量的channel，而且对于子协程继续派生出来的协程不方便控制。这时可以使用 context来进行处理。