Go 语言中如何利用协程实现等待机制？

通过协程实现等待机制

在 go 语言中，可以通过 channel 或 context 实现协程之间的等待。

channel 方式

创建一个带有缓冲区的 channel，子协程在完成任务后向 channel 发送消息，主协程从 channel 接收消息等待子协程完成。

package main

import (
    "fmt"
    "time"
)

func main() {
    ch := make(chan int, 10)

    for i := 0; i < 10; i++ {
        go func(ch chan int, key int) {
            fmt.println("goroutine start: ", key)
            time.sleep(time.duration(key) * time.second)
            fmt.println("goroutine end: ", key)
            ch <- key
        }(ch, i)
    }

    fmt.println("main start")
    for i := 0; i < 10; i++ {
        key := <-ch
        fmt.println("main get: ", key)
    }
    fmt.println("main end")
}

context 方式

使用 context.context 来传播取消信号，当父 context 被取消时，所有关联的子协程也会自动取消。

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    ctx, cancel := context.WithCancel(context.Background())

    for i := 0; i < 10; i++ {
        go func(ctx context.Context, key int) {
            fmt.Println("goroutine start: ", key)
            select {
            case <-ctx.Done():
                fmt.Println("goroutine canceled: ", key)
            case <-time.After(time.Duration(key) * time.Second):
                fmt.Println("goroutine end: ", key)
            }
        }(ctx, i)
    }

    fmt.Println("main start")
    time.Sleep(time.Duration(3) * time.Second)
    cancel()
    fmt.Println("main end")
}

以上就是Go 语言中如何利用协程实现等待机制？的详细内容，更多请关注其它相关文章！