go语言学习笔记 — 进阶 — 并发编程（11）：同步sync，等待组（sync.WaitGroup）—— 保证在并发环境中完成指定数量的任务

Posted 2021-11-22 Locutus

除了使用通道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
*/