Golang中的同步工具sync.WaitGroup详解_开发_开发者

开发者 https://www.devze.com 2023-11-20 14:58 出处：网络作者：路多辛

目录sync.WaitGroup使用方法和示例使用过程中需要注意的点sync.WaitGroup sync.WaitGroup作用是等待一组goro编程utine执行完成，当使用多个goroutine处理任务的时候，如果要等待所有的goroutine都执行完成后再执行下

sync.WaitGroup

sync.WaitGroup作用是等待一组goro编程utine执行完成，当使用多个goroutine处理任务的时候，如果要等待所有的goroutine都执行完成后再执行下一步操作，就可以使用sync.WaitGroup来实现。

sync.WaitGroup有三个方法：

Add(delta int)：增加或减少等待goroutine的数量，delta可以为负数；
Done()：标记goroutine已执行完成，等价于Add(-1)；
Wait()：等待所有的goroutine执行完成。

使用方法和示例

假如有n个任务需要处理，可以先创建一个WaitGroup，用于等待所有的任务完成：

var wg sync.WaitGroup

在处理每一个任务之前，调用一次Add方法：

wg.Add(1)

在每一个任务处理完成之后，调用一次Done方法：

wg.Done()

最后，在等待所有的任务完成之前，需要调用一次Wait方法：

wg.Wait()

具体示例代码如下：

package main
 
import (
	"log"
	"sync"
	"time"
)
 
func main() {
	// 创建一个等待组
	var wg sync.WaitGroup
 
	for i := 1; i <= 5; i++ {
		// 每一个任务开始前, 增加1
		wg.Add(1)
 
		go func(index int) {
			// 处理业务逻辑
			//...
			time.Shttp://www.devze.comleep(1 * time.Second)
			log.Printf("第 %d 个gorohttp://www.devze.comutine执行完了", index)
			wg.Done()
		}(i)
	}
	// 等待所有的任务完成
	wg.Wait()
	log.Println("所有goroutine都执行完了")
}