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

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《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")
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。