Go 协程如何实现等待多个协程完成？

一分耕耘，一分收获！既然都打开这篇《Go 协程如何实现等待多个协程完成？》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

如何使用 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 协程如何实现等待多个协程完成？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！