wait group 的用法
时间: 2024-02-18 12:37:52 浏览: 26
在 Go 语言中,`sync.WaitGroup` 可以用于等待一组操作完成后再继续执行下一步操作。它通常用于协调多个 goroutine 的执行。
`WaitGroup` 是一个结构体,它有三个方法:
- `Add(delta int)`:增加 WaitGroup 的计数器,delta 参数为要增加的计数器值。
- `Done()`:减少 WaitGroup 的计数器,相当于完成了一个 goroutine 的操作。
- `Wait()`:阻塞代码执行,直到 WaitGroup 的计数器归零。
下面是一个简单的示例代码,用于演示 `WaitGroup` 的用法:
```go
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
wg.Add(1)
go worker(i, &wg)
}
wg.Wait()
fmt.Println("All workers done")
}
```
在这个例子中,我们创建了 5 个 goroutine,并且每个 goroutine 都会睡眠 1 秒钟。`WaitGroup` 的初始计数器值为 0,每个 `worker` 函数调用 `wg.Add(1)` 后,计数器的值就会加 1。`wg.Wait()` 会一直等待,直到计数器的值归零,也就是所有的 goroutine 都执行完成了。
输出结果如下:
```
Worker 2 starting
Worker 1 starting
Worker 4 starting
Worker 3 starting
Worker 5 starting
Worker 1 done
Worker 3 done
Worker 2 done
Worker 5 done
Worker 4 done
All workers done
```
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)