golang sync.WaitGroup使用

WaitGroup用于等待一组线程的结束。父线程调用Add方法来设定应等待的线程的数量,每一次执行Add都会增加线程组的数量。每个被等待的线程在结束时应调用Done方法。同时,主线程里可以调用Wait方法阻塞至所有线程结束。

方法介绍
  1. func (*WaitGroup) Add(delta int)
    Add方法向内部计数加上delta,delta可以是负数;如果内部计数器变为0,Wait方法阻塞等待的所有线程都会释放,如果计数器小于0,方法panic。注意Add加上正数的调用应在Wait之前,否则Wait可能只会等待很少的线程。一般来说本方法应在创建新的线程或者其他应等待的事件之前调用。

  2. func (*WaitGroup) Done
    Done方法减少WaitGroup计数器的值,应在线程的最后执行,Done的执行应标志着一个goroutine的结束

  3. func (*WaitGroup) Wait
    Wait方法阻塞直到WaitGroup计数器减为0。如果WaitGroup不为0,那么程序就会一直阻塞在Wait函数这里

代码示例
package main

import (
	"fmt"
	"sync"
	"net/http"
)

func main() {
	var wg sync.WaitGroup
	var urls = []string{
		"http://www.baidu.com",
		"http://www.github.com/",
		"https://golang.google.cn/",
	}
	for _, url := range urls {
		// 增加WaitGroup的数量
		wg.Add(1)
		// 分发一个goroutine请求对应的url
		go func(url string) {
			// 当这个goroutine完成的时候,减少WaitGroup的数量
			defer wg.Done()
			// Fetch the URL.
			res, err := http.Get(url)
			if err != nil{
				fmt.Println(err)
				return
			}
			fmt.Println(res.StatusCode)
		}(url)
	}
	// 等待至所有的请求都完成
	wg.Wait()
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值