WaitGroup是Go语言中的一个并发同步原语,用于等待一组goroutine的完成。它提供了三个主要的方法:Add、Done和Wait。
Add方法用于向WaitGroup中添加要等待的goroutine的数量。Done方法用于表示一个goroutine的完成,即从WaitGroup中减去一个等待的goroutine。Wait方法会阻塞调用它的goroutine,直到所有的等待的goroutine都完成。
WaitGroup的内部实现主要涉及一个计数器和一个条件变量。计数器用于记录有多少个goroutine还未完成,条件变量用于实现goroutine的阻塞和唤醒。
在WaitGroup的初始化过程中,计数器被设置为0,并且条件变量也被初始化。每次调用Add方法时,计数器会增加指定的数量。每次调用Done方法时,计数器会减少1。在调用Wait方法时,如果计数器大于0,则当前的goroutine会被阻塞,并等待条件变量的通知。当计数器减少到0时,所有等待的goroutine都会被唤醒,继续执行后续的代码。
需要注意的是,WaitGroup的计数器是一个有符号的整数,可以为负值。这是因为在某些情况下,可能会调用Add方法增加计数器的数量,而不调用Done方法减少计数器的数量。这种情况下,计数器可能会变成负值。当计数器为负值时,Wait方法会立即返回,而不会阻塞当前的goroutine。
总之,WaitGroup通过计数器和条件变量的组合,实现了一种简单而高效的等待一组goroutine的完成的机制。它是Go语言中常用的并发同步原语之一,可以方便地控制多个goroutine的并发执行。