Go采坑记: sync.WaitGroup 指针引用问题
Go采坑记: sync.WaitGroup 指针引用问题
机械视角
发布于 2019-10-23 11:23:05
发布于 2019-10-23 11:23:05
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)
}本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2018-09-18，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读