详解如何在Golang中监听多个channel

更新时间：2023年03月16日 10:54:45 作者：洛天枫

这篇文章主要为大家详细介绍了如何在Golang中实现监听多个channel，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起学习一下

select 关键字

我们可以使用select关键字来同时监听多个goroutine。

package main

import (
    "fmt"
    "time"
)

func main() {
    c1 := make(chan string)
    c2 := make(chan string)

    go func() {
        time.Sleep(1 * time.Second)
        c1 <- time.Now().String()
    }()

    go func() {
        time.Sleep(2 * time.Second)
        c2 <- time.Now().String()
    }()

    for i := 0; i < 2; i++ {
        select {
        case res1 := <-c1:
            fmt.Println("from c1:", res1)
        case res2 := <-c2:
            fmt.Println("from c2:", res2)
        }
    }
}

from c1: 2022-09-04 14:30:39.4469184 -0400 EDT m=+1.000172801
from c2: 2022-09-04 14:30:40.4472699 -0400 EDT m=+2.000524401

上面的代码显示了select关键字是如何工作的：

我们首先创建两个 channel c1和c2来监听。
然后我们生成两个goroutine，分别向c1和c2发送当前时间。
在for循环中，我们创建了一个select语句，并定义了两个 case：第一个 case 是我们何时可以从c1接收，第二个 case 是我们何时可以从c2接收。

你可以看到select语句和switch语句在设计上非常相似。两者都定义了不同的情况，并在满足某种情况时运行相应的代码。另外，我们可以看到select语句是阻塞的。也就是说，它将等待，直到其中一个 case 被满足。

我们为这个循环迭代了两次，因为只有两个goroutine需要监听。更确切地说，每个goroutine都是一个fire-and-forget goroutine，意味着它们在返回之前只向一个通道发送一次。因此，在这段代码中，任何时候都有两个消息的最大值，而我们只需要选择两次。

如果我们不知道工作何时会结束呢

有时我们不知道有多少个工作。在这种情况下，把 select 语句放在一个while循环里。

package main

import (
    "fmt"
    "math/rand"
    "time"
)

func main() {
    c1 := make(chan string)
    rand.Seed(time.Now().UnixNano())

    for i := 0; i < rand.Intn(10); i++ {
        go func() {
            time.Sleep(1 * time.Second)
            c1 <- time.Now().String()
        }()
    }

    for {
        select {
        case res1 := <-c1:
            fmt.Println("from c1:", res1)
        }
    }
}

因为我们让一个随机数的goroutines运行，所以我们不知道有多少个作业。值得庆幸的是，底部包裹着select语句的for循环将捕获每一个输出。让我们看看如果我们运行这段代码会发生什么。

from c1: 2022-09-04 14:48:47.5145341 -0400 EDT m=+1.000257801
from c1: 2022-09-04 14:48:47.5146126 -0400 EDT m=+1.000336201
from c1: 2022-09-04 14:48:47.5146364 -0400 EDT m=+1.000359901
fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan receive]:
main.main()
/home/jacob/blog/testing/listening-to-multiple-channels-in-go/main.go:22 +0x128
exit status 2

嗯，select 语句按预期收到了三次，但由于死锁，程序出错了。为什么会出现这种情况？

记住，在没有任何条件的情况下，Go中的for循环会永远运行。这意味着 select 语句将永远尝试接收。然而，要运行的作业数量是有限的。即使没有更多的工作，select 语句仍然会尝试接收。

还记得在本系列的第一篇文章中我说过，如果你在发送方没有准备好的时候试图从一个没有缓冲的通道中接收，你的程序就会陷入死锁。这正是我们例子中的情况。

那么我们如何解决这个问题呢？我们可以使用以前文章中涉及的概念的组合：退出通道和WaitGroups。

package main

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

func main() {
    c1 := make(chan string)
    exit := make(chan struct{})
    rand.Seed(time.Now().UnixNano())
    var wg sync.WaitGroup

    go func() {
        numJob := rand.Intn(10)
        fmt.Println("number of jobs:", numJob)
        for i := 0; i < numJob; i++ {
            wg.Add(1)
            go func() {
                defer wg.Done()
                time.Sleep(1 * time.Second)
                c1 <- time.Now().String()
            }()
        }
        wg.Wait()
        close(exit)
    }()

    for {
        select {
        case res1 := <-c1:
            fmt.Println("from c1:", res1)
        case <-exit:
            return
        }
    }
}

3
from c1: 2022-09-04 15:09:08.6936976 -0400 EDT m=+1.000287801
from c1: 2022-09-04 15:09:08.6937788 -0400 EDT m=+1.000369101
from c1: 2022-09-04 15:09:08.6937949 -0400 EDT m=+1.000385101

