[关闭]
@phper 2018-03-13T09:57:06.000000Z 字数 6671 阅读 3272

22.信道(channel)

Golang

原文:https://golangbot.com/channels/


欢迎来到Golang 系列教程的第 22 章。

在上一教程里,我们探讨了如何使用 Go 协程(Goroutine)来实现并发。我们接着在本教程里学习信道(Channel),学习如何通过信道来实现 Go 协程间的通信。

什么是信道?

信道可以想像成 Go 协程之间通信的管道。如同管道中的水会从一端流到另一端,通过使用信道,数据也可以从一端发送,在另一端接收。

信道的声明

所有信道都关联了一个类型。信道只能运输这种类型的数据,而运输其他类型的数据都是非法的。

chan T 表示 T 类型的信道。

信道的零值为 nil。信道的零值没有什么用,应该像对 map 和切片所做的那样,用 make 来定义信道。

下面编写代码,声明一个信道。

  1. package main
  2. import "fmt"
  3. func main() {
  4. var a chan int
  5. if a == nil {
  6. fmt.Println("channel a is nil, going to define it")
  7. a = make(chan int)
  8. fmt.Printf("Type of a is %T", a)
  9. }
  10. }

在线运行程序

由于信道的零值为 nil,在第 6 行,信道 a 的值就是 nil。于是,程序执行了 if 语句内的语句,定义了信道 a。程序中 a 是一个 int 类型的信道。该程序会输出:

  1. channel a is nil, going to define it
  2. Type of a is chan int

简短声明通常也是一种定义信道的简洁有效的方法。

  1. a := make(chan int)

这一行代码同样定义了一个 int 类型的信道 a

通过信道进行发送和接收

如下所示,该语法通过信道发送和接收数据。

  1. data := <- a // 读取信道 a
  2. a <- data // 写入信道 a

信道旁的箭头方向指定了是发送数据还是接收数据。

在第一行,箭头对于 a 来说是向外指的,因此我们读取了信道 a 的值,并把该值存储到变量 data

在第二行,箭头指向了 a,因此我们在把数据写入信道 a

发送与接收默认是阻塞的

发送与接收默认是阻塞的。这是什么意思?当把数据发送到信道时,程序控制会在发送数据的语句处发生阻塞,直到有其它 Go 协程从信道读取到数据,才会解除阻塞。与此类似,当读取信道的数据时,如果没有其它的协程把数据写入到这个信道,那么读取过程就会一直阻塞着。

信道的这种特性能够帮助 Go 协程之间进行高效的通信,不需要用到其他编程语言常见的显式锁或条件变量。

信道的代码示例

理论已经够了:)。接下来写点代码,看看协程之间通过信道是怎么通信的吧。

我们其实可以重写上章学习Go协程 时写的程序,现在我们在这里用上信道。

首先引用前面教程里的程序。

  1. package main
  2. import (
  3. "fmt"
  4. "time"
  5. )
  6. func hello() {
  7. fmt.Println("Hello world goroutine")
  8. }
  9. func main() {
  10. go hello()
  11. time.Sleep(1 * time.Second)
  12. fmt.Println("main function")
  13. }

在线运行程序

这是上一篇的代码。我们使用到了休眠,使 Go 主协程等待 hello 协程结束。如果你看不懂,建议你阅读上一教程Go 协程。

我们接下来使用信道来重写上面代码。

  1. package main
  2. import (
  3. "fmt"
  4. )
  5. func hello(done chan bool) {
  6. fmt.Println("Hello world goroutine")
  7. done <- true
  8. }
  9. func main() {
  10. done := make(chan bool)
  11. go hello(done)
  12. <-done
  13. fmt.Println("main function")
  14. }

在线运行程序

在上述程序里,我们在第 12 行创建了一个 bool 类型的信道 done,并把 done 作为参数传递给了 hello 协程。在第 14 行,我们通过信道 done 接收数据。这一行代码发生了阻塞,除非有协程向 done 写入数据,否则程序不会跳到下一行代码。于是,这就不需要用以前的 time.Sleep 来阻止 Go 主协程退出了。

<-done 这行代码通过协程(译注:原文笔误,信道)done 接收数据,但并没有使用数据或者把数据存储到变量中。这完全是合法的。

现在我们的 Go 主协程发生了阻塞,等待信道 done 发送的数据。该信道作为参数传递给了协程 hellohello 打印出 Hello world goroutine,接下来向 done 写入数据。当完成写入时,Go 主协程会通过信道 done 接收数据,于是它解除阻塞状态,打印出文本 main function

该程序输出如下:

  1. Hello world goroutine
  2. main function

我们稍微修改一下程序,在 hello 协程里加入休眠函数,以便更好地理解阻塞的概念。

  1. package main
  2. import (
  3. "fmt"
  4. "time"
  5. )
  6. func hello(done chan bool) {
  7. fmt.Println("hello go routine is going to sleep")
  8. time.Sleep(4 * time.Second)
  9. fmt.Println("hello go routine awake and going to write to done")
  10. done <- true
  11. }
  12. func main() {
  13. done := make(chan bool)
  14. fmt.Println("Main going to call hello go goroutine")
  15. go hello(done)
  16. <-done
  17. fmt.Println("Main received data")
  18. }

