与Channel区别
Channel能够很好的帮助我们控制并发,但是在开发习惯上与显示的表达不太相同,所以在Go语言中可以利用sync包中的WaitGroup实现并发控制,更加直观。
基本使用示例
我们将之前的示例加以改造,引入sync.WaitGroup来实现并发控制。
- 首先我们在主函数中定义WaitGroup
var wg sync.WaitGroup
- 每执行一个任务,则调用Add()方法
wg.Add(1)
- 在主函数中我们利用Wait()方法等待并发结束
wg.Wait()
- 在调用的函数中,我们需要将WaitGroup以指针方式传入,否则将造成Deadlock
// 主函数内
go ready(5, &wg)
// 函数
func ready(s int, wg *sync.WaitGroup)
- 同时在函数执行完成后,调用wg.Done,我们使用defer实现
defer wg.Done()
完整代码
package main
import (
"fmt"
"sync"
"time"
)
func ready(s int, wg *sync.WaitGroup) {
defer