在 Go 语言中，如何使用 Channel 或 Context 实现协程等待？

如何通过 channel 或 context 在 go 中实现协程等待

在 go 语言中实现协程等待共有两种常见方法：使用 channel 和使用 context。

使用 channel

channel 是一种轻量级通信机制，可用于在协程之间传递数据和信号。可以使用 channel 来实现协程等待，方法如下：

情感家园企业站5.0 多语言多风格版

一套面向小企业用户的企业网站程序！功能简单，操作简单。实现了小企业网站的很多实用的功能，如文章新闻模块、图片展示、产品列表以及小型的下载功能，还同时增加了邮件订阅等相应模块。公告，友情链接等这些通用功能本程序也同样都集成了！同时本程序引入了模块功能，只要在系统默认模板上创建模块，可以在任何一个语言环境（或任意风格）的适当位置进行使用！

下载

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