除了使用通道channel和互斥锁sync.Mutex进行两个并发程序间的同步,我们还可以使用等待组(sync.WaitGroup)进行多个任务之间的同步。
方法名 | 功能 |
---|---|
(wg *WaitGroup)Add(delta int) | 等待组的计数器+1 |
(wg *WaitGroup)Done() | 等待组的计数器-1 |
(wg *WaitGroup)Wait() | 当等待组计数器不等于0时,阻塞直到变0 |
等待组sync.WaitGroup内部有一个计数器,我们可以通过方法调用来实现计数器值的增加和减少:当我们启动N个并发任务进行工作时,就把等待组的计数器值增加N,每完成一个任务,这个值就减1;同时在另一个goroutine中,当等待组的计数器值为0时,表示所有并发任务已经完成。
package main
import (
"fmt"
"net/http"
"sync"
"testing"
)
func main() {
var wg sync.WaitGroup // 声明一个等待组。一组等待任务只需要一个等待组,而不是每个任务都需要一个等待组。
urls := []string{ // 准备一个网站的字符串切片
"http://www.github.com/",
"https://www.github.com/",
"https://www.golangtc.com",
}
// 遍历字符串切片中的地址
for _, url := range urls {
wg.Add(1) // 每个并发任务开始时,把等待组加1
// 启动一个匿名函数的并发任务
go func(url string) {
defer wg.Done() // 每个并发的匿名函数结束时,都会执行这一句,表示任务完成。等效于wg.Add(-1)
_, err := http.Get(url) // 使用Get函数访问url,这个操作会一直阻塞(持续访问),直到网站响应或超时
fmt.Println(url, err) // 在网站响应或超时后,打印这个网站的地址或错误
}(url) // 通过参数url入参,为了避免url变量通过闭包引用后被修改的问题
wg.Wait() // 等待所有任务完成后,wg.Wait()就会停止阻塞。
fmt.Println("over")
}
}
/*
http://www.github.com/ <nil>
over
https://www.github.com/ <nil>
over
https://www.golangtc.com <nil>
over
*/