脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|shell|

服务器之家 - 脚本之家 - Golang - 详解如何在Golang中监听多个channel

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

2023-03-17 13:50洛天枫 Golang

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

select 关键字

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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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循环里。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
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内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://juejin.cn/post/7210944562222432312

延伸 · 阅读

精彩推荐
  • Golanggo web 处理表单的输入的说明

    go web 处理表单的输入的说明

    今天给大家普及go表单输入方面的知识点,整体代码分为前端页面和后端处理方法,通过代码给大家介绍的很详细,对大家的学习或工作具有一定的参考借...

    chengqiuming5872021-08-07
  • Golanggolang 实现两个结构体复制字段

    golang 实现两个结构体复制字段

    这篇文章主要介绍了golang 实现两个结构体复制字段,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    butterfly52113148962021-06-06
  • GolangGo语言切片前或中间插入项与内置copy()函数详解

    Go语言切片前或中间插入项与内置copy()函数详解

    这篇文章主要介绍了Go语言切片前或中间插入项与内置copy()函数详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    弃更内容请谨慎查看12072021-06-01
  • Golanggo语言代码生成器code generator使用示例介绍

    go语言代码生成器code generator使用示例介绍

    这篇文章主要为大家介绍了go语言代码生成器code generator的使用简单介绍,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加...

    油腻中年李大鹅8112022-10-08
  • GolangGo 内联优化让程序员爱不释手

    Go 内联优化让程序员爱不释手

    这篇文章主要介绍了Go 内联优化让程序员爱不释手,内联是在编译过程中自动进行的一类基本优化之一,文章围绕主题展开更多详细介绍,具有一定的参考...

    煎鱼eddycjy10352022-10-24
  • Golanggo goroutine 怎样进行错误处理

    go goroutine 怎样进行错误处理

    在 Go 语言程序开发中,goroutine 的使用是比较频繁的,因此在日常编码的时候 goroutine 里的错误处理,怎么做会比较好呢,本文就来详细介绍一下...

    mb60e703e6a88976212021-08-16
  • GolangGolang加密解密之RSA(附带php)

    Golang加密解密之RSA(附带php)

    安全总是很重要的,各个语言对于通用的加密算法都会有实现。本文先是对RSA算法进行了简单介绍,后才进行介绍如何用Go实现RSA的加密解密,下面一起来...

    daisy2992020-05-01
  • GolangGo语言入门exec的基本使用示例

    Go语言入门exec的基本使用示例

    这篇文章主要为大家介绍了Go语言入门exec在go语言中的基本使用示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪...

    宇宙之一粟3492022-10-09