在线运行程序

在上面程序里,我们向 hello 函数里添加了 4 秒的休眠(第 10 行)。

程序首先会打印 Main going to call hello go goroutine。接着会开启 hello 协程,打印 hello go routine is going to sleep。打印完之后,hello 协程会休眠 4 秒钟,而在这期间,主协程会在 <-done 这一行发生阻塞,等待来自信道 done 的数据。4 秒钟之后,打印 hello go routine awake and going to write to done,接着再打印 Main received data

信道的另一个示例

我们再编写一个程序来更好地理解信道。该程序会计算一个数中每一位的平方和与立方和,然后把平方和与立方和相加并打印出来。

例如,如果输出是 123,该程序会如下计算输出:

  1. squares = (1 * 1) + (2 * 2) + (3 * 3)
  2. cubes = (1 * 1 * 1) + (2 * 2 * 2) + (3 * 3 * 3)
  3. output = squares + cubes = 49

我们会这样去构建程序:在一个单独的 Go 协程计算平方和,而在另一个协程计算立方和,最后在 Go 主协程把平方和与立方和相加。

  1. package main
  2. import (
  3. "fmt"
  4. )
  5. func calcSquares(number int, squareop chan int) {
  6. sum := 0
  7. for number != 0 {
  8. digit := number % 10
  9. sum += digit * digit
  10. number /= 10
  11. }
  12. squareop <- sum
  13. }
  14. func calcCubes(number int, cubeop chan int) {
  15. sum := 0
  16. for number != 0 {
  17. digit := number % 10
  18. sum += digit * digit * digit
  19. number /= 10
  20. }
  21. cubeop <- sum
  22. }
  23. func main() {
  24. number := 589
  25. sqrch := make(chan int)
  26. cubech := make(chan int)
  27. go calcSquares(number, sqrch)
  28. go calcCubes(number, cubech)
  29. squares, cubes := <-sqrch, <-cubech
  30. fmt.Println("Final output", squares + cubes)
  31. }

在线运行程序

在第 7 行,函数 calcSquares 计算一个数每位的平方和,并把结果发送给信道 squareop。与此类似,在第 17 行函数 calcCubes 计算一个数每位的立方和,并把结果发送给信道 cubop

这两个函数分别在单独的协程里运行(第 31 行和第 32 行),每个函数都有传递信道的参数,以便写入数据。Go 主协程会在第 33 行等待两个信道传来的数据。一旦从两个信道接收完数据,数据就会存储在变量 squarescubes 里,然后计算并打印出最后结果。该程序会输出:

  1. Final output 1536

死锁

使用信道需要考虑的一个重点是死锁。当 Go 协程给一个信道发送数据时,照理说会有其他 Go 协程来接收数据。如果没有的话,程序就会在运行时触发 panic,形成死锁。

同理,当有 Go 协程等着从一个信道接收数据时,我们期望其他的 Go 协程会向该信道写入数据,要不然程序就会触发 panic。

  1. package main
  2. func main() {
  3. ch := make(chan int)
  4. ch <- 5
  5. }

在线运行程序

在上述程序中,我们创建了一个信道 ch,接着在下一行 ch <- 5,我们把 5 发送到这个信道。对于本程序,没有其他的协程从 ch 接收数据。于是程序触发 panic,出现如下运行时错误。

  1. fatal error: all goroutines are asleep - deadlock!
  2. goroutine 1 [chan send]:
  3. main.main()
  4. /tmp/sandbox249677995/main.go:6 +0x80

单向信道

我们目前讨论的信道都是双向信道,即通过信道既能发送数据,又能接收数据。其实也可以创建单向信道,这种信道只能发送或者接收数据。

  1. package main
  2. import "fmt"
  3. func sendData(sendch chan<- int) {
  4. sendch <- 10
  5. }
  6. func main() {
  7. sendch := make(chan<- int)
  8. go sendData(sendch)
  9. fmt.Println(<-sendch)
  10. }

在线运行程序

上面程序的第 10 行,我们创建了唯送(Send Only)信道 sendchchan<- int 定义了唯送信道,因为箭头指向了 chan。在第 12 行,我们试图通过唯送信道接收数据,于是编译器报错:

  1. main.go:11: invalid operation: <-sendch (receive from send-only type chan<- int)

一切都很顺利,只不过一个不能读取数据的唯送信道究竟有什么意义呢?

这就需要用到信道转换(Channel Conversion)了。把一个双向信道转换成唯送信道或者唯收(Receive Only)信道都是行得通的,但是反过来就不行。

  1. package main
  2. import "fmt"
  3. func sendData(sendch chan<- int) {
  4. sendch <- 10
  5. }
  6. func main() {
  7. chnl := make(chan int)
  8. go sendData(chnl)
  9. fmt.Println(<-chnl)
  10. }

在线运行程序

