登录
首页 >  Golang >  Go教程

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

时间:2024-11-08 17:37:01 181浏览 收藏

最近发现不少小伙伴都对Golang很感兴趣,所以今天继续给大家介绍Golang相关的知识,本文《Go 语言中如何利用协程实现等待机制?》主要内容涉及到等等知识点,希望能帮到你!当然如果阅读本文时存在不同想法,可以在评论中表达,但是请勿使用过激的措辞~

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学习网公众号。

相关阅读
更多>
最新阅读
更多>
课程推荐
更多>