golang中等待所有goroutine执行完成.

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")
}

以上.

你可能感兴趣的:(golang中等待所有goroutine执行完成.)