在Golang中,WaitGroup是一种用于等待一组goroutine完成执行的机制。它可以用来确保在主goroutine结束之前,所有的子goroutine都已经执行完毕。
要使用WaitGroup,首先需要导入`sync`包。然后,可以创建一个WaitGroup对象,并使用`Add`方法来设置需要等待的goroutine数量。接下来,在每个子goroutine的末尾使用`Done`方法来标记一个子goroutine的结束。最后,在主goroutine中使用`Wait`方法来等待所有的子goroutine完成执行。
下面是一个使用WaitGroup的简单示例:
```go
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("Goroutine", i, "is done")
}(i)
}
wg.Wait()
fmt.Println("All goroutines finished executing")
}
```
在上面的示例中,我们创建了一个包含5个子goroutine的循环。每个子goroutine都会输出一个简单的消息,并在结束时调用`Done`方法。最后,我们在主goroutine中调用`Wait`方法来等待所有的子goroutine完成执行。
运行上面的代码,你会看到类似以下的输出:
```
Goroutine 0 is done
Goroutine 1 is done
Goroutine 3 is done
Goroutine 4 is done
Goroutine 2 is done
All goroutines finished executing
```
可以看到,所有的子goroutine都已经执行完毕,然后主goroutine输出了"所有goroutine都已经执行完毕"的消息。
总结来说,WaitGroup是一种在Golang中等待一组goroutine完成执行的机制。通过调用`Add`方法设置需要等待的goroutine数量,调用`Done`方法标记一个goroutine的结束,以及调用`Wait`方法来等待所有的goroutine完成执行。这是在并发编程中非常常用的一种技术。