【Golang】Sync.WaitGroup的使用

MaybeHC

发布于 2024-04-23 19:36:36

890

发布于 2024-04-23 19:36:36

示例

这里我们直接来用代码进行讲解，以下面的代码为例我们需要等待协程结束后查询count的，所以最后在末尾处使用了睡眠time.Sleep(time.Second* 2) ，目前的是一个简单的业务逻辑我们可以使用睡眠来等待，但是如果遇到业务复杂的情况，时间是无法进行预料的，这样的方法就不能使用了，

func main(){
	count := 0
	for i:=0 ;i<10000;i++{
		go func() {
			//数据处理
			count++
		}()
	}
	time.Sleep(time.Second* 2)
	fmt.Println(count)
}

这里我们可以使用WaitGroup来对代码进行优化

Sync.WaitGroup

简单使用就是在创建一个任务的时候wg.Add(1), 任务完成的时候使用wg.Done()来将任务减一。使用wg.Wait()来阻塞等待所有任务完成。

func main(){
	wg:= sync.WaitGroup{}
	count := 0
	for i:=0 ;i<10000;i++{
		wg.Add(1)
		go func() {
			defer wg.Done()
			count++
		}()
	}
	wg.Wait()
	fmt.Println(count)
}

代码使用WaitGroup，这样就无需通过估算时间睡眠这种方式，等待协程的结束，也可以这样写，如下：

func main(){
	wg:= sync.WaitGroup{}
	count := 0
    wg.Add(10000)
	for i:=0 ;i<10000;i++{
		go func() {
			defer wg.Done()
			count++
		}()
	}
	wg.Wait()
	fmt.Println(count)
}