Go中使用Channel等待所有协程结束

最新推荐文章于 2024-07-21 13:35:18 发布

路上走着

最新推荐文章于 2024-07-21 13:35:18 发布

阅读量2.7k

点赞数

本文链接：https://blog.csdn.net/zengchen73/article/details/81088831

版权

本文介绍了如何在Go中使用Channel确保所有协程执行完毕后再退出。通过成对的Channel通信，实现了按顺序、批量处理和完全随机的协程控制方式，并特别讨论了利用Channel进行树遍历的应用场景。

摘要由CSDN通过智能技术生成

让main方法等待所有协程执行完毕再退出。可能一般思路是设置一个共有变量，然后通过修改这个变量的状态。这是通过共享变量来通信的方式，而go要做的是，通过通信来共享内存。

1. 按顺序执行

每次通信进行成对通信，当main向协程发送一个写channel时，同时也等待协程返回一个读channel。

这两个channel一定是成对的，所以构造一个结构体


type worker struct {
    in chan int
    done chan bool
}

func chanDemo1(){
    var workers [10]worker

    for i := 0; i < 10; i++ {
        workers[i] = createWorker1(i)
    }

    for i := 0; i < 10; i++ {
        workers[i].in <- 'a' + i
        <- workers[i].done
    }

    for i := 0; i < 10; i++ {
        workers[i].in <- 'A' + i
        <- workers[i].done
    }

}

func createWorker1(id int) worker {
    work := worker{
        in: make(chan int),
        done: make(chan bool),
    }
    go func() {
        for {
            fmt.Printf("Work %d receiverd %c\n", id, <- work.in)