使用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()来等待所有子协程完成后主协程才退出!

此条目发表在 网站开发, 网站架构 分类目录,贴了 , 标签。将固定链接加入收藏夹。

评论功能已关闭。