如何使用 go 协程实现等待多个协程

为了实现主协程等待多个子协程执行完毕的效果，类似于 sync.waitgroup，可以使用 go 中的 channel 或 context。

使用 channel

通过创建一个固定容量的 channel，可以实现此功能。当每个子协程完成时，它会向这个 channel 发送一个值以表明完成。

AI舌诊面诊问诊医疗系统+诊疗商城系统源码

AI 舌诊面诊问诊医疗系统 + 诊疗商城系统源码是一套结合 AI 辅助诊断与商品转化的健康类程序，基于浩创云、ThinkPHP（PHP）开发，支持微信服务号与 H5 端，适合做中医舌诊面诊问答、健康建议推送以及配套诊疗商品的运营。

下载

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
    }
}