我们创建一个 exit channel 和一个WaitGroup。
每次运行的作业数都是随机的。对于numJobs的次数，我们启动goroutines。为了等待作业的完成，我们把它们加到wg中。当一个工作完成后，我们从wg中减去一个。
一旦所有工作完成，我们就关闭exit channel。
我们将上述部分包裹在一个goroutine中，因为我们希望所有的部分都是无阻塞的。如果我们不把它包在一个goroutine中，wg.Wait()就会等待，直到作业完成。这将阻塞代码，并且不会让底部的for-select语句运行。
此外，由于c1是一个未缓冲的通道，等待所有的goroutine将消息发送到c1，将导致许多消息被发送到c1而没有for-select语句接收。这就导致了一个死锁，因为当发送者准备好时，接收者还没有准备好。

如何使select成为非阻塞式的

select语句默认是阻塞的。我们如何使其成为非阻塞的呢？这很简单--我们只需添加一个 default case。

package main

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

func main() {
    ashleyMsg := make(chan string)
    brianMsg := make(chan string)
    exit := make(chan struct{})
    rand.Seed(time.Now().UnixNano())
    var wg sync.WaitGroup

    go func() {
        numJob := rand.Intn(10)
        fmt.Println("number of jobs:", numJob)
        for i := 0; i < numJob; i++ {
            wg.Add(2)
            go func() {
                defer wg.Done()
                time.Sleep(time.Duration(rand.Intn(10)) * time.Millisecond)
                ashleyMsg <- "hi"
            }()
            go func() {
                defer wg.Done()
                time.Sleep(time.Duration(rand.Intn(10)) * time.Millisecond)
                brianMsg <- "what's up"
            }()
        }
        wg.Wait()
        close(exit)
    }()

    for {
        select {
        case res1 := <-ashleyMsg:
            fmt.Println("ashley:", res1)
        case res2 := <-brianMsg:
            fmt.Println("brian:", res2)
        case <-exit:
            fmt.Println("chat ended")
            return
        default:
            fmt.Println("...")
            time.Sleep(time.Millisecond)
        }
    }
}

...
number of jobs: 4
brian: what's up
...
ashley: hi
...
...
brian: what's up
ashley: hi
ashley: hi
brian: what's up
...
...
ashley: hi
...
brian: what's up
...
chat ended

除了蹩脚的对话之外，我们可以看到默认情况下的工作方式。与其等待聊天记录的到来，我们可以在没有频道可以接收的时候做一些事情。在这个例子中，我们只是打印出省略号，但你可以做任何你想做的事情。

到此这篇关于详解如何在Golang中监听多个channel的文章就介绍到这了,更多相关Golang监听多个channel内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Go语言实现UDP版聊天小工具的示例详解
这篇文章主要为大家详细介绍了如何利用Go语言实现聊天小工具（UDP版），文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2023-03-03
golang中之strconv包的具体使用方法
这篇文章主要介绍了golang中之strconv包的具体使用方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-10-10
Go语言Gin框架实现HTML页面渲染
Web开发中，我们经常要面对如何将数据渲染到前端的问题，这就涉及到了模板引擎的知识，Go语言的Gin框架就提供了强大的HTML模板渲染功能,本文就来为大家介绍，有需要的朋友可以借鉴参考下,希望能够有所帮助
2024-01-01
go语言实现http服务端与客户端的例子
今天小编就为大家分享一篇go语言实现http服务端与客户端的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-08-08
解决vscode中golang插件依赖安装失败问题
这篇文章主要介绍了解决vscode中golang插件依赖安装失败问题,本文给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下
2019-08-08
Go语言net包RPC远程调用三种方式http与json-rpc及tcp
这篇文章主要为大家介绍了Go语言net包RPC远程调用三种方式分别使用http与json-rpc及tcp的示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助
2021-11-11
深入理解Go中的项目代码布局
Go 语言的创世项目其实就是 Go 语言项目自身,是全世界第一个 Go 语言项目,本文主要来和大家详细介绍一下Go中的项目代码布局,需要的可以了解下
2023-11-11
golang官方嵌入文件到可执行程序的示例详解
这篇文章主要介绍了golang官方嵌入文件到可执行程序,本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-02-02
Go语言实现简单留言板的方法
这篇文章主要介绍了Go语言实现简单留言板的方法,涉及数据库、模板页面元素等留言板相关技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-02-02
Golang使用Gin框架实现http分块传输
这篇文章主要为大家详细介绍了Golang中如何使用Gin框架实现http分块传输功能，文中的示例代码讲解详细，具有一定的学习价值，需要的可以参考一下
2023-05-05