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

除了使用通道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
*/
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值