Go里面sync.WaitGroup指针引用问题

最新推荐文章于 2024-12-10 11:45:26 发布

weixin_33967071

最新推荐文章于 2024-12-10 11:45:26 发布

阅读量570

点赞数 1

CC 4.0 BY-SA版权

文章标签： golang python

原文链接：https://my.oschina.net/Kanonpy/blog/2054788

2019独角兽企业重金招聘Python工程师标准>>>

WaitGroup：主要包括Add,Done,Wait三个方法，Add表示添加一个goroutine，Done等于Add(-1)，表示一个goroutine结束，wait表示主线程一直等到所有的goroutine执行完成，并且阻塞主线程的执行，直到所有的goroutine执行完成.

但在使用的时候发现进程一直阻塞，代码如下：

func Exec(url string, wg sync.WaitGroup) error{
	statusCode, body, err := fasthttp.Get(nil, url)
	if err != nil{
		log.Fatalf("err,%s", err)
		return err
	}
	if statusCode != fasthttp.StatusOK {
		log.Fatalf("Unexpected status code: %d")
		return err
	}
	ParseBody(body)
	wg.Done()
	return nil
}

func main(){
	var URL string
	URL = "http://192.168.2.112:61208/"
	num := 100
	var wg sync.WaitGroup
	wg.Add(num)
	startTime := time.Now()
	for i:=0; i<num; i++{
		go Exec(URL, wg)
	}
	wg.Wait()
	endTime := time.Now()
	spendTime := endTime.Sub(startTime)
	log.Println(spendTime)
}

将sync.WaitGroup放在main中做done就可以正常运行：

func main(){
	var URL string
	URL = "http://192.168.2.112:61208/"
	num := 100
	var wg sync.WaitGroup
	wg.Add(num)
	startTime := time.Now()
	for i:=0; i<num; i++{
		go func(){
			statusCode, body, err := fasthttp.Get(nil, url)
			if err != nil{
				log.Fatalf("err,%s", err)
			}
			if statusCode != fasthttp.StatusOK {
				log.Fatalf("Unexpected status code: %d")
			}
			ParseBody(body)
			wg.Done() 
		}
	}
	wg.Wait()
	endTime := time.Now()
	spendTime := endTime.Sub(startTime)
	log.Println(spendTime)
}

一直阻塞应该是死锁了，也就是wg.Done()没起作用，原来是wg 给拷贝传递到了 goroutine 中，导致只有 Add 操作，其实 Done操作是在 wg 的副本执行导致的~~ 将wg 的传入类型改为 *sync.WaitGrou,这样就能引用到正确的WaitGroup了

func Exec(url string, wg *sync.WaitGroup) error{
	statusCode, body, err := fasthttp.Get(nil, url)
	if err != nil{
		log.Fatalf("err,%s", err)
		return err
	}
	if statusCode != fasthttp.StatusOK {
		log.Fatalf("Unexpected status code: %d")
		return err
	}
	ParseBody(body)
	wg.Done()
	return nil
}

func main(){
	var URL string
	URL = "http://192.168.2.112:61208/"
	num := 100
	var wg sync.WaitGroup
	wg.Add(num)
	startTime := time.Now()
	for i:=0; i<num; i++{
		go Exec(URL, &wg)
	}
	wg.Wait()
	endTime := time.Now()
	spendTime := endTime.Sub(startTime)
	log.Println(spendTime)
}

转载于:https://my.oschina.net/Kanonpy/blog/2054788