如何通过 channel 或 context 在 go 中实现协程等待
在 go 语言中实现协程等待共有两种常见方法：使用 channel 和使用 context。
使用 channel
channel 是一种轻量级通信机制，可用于在协程之间传递数据和信号。可以使用 channel 来实现协程等待，方法如下：
package main import ( "fmt" "math/rand" "sync" "time" ) func main() { // 创建一个可缓冲 channel，其中缓冲量为协程数量。 ch := make(chan int, 10) // 启动 10 个协程。 for i := 0; i < 10; i++ { go func(ch chan int, key int) { fmt.println("goroutine start:", key) t := rand.intn(5) + 1 time.sleep(time.duration(t) * time.second) fmt.println("goroutine end:", key) ch <- key // 发送协程 id，表示协程已完成。 }(ch, i) } fmt.println("main start") // 监听 channel，等待所有协程完成。 for i := 0; i < 10; i++ { key := <-ch // 从 channel 接收协程 id。 fmt.println("main get:", key) } fmt.println("main end") }
登录后复制
使用 context
context 是 go 语言中用于处理协程生命周期和取消的机制。context 可以用来实现协程等待，方法如下：
package main import ( "context" "fmt" "math/rand" "sync" "time" ) func main() { var wg sync.WaitGroup // 使用 WaitGroup 计数未完成的协程。 // 创建一个 context 并将其传递给协程。 ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() // 启动 10 个协程。 for i := 0; i < 10; i++ { wg.Add(1) go func(ctx context.Context, key int) { defer wg.Done() // 减少计数。 fmt.Println("goroutine start:", key) t := rand.Intn(5) + 1 time.Sleep(time.Duration(t) * time.Second) fmt.Println("goroutine end:", key) }(ctx, i) } fmt.Println("main start") // 等待所有协程完成或超时。 wg.Wait() fmt.Println("main end") }
登录后复制
以上就是在 Go 语言中，如何使用 Channel 或 Context 实现协程等待？的详细内容，更多请关注php中文网其它相关文章！
微信扫一扫打赏
支付宝扫一扫打赏
