下面代码的输出结果是什么?
package main
import (
"fmt"
)
func main() {
for i := 0; i < 10; i++ {
go func() {
fmt.Println(i)
}()
}
}
运行该代码输出的结果是什么? 可能有以下答案
- 大概率什么都没有输出
原因是: 只要主协程也就是 main 进程执行完毕,其它的 Go 协程就会立即结束,主协程也就是 Go 程序是不会等待其它 Go 协程的,
而此时由于 for 循环启动的 Go 协程完全有可能还有运行,从而导致任何没有输出任何内容。
- 小概率有可能是 10 个 10
原因是: 假如 for 循坏启动的 10 个 协程在 main 协程执行完毕之前已运行,由于协程的调度需要一定的时间,而通常这个时间是大于
for 循坏的时间的,所以等 10 个协程启动的时候,for 循环已经执行到最后的变量 i=10 了,所以就会打印 10 个 10。
- 小概率有可能是乱序的 1 到 9 等
原因: 如果 Go 协程启动的时间较快,在 for 循环每执行到一个变量时该协程就会立即启动,那么就会打印 1 到 9 不等,最终原因
还是与 Go 协程的调度机制有关。
那么问题来了,如果想要让子协程执行完毕后退出,有什么办法呢?
我们可以使用 sync.WaitGroup
来实现,请看下面代码
func main() {
var wg sync.WaitGroup
for i := 0; i < 10; i++ {
wg.Add(1) // 每启动一个协程就加 1
go func() {
defer wg.Done() // 协程退出时自动减 1
fmt.Println(i)
}()
}
wg.Wait() // 等待所有协程执行完毕
}
输出结果为:
10
10
10
10
10
10
10
10
10
10
如果想要让其打印出 0,1,2,3,4 … 等,需要怎样修改代码呢?
只需要在调用协程函数的时候传入对应的变量就行,如下代码:
func main() {
var wg sync.WaitGroup
for i := 0; i < 10; i++ {
wg.Add(1)
go func(n int) {
defer wg.Done()
fmt.Println(n)
}(i) // 传入外面循环变量的值
}
wg.Wait()
}