Golang gorouting 并发控制 sync.WaitGroup 介绍与使用

WaitGroup简称同步组,用于等待goroutines结束的。

官方文档:

type WaitGroup

A WaitGroup waits for a collection of goroutines to finish. The main goroutine calls Add to set the number 
of goroutines to wait for. Then each of the goroutines runs and calls Done when finished. At the same time, 
Wait can be used to block until all goroutines have finished.

A WaitGroup must not be copied after first use.

翻译过来就是:WaitGroup用来等待goroutings退出,主进程调用Add设置等待的协程数量,在协程里面结束的时候调用Done方法。Wait方法将会阻塞直到所有协程都调用Done方法退出。

WaitGroup不能被复制使用,可以以指针的方式传递。

func (wg *WaitGroup) Add(delta int)
增加wg计数器

func (wg *WaitGroup) Done()
wg计数器减一

func (wg *WaitGroup) Wait()
等待直到wg计数器变为0

用例:

// wg.go
package main

import (
	"fmt"
	"sync"
)

func child(wg *sync.WaitGroup, i int) {
	fmt.Printf("child:%d, exit \n", i)
	wg.Done()
}

func main() {

	//定义一个WaitGroup变量
	var wg sync.WaitGroup

	//添加gorouting
	for i := 0; i <= 5; i++ {
		wg.Add(1)
		go child(&wg, i)
	}

	//等待gorouting退出
	wg.Wait()
	fmt.Printf("all gorouting exit\n")
	return
}

运行结果:

Golang gorouting 并发控制 sync.WaitGroup 介绍与使用_第1张图片

参考文档:

1. https://golang.org/pkg/sync/#WaitGroup

你可能感兴趣的:(Go语言开发)