阅读量:1
在Go语言的并发编程中，WaitGroup是一个非常有用的工具，可以帮助我们等待一组并发操作的完成。除了基本的用法之外，WaitGroup还有一些高级用法，可以使我们的并发编程更加灵活和高效。
- 使用Add方法动态增加等待的数量
在一些情况下，我们可能无法提前知道并发操作的数量，但是我们仍然希望能够等待它们全部完成。这时，我们可以使用WaitGroup的Add方法动态增加等待的数量。例如：
var wg sync.WaitGroup // 模拟并发操作 for i := 0; i < 10; i++ { wg.Add(1) go func() { // 并发操作的逻辑 wg.Done() }() } wg.Wait() 在上面的例子中，我们通过循环增加了10个并发操作，而不需要提前知道它们的数量。每个并发操作在完成后都会调用WaitGroup的Done方法来标记自己的完成，最后我们通过Wait方法等待所有的操作完成。
- 使用WaitGroup的Wait方法设置超时
有时候，我们希望在一段时间内等待并发操作的完成，如果超过了这个时间仍然没有完成，就不再等待。这时，我们可以使用WaitGroup的Wait方法结合select语句来实现超时等待。例如：
var wg sync.WaitGroup // 模拟并发操作 for i := 0; i < 10; i++ { wg.Add(1) go func() { // 并发操作的逻辑 wg.Done() }() } timeout := 5 * time.Second done := make(chan struct{}) go func() { wg.Wait() close(done) }() select { case <-done: // 所有操作完成 case <-time.After(timeout): // 超时 } 在上面的例子中，我们使用一个额外的goroutine来执行Wait方法，并在等待完成后关闭一个通道done。然后，在主goroutine中通过select语句等待done通道的关闭或者超时，来决定是否继续等待。
- 使用WaitGroup的Wait方法实现并发任务的部分完成
有时候，我们希望在一组并发任务中，只要有其中一部分完成了，就可以继续执行后续的操作，而不需要等待全部任务完成。这时，我们可以使用WaitGroup的Wait方法结合channel来实现部分完成。例如：
var wg sync.WaitGroup // 模拟并发操作 for i := 0; i < 10; i++ { wg.Add(1) go func() { // 并发操作的逻辑 wg.Done() }() } done := make(chan struct{}) go func() { wg.Wait() close(done) }() // 部分完成的逻辑 select { case <-done: // 所有操作完成 default: // 部分操作完成 } 在上面的例子中，我们同样使用一个额外的goroutine来执行Wait方法，并在等待完成后关闭一个通道done。然后，在主goroutine中使用select语句判断done通道是否关闭，来决定是继续等待还是执行部分完成的逻辑。
总结：
WaitGroup的高级用法可以帮助我们更加灵活地处理并发操作的完成。通过动态增加等待的数量，设置超时，或者实现部分完成，我们可以更好地控制并发编程的流程和效率。在实际的并发编程中，根据具体的需求选择合适的WaitGroup的用法，可以使我们的代码更加可靠和高效。
