Go語言協程數目控制

Go语言有天生良好的多并发特征，依赖goroutines可以很好的实现多并发功能。但是，在多并发之下，控制goroutines的数量，也是需要考虑的问题。如果不限制goroutines，在较大量的goroutines影响下可能会消耗较多的CPU单元、内存、磁盘IO和网络性能。   控制的方法也很简单，可以通过channal方式限制。如下例：  

func main() {
    var wg sync.WaitGroup
    // 生成token
    tokens := make(chan int, 5)

    for i := 0; i < 20; i++ {
        // 获取
        tokens <- 1
        wg.Add(1)
        go func() {
            time.Sleep(1000 * time.Millisecond)
            fmt.Println("go is here!")
            // 缴回
            <-tokens
            defer wg.Done()
        }()
    }
    wg.Wait()
}

 

  则可以比较好的控制运行，除此之外，也可以更简单的使用CPU单元来限制，只需要替换一句：

tokens := make(chan int, runtime.NumCPU())