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

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

碧海醫心

发布： 2024-11-09 14:12:02

原创

652人浏览过

在 go 语言中，如何使用 channel 或 context 实现协程等待？

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

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

使用 channel

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

Softr Studio

最简单的无代码web开发平台

查看详情

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

登录后复制

以上就是在 Go 语言中，如何使用 Channel 或 Context 实现协程等待？的详细内容，更多请关注php中文网其它相关文章！