通过协程实现等待机制
在 go 语言中，可以通过 channel 或 context 实现协程之间的等待。
channel 方式
创建一个带有缓冲区的 channel，子协程在完成任务后向 channel 发送消息，主协程从 channel 接收消息等待子协程完成。
package main
import (
"fmt"
"time"
)
func main() {
ch := make(chan int, 10)
for i := 0; i < 10; i++ {
go func(ch chan int, key int) {
fmt.println("goroutine start: ", key)
time.sleep(time.duration(key) * time.second)
fmt.println("goroutine end: ", key)
ch <- key
}(ch, i)
}
fmt.println("main start")
for i := 0; i < 10; i++ {
key := <-ch
fmt.println("main get: ", key)
}
fmt.println("main end")
}context 方式
使用 context.context 来传播取消信号，当父 context 被取消时，所有关联的子协程也会自动取消。
package main
import (
"context"
"fmt"
"time"
)
func main() {
ctx, cancel := context.WithCancel(context.Background())
for i := 0; i < 10; i++ {
go func(ctx context.Context, key int) {
fmt.Println("goroutine start: ", key)
select {
case <-ctx.Done():
fmt.Println("goroutine canceled: ", key)
case <-time.After(time.Duration(key) * time.Second):
fmt.Println("goroutine end: ", key)
}
}(ctx, i)
}
fmt.Println("main start")
time.Sleep(time.Duration(3) * time.Second)
cancel()
fmt.Println("main end")
} 
