使用sync.WaitGroup来同步golang程序

golang中有2种方式同步程序，一种使用channel，另一种就是sync.WaitGroup,今天我们介绍比较简单的sync.WaitGroup。

sync.WaitGroup只有3个方法，Add()，Done()，Wait()。其中Done()是Add(-1)的别名。简单的来说，使用Add()添加计数，Done()减掉一个计数，计数不为0, 阻塞Wait()的运行。
要注意的有一点。sync文档已经说明了的，在运行main函数的goroutine里运行Add()函数，在其他的goroutine里面运行Done()函数。

代码示例如下：

package main
import (
    "runtime"
    "time"
    "fmt"
    "sync"
)
//定义任务队列
var waitgroup sync.WaitGroup
func xtgxiso(num int) {
    fmt.Println(num)
    time.Sleep(1*time.Second)
    waitgroup.Done() //任务完成，将任务队列中的任务数量-1，其实.Done就是.Add(-1)
}
func main() {
    //设置最大的可同时使用的CPU核数和实际cpu核数一致
    runtime.GOMAXPROCS(runtime.NumCPU())
    for i := 0; i < 10; i++ {
        waitgroup.Add(1) //每创建一个goroutine，就把任务队列中任务的数量+1
        go xtgxiso(i)
    }
    waitgroup.Wait() //Wait()这里会发生阻塞，直到队列中所有的任务结束就会解除阻塞
    fmt.Println("finish")
}

这样我们就完成了一个简单的程序，用runtime.GOMAXPROCS(runtime.NumCPU())来控制最大协程数量，用waitgroup.Wait()来等待所有子协程完成后主协程才退出！

参考:
http://www.xtgxiso.com/%E4%BD%BF%E7%94%A8sync-waitgroup%E6%9D%A5%E5%90%8C%E6%AD%A5golang%E7%A8%8B%E5%BA%8F/

使用sync.WaitGroup来同步golang程序

微信号：腾讯云折扣返现

技术讨论QQ群： 863102326

作者: catherine

搜索

标签