golang中,创建一个channel并开始执行后便脱离了主程序控制,那么,类似于其他语言中, 多线程中的每个线程join到主线程,golang中 如何等待所有的 goroutine 都执行完毕?
查了下大概这个样子.
- 用 channel 进行同步
func main() {
ch := make(chan int, 2)
go func() {
for i := 0; i < 10; i++ {
time.Sleep(1 * time.Second)
fmt.Println("go routine1", i)
}
ch <- 1
}()
go func() {
for i := 0; i < 10; i++ {
time.Sleep(1 * time.Second)
fmt.Println("go routine2", i)
}
ch <- 2
}()
for i := 0; i < 2; i++ {
<-ch
}
fmt.Println("main exit")
}
- sync.WaitGroup
官方推荐方法,不需要知道 gotoutine 的个数
import "fmt"
import "time"
import "sync"
func main() {
var wg sync.WaitGroup
for i := 0; i < 10; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
time.Sleep(1 * time.Second)
fmt.Println(i)
}(i)
}
wg.Wait() // 等待
fmt.Println("main exit")
}
以上.
有疑问加站长微信联系(非本文作者)