如何使用 go 协程实现等待多个协程
为了实现主协程等待多个子协程执行完毕的效果，类似于 sync.waitgroup，可以使用 go 中的 channel 或 context。
使用 channel
通过创建一个固定容量的 channel，可以实现此功能。当每个子协程完成时，它会向这个 channel 发送一个值以表明完成。
num := 10
ch := make(chan int, num)
for i := 0; i < num; i++ {
go func(ch chan int, key int) {
// ...
ch <- key
}(ch, i)
}
for i := 0; i < num; i++ {
key := <-ch
// ...
}使用 context
也可以使用 context 来实现等待多个协程。context 提供了一种取消和管理协程生命周期的机制。
ctx, cancel := context.withtimeout(context.background(), 10*time.second)
defer cancel()
for i := 0; i < num; i++ {
go func(ctx context.context, key int) {
// ...
}(ctx, i)
}
select {
case <-ctx.done():
// ...
}自定义 waitgroup
另一种方式是使用自定义的 waitgroup 结构体，它封装了 channel 和计数器。
type CustomWaitGroup struct {
ch chan int
num int
}
func NewCustomWaitGroup(num int) *CustomWaitGroup {
return &CustomWaitGroup{
ch: make(chan int, num),
num: num,
}
}
func (r *CustomWaitGroup) Done() {
r.ch <- 1
}
func (r *CustomWaitGroup) Wait() {
for i := 0; i < r.num; i++ {
<-r.ch
}
}以上就是Go 协程如何实现等待多个协程完成？的详细内容，更多请关注php中文网其它相关文章！
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移，垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是，许多工具可以让 Windows 保持平稳运行。
广告![ThinkPHP5快速开发企业站点[全程实录]](https://img.php.cn/upload/course/000/000/068/6253d918a3ce7278.png)
收藏
收藏
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
773
