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可以有效地等待一组并发任务完成,确保在所有任务完成之前不会提前退出主程序。