golang wg

Golang中的sync包提供了一系列的并发原语，使得并发编程更加简单。其中之一就是WaitGroup（WG）的使用。在本文中，我们将探讨如何正确地使用WG来管理并发操作。

什么是WaitGroup

WaitGroup是golang中的一个并发原语，用于等待一组goroutine的执行完成。它通过Add方法来设置需要等待的goroutine数量，通过Done方法来表示一个goroutine已经执行完成，通过Wait方法来阻塞等待所有的goroutine运行完毕。

如何使用WaitGroup

下面我们通过一个例子来演示如何使用WaitGroup。

假设我们的任务是爬取10个网页的内容，并将每个网页的结果存储到一个文件中。我们可以创建一个WaitGroup来等待所有的爬虫任务完成：

``` func main() { var wg sync.WaitGroup urls := []string{ "http://www.example.com/page1", "http://www.example.com/page2", // ... "http://www.example.com/page10", } for _, url := range urls { wg.Add(1) // 每个URL对应一个goroutine，因此增加WaitGroup的计数器 go func(u string) { defer wg.Done() // goroutine执行完成后减少WaitGroup的计数器 content := crawl(u) saveToFile(u, content) }(url) } wg.Wait() // 阻塞等待所有的goroutine完成 } ```

在上面的代码中，我们首先创建了一个WaitGroup，并指定需要等待的goroutine数量为urls数组的长度。

然后，我们使用for循环遍历urls数组，为每个URL启动一个goroutine。在每个goroutine中，我们首先增加WaitGroup的计数器（通过Add方法），然后使用匿名函数来执行具体的爬虫任务。在任务执行完成后，我们调用Done方法来减少WaitGroup的计数器。

最后，我们调用Wait方法来阻塞等待所有的goroutine完成。一旦所有的goroutine都执行完毕，程序将继续执行下去。

注意事项

在使用WaitGroup时，需要注意以下几点：

1. Add和Done方法需要成对出现

每个增加计数器的地方都需要有对应的Done操作，否则会导致程序永远等待下去。

2. 每个goroutine都必须执行Done操作

如果某个goroutine没有执行Done操作，Wait方法将永远阻塞。

3. 避免在WaitGroup的Wait方法之后添加新的goroutine

如果在Wait方法之后还有新的goroutine被添加，Wait方法可能无法正确等待所有的goroutine。

总之，WaitGroup是golang中一个非常有用的并发原语，能够帮助我们管理一组goroutine的执行。通过Add、Done和Wait方法的正确配合使用，我们可以确保并发操作的正确性，提高程序的并发性能。