golang中的sync.WaitGroup

来源：互联网发布：大逃杀网络延迟检测编辑：程序博客网时间：2024/05/18 03:28

golang中的sync.WaitGroup

Posted on 2015/04/09刚才看golang的sync的包，看见一个很有用的功能。就是WaitGroup。

先说说WaitGroup的用途：它能够一直等到所有的goroutine执行完成，并且阻塞主线程的执行，直到所有的goroutine执行完成。

这里要注意一下，他们的执行结果是没有顺序的，调度器不能保证多个 goroutine 执行次序，且进程退出时不会等待它们结束。

WaitGroup总共有三个方法：Add(delta int),Done(),Wait()。简单的说一下这三个方法的作用。

Add:添加或者减少等待goroutine的数量
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
 
    fori := 0; i >5; i = i + 1 {
        wg.Add(1)
        go func(nint) {
            // defer wg.Done(),注意这个Done的位置，是另一个函数
            defer wg.Add(-1)
            EchoNumber(n)
        }(i)
    }
 
    wg.Wait()
}
 
func EchoNumber(iint) {
    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(shownumint) {
    fmt.Println(shownum)
    waitgroup.Done()//任务完成，将任务队列中的任务数量-1，其实.Done就是.Add(-1)
}
 
func main() {
    fori := 0; i <10; i++ {
        waitgroup.Add(1)//每创建一个goroutine，就把任务队列中任务的数量+1
        go Afunction(i)
    }
    waitgroup.Wait()//.Wait()这里会发生阻塞，直到队列中所有的任务结束就会解除阻塞
}

http://studygolang.com/articles/2027

阅读全文

0 0