golang中的sync.WaitGroup
先说说WaitGroup的用途：它能够一直等到所有的goroutine执行完成，并且阻塞主线程的执行，直到所有的goroutine执行完成。
这里要注意一下，他们的执行结果是没有顺序的，调度器不能保证多个 goroutine 执行次序，且进程退出时不会等待它们结束。
WaitGroup总共有三个方法：Add(delta int),Done(),Wait()。简单的说一下这三个方法的作用。
Done:相当于Add(-1)
Wait:执行阻塞，直到所有的WaitGroup数量变成0
如：
|
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
|
package
main
import
(
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
for
i :=
0
; i >
5
; i = i +
1
{
wg.Add(
1
)
go func(n
int
) {
// defer wg.Done(),注意这个Done的位置，是另一个函数
defer wg.Add(-
1
)
EchoNumber(n)
}(i)
}
wg.Wait()
}
func EchoNumber(i
int
) {
time.Sleep(3e9)
fmt.Println(i)
}
|
golang中的同步是通过sync.WaitGroup来实现的．WaitGroup的功能：它实现了一个类似队列的结构，可以一直向队列中添加任务，当任务完成后便从队列中删除，如果队列中的任务没有完全完成，可以通过Wait()函数来出发阻塞，防止程序继续进行，直到所有的队列任务都完成为止．
WaitGroup的特点是Wait()可以用来阻塞直到队列中的所有任务都完成时才解除阻塞，而不需要sleep一个固定的时间来等待．但是其缺点是无法指定固定的goroutine数目．但是其缺点是无法指定固定的goroutine数目．可能通过使用channel解决此问题。
另一个例子：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
package
main
import
(
"fmt"
"sync"
)
//声明一个全局变量
var waitgroup sync.WaitGroup
func Afunction(shownum
int
) {
fmt.Println(shownum)
}
func main() {
for
i :=
0
; i <
10
; i++ {
go Afunction(i)
}
waitgroup.Wait()
//.Wait()这里会发生阻塞，直到队列中所有的任务结束就会解除阻塞
}
|
