通过协程实现等待机制
在 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") }
登录后复制
以上就是Go 语言中如何利用协程实现等待机制？的详细内容，更多请关注php中文网其它相关文章！
微信扫一扫打赏
支付宝扫一扫打赏