在上述程序的第 10 行,我们创建了一个双向信道 cha1。在第 11 行 cha1 作为参数传递给了 sendData 协程。在第 5 行,函数 sendData 里的参数 sendch chan<- intcha1 转换为一个唯送信道。于是该信道在 sendData 协程里是一个唯送信道,而在 Go 主协程里是一个双向信道。该程序最终打印输出 10

关闭信道和使用 for range 遍历信道

数据发送方可以关闭信道,通知接收方这个信道不再有数据发送过来。

当从信道接收数据时,接收方可以多用一个变量来检查信道是否已经关闭。

  1. v, ok := <- ch

上面的语句里,如果成功接收信道所发送的数据,那么 ok 等于 true。而如果 ok 等于 false,说明我们试图读取一个关闭的通道。从关闭的信道读取到的值会是该信道类型的零值。例如,当信道是一个 int 类型的信道时,那么从关闭的信道读取的值将会是 0

  1. package main
  2. import (
  3. "fmt"
  4. )
  5. func producer(chnl chan int) {
  6. for i := 0; i < 10; i++ {
  7. chnl <- i
  8. }
  9. close(chnl)
  10. }
  11. func main() {
  12. ch := make(chan int)
  13. go producer(ch)
  14. for {
  15. v, ok := <-ch
  16. if ok == false {
  17. break
  18. }
  19. fmt.Println("Received ", v, ok)
  20. }
  21. }

在线运行程序

在上述的程序中,producer 协程会从 0 到 9 写入信道 chn1,然后关闭该信道。主函数有一个无限的 for 循环(第 16 行),使用变量 ok(第 18 行)检查信道是否已经关闭。如果 ok 等于 false,说明信道已经关闭,于是退出 for 循环。如果 ok 等于 true,会打印出接收到的值和 ok 的值。

  1. Received 0 true
  2. Received 1 true
  3. Received 2 true
  4. Received 3 true
  5. Received 4 true
  6. Received 5 true
  7. Received 6 true
  8. Received 7 true
  9. Received 8 true
  10. Received 9 true

for range 循环用于在一个信道关闭之前,从信道接收数据。

接下来我们使用 for range 循环重写上面的代码。

  1. package main
  2. import (
  3. "fmt"
  4. )
  5. func producer(chnl chan int) {
  6. for i := 0; i < 10; i++ {
  7. chnl <- i
  8. }
  9. close(chnl)
  10. }
  11. func main() {
  12. ch := make(chan int)
  13. go producer(ch)
  14. for v := range ch {
  15. fmt.Println("Received ",v)
  16. }
  17. }

在线运行程序

在第 16 行,for range 循环从信道 ch 接收数据,直到该信道关闭。一旦关闭了 ch,循环会自动结束。该程序会输出:

  1. Received 0
  2. Received 1
  3. Received 2
  4. Received 3
  5. Received 4
  6. Received 5
  7. Received 6
  8. Received 7
  9. Received 8
  10. Received 9

我们可以使用 for range 循环,重写信道的另一个示例这一节里面的代码,提高代码的可重用性。

如果你仔细观察这段代码,会发现获得一个数里的每位数的代码在 calcSquarescalcCubes 两个函数内重复了。我们将把这段代码抽离出来,放在一个单独的函数里,然后并发地调用它。

  1. package main
  2. import (
  3. "fmt"
  4. )
  5. func digits(number int, dchnl chan int) {
  6. for number != 0 {
  7. digit := number % 10
  8. dchnl <- digit
  9. number /= 10
  10. }
  11. close(dchnl)
  12. }
  13. func calcSquares(number int, squareop chan int) {
  14. sum := 0
  15. dch := make(chan int)
  16. go digits(number, dch)
  17. for digit := range dch {
  18. sum += digit * digit
  19. }
  20. squareop <- sum
  21. }
  22. func calcCubes(number int, cubeop chan int) {
  23. sum := 0
  24. dch := make(chan int)
  25. go digits(number, dch)
  26. for digit := range dch {
  27. sum += digit * digit * digit
  28. }
  29. cubeop <- sum
  30. }
  31. func main() {
  32. number := 589
  33. sqrch := make(chan int)
  34. cubech := make(chan int)
  35. go calcSquares(number, sqrch)
  36. go calcCubes(number, cubech)
  37. squares, cubes := <-sqrch, <-cubech
  38. fmt.Println("Final output", squares+cubes)
  39. }

在线运行程序

上述程序里的 digits 函数,包含了获取一个数的每位数的逻辑,并且 calcSquarescalcCubes 两个函数并发地调用了 digits。当计算完数字里面的每一位数时,第 13 行就会关闭信道。calcSquarescalcCubes 两个协程使用 for range 循环分别监听了它们的信道,直到该信道关闭。程序的其他地方不变,该程序同样会输出:

  1. Final output 1536

本教程的内容到此结束。关于信道还有一些其他的概念,比如缓冲信道(Buffered Channel)、工作池(Worker Pool)和 select。我们会在接下来的教程里专门介绍它们。感谢阅读。祝你愉快。

添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注