让我们看看这个精心设计的例子:
package main
import "fmt"
func printElo() {
fmt.Printf("Elo\n")
}
func printHello() {
fmt.Printf("Hello\n")
}
func main() {
fmt.Printf("This will print.")
i := 0
for i < 10 {
go printElo()
go printHello()
i++
}
}
该程序的输出将只是"This will print".Goroutines printElo()和printHello的输出将不会发出，因为我猜main()函数线程将在Goroutines甚至有机会开始执行之前完成.
让类似的代码在Golang中工作而不过早终止的惯用方法是什么？
要做到这一点，最简单、最干净且"可伸缩"的方法是使用sync.WaitGroup:
var wg = &sync.WaitGroup{}
func printElo() {
defer wg.Done()
fmt.Printf("Elo\n")
}
func printHello() {
defer wg.Done()
fmt.Printf("Hello\n")
}
func main() {
fmt.Printf("This will print.")
i := 0
for i < 10 {
wg.Add(1)
go printElo()
wg.Add(1)
go printHello()
i++
}
wg.Wait()
}
输出(在Go Playground上试用):
This will print.Hello
Elo
Hello
Elo
Hello
Elo
Hello
Elo
Hello
Elo
Hello
Elo
Hello
Elo
Hello
Elo
Hello
Elo
Hello
Elo
使用sync.WaitGroup时要遵循的简单"规则":
go语句之前，在"原始"goroutine(开始一个新的)中调用WaitGroup.Add()WaitGroup.Done()，因此即使goroutinepanic 也会拨打WaitGroup传递给其他函数(而不使用包级别变量)，必须传递指向它的指针，否则WaitGroup( struct )将被复制，并且在副本上调用的Done()方法在原始函数上不会被观察到