Go采坑记: sync.WaitGroup 指针引用问题

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

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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就可以正常运行:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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了

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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)
}
shikanon wechat
欢迎您扫一扫,订阅我滴↑↑↑的微信公众号!