WaitGroup | 青训营笔记

88 阅读1分钟

WaitGroup

sync.WaitGroup是一种用于协调并发任务的机制。WaitGroup是一个结构体,它可以跟踪一组并发任务的完成情况。

type WaitGroup struct {
	// contains filtered or unexported fields
}

WaitGroup提供了三个主要的方法:

  • Add(delta int):用于指定等待的任务数量。delta参数表示需要等待的任务数量的增量,可以是负数。当调用Add方法时,WaitGroup内部的计数器会增加相应的值。
func (wg *WaitGroup) Add(delta int)
  • Done():用于标记一个任务已经完成。每次调用Done方法时,WaitGroup内部的计数器会减1。
func (wg *WaitGroup) Done()
  • Wait():用于阻塞当前的goroutine,直到计数器归零。当调用Wait方法时,如果计数器的值大于0,则会阻塞当前goroutine,直到计数器的值变为0。一旦计数器的值为0,Wait方法会立即返回,当前的goroutine可以继续执行后续的操作。
func (wg *WaitGroup) Wait()

示例

用一个简单的例子示范WaitGroup

package main

import (
	"fmt"
	"sync"
)

func main() {
	var wg sync.WaitGroup
	for i := 0; i < 5; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			fmt.Println("Task", i, "is done")
		}(i)
	}
	wg.Wait()
	fmt.Println("All tasks are done")
}

在上面的示例中,首先创建了一个WaitGroup实例wg。然后,在一个循环中创建了5个goroutine,每个goroutine执行一个任务。在每个goroutine的匿名函数中,调用了Done方法来标记任务的完成。最后,调用了Wait方法来等待所有任务的完成。

使用WaitGroup可以有效地等待一组并发任务完成,确保在所有任务完成之前不会提前退出主程序。