
如何使用 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
}
}









