Go语言中的通道

通道(channel)是Go 语言中一种特殊的数据类型,通道本身就是并发安全的,可以通过它在多个 goroutine 之间传递数据。通道是Go 语言编程理念:“Do not communicate by sharing memory; instead, share memory by communicating”(不要通过共享数据来通信,而应该通过通信来共享数据。)的完美实现,在并发编程中经常会遇到它。下面来介绍一下通道的使用方法。

通道的发送和接收

通道包括双向通道和单向通道,这里双向通道只的是支持发送和接收的通道,而单向通道是只能发送或者只能接收的通道。

双向通道

使用make函数声明并初始化一个通道:

1
ch1 := make(chan string, 3)
  • chan 是表示通道类型的关键字
  • string 表示该通道类型的元素类型
  • 3 表示该通道的容量为3,最多可以缓存3个元素值。

一个通道相当于一个先进先出(FIFO)的队列,使用操作符 <- 进行元素值的发送和接收:

1
ch1 <- "1"  //向通道ch1发送数据 "1"

接收元素值:

1
elem1 := <- ch1 // 接收通道中的元素值

首先接收到的元素为先存入通道中的元素值,也就是先进先出:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
package main

import "fmt"

func main() {
str1 := []string{"hello","world", "!"}
ch1 := make(chan string, len(str1))

for _, str := range str1 {
ch1 <- str
}

for i := 0; i < len(str1); i++ {
elem := <- ch1
fmt.Println(elem)
}
}

执行结果:

1
2
3
hello
world
!

单向通道

单向通道包括只能发送的通道和只能接收的通道:

1
2
var WriteChan = make(chan<- interface{}, 1) // 只能发送不能接收的通道
var ReadChan = make(<-chan interface{}, 1) // 只能接收不能发送的通道

单向通道的这种特性可以用来约束函数的输入类型或者输出类型,比如下面的例子约束了只能从通道中接收元素值:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
package main

import (
"fmt"
)

func OnlyReadChan(num int) <-chan int {
ch := make(chan int, 1)
ch <- num
close(ch)
return ch
}

func main() {

Chan1 := OnlyReadChan(6)
num := <- Chan1
fmt.Println(num)
}

执行结果:

1
6

通道阻塞

通道操作是并发安全的,在同一时刻,只会执行对同一个通道的任意个发送操作中的某一个,直到这个元素值被完全复制进该通道之后,其他针对该通道的发送操作才可能被执行。接收操作也一样。另外,对于通道中的同一个元素值来说,发送操作和接收操作之间也是互斥的。

发送操作和接收操作是原子操作,也就是说,发送操作绝不会出现只复制了一部分的情况,要么还没有复制,要么已经复制完毕。接收操作在准备好元素值的副本之后,一定会删除掉通道中的原值,绝不会出现通道中仍有残留的情况。在进行发送操作和接收操作时,代码会一直阻塞在那里,完成操作后才会继续执行后面的代码。通道的发送操作和接收操作是很快的,那么什么情况下会出现长时间的阻塞呢?下面介绍几种情况。

缓冲通道的阻塞

缓冲通道是容量大于0的通道,也就是可以缓存数据的通道。

1、发送阻塞

如果缓冲通道已经填满,如果有goroutine继续向该通道发送数据就会阻塞。请看下面的例子:

1
2
3
4
5
6
7
package main

func main() {
ch1 := make(chan int, 1)
ch1 <- 1
ch1 <- 2
}

执行结果:

1
2
3
4
fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan send]:
...........

如果通道可以接收数据(有元素被接收),通道会通知最先等待发送操作的 goroutine再次执行发送操作。

2、接收阻塞

类似的,如果通道已空,如果继续进行接收操作就会被阻塞。

1
2
3
4
5
6
package main

func main() {
ch1 := make(chan int, 1)
<- ch1
}

执行结果:

1
2
3
4
fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan receive]:
...........

非缓冲通道

非缓冲通道是容量为0的通道,不能缓存数据。

非缓冲通道的数据传递是同步的,发送操作或者接收操作在执行后就会阻塞,需要对应的接收操作或者发送操作执行才会继续传递。由此可以看出缓冲通道使用的是异步方式进行数据传递。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
package main

import (
"fmt"
)

func main() {
str1 := []string{"hello","world", "!"}
ch1 := make(chan string, 0)

go func() {
for _, str := range str1 {
ch1 <- str
}
}()

for i := 0; i < len(str1); i++ {
elem := <- ch1
fmt.Println(elem)
}

}

执行结果:

1
2
3
hello
world
!

上面的代码中3个goroutine向通道写了三次数据,必须有三次接收,不然会阻塞。

对值为nil的通道进行发送操作和接收操作也会发生阻塞:

1
2
3
var ch1 chan int
ch1 <- 1 // 阻塞
<-ch1 // 阻塞

通道关闭

可以使用close()方法来关闭通道,通道关闭后,不能再对通道进行发送操作,可以进行接收操作。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
package main

import "fmt"

func main() {
ch1 := make(chan int, 1)
ch1 <- 1
close(ch1)

ele := <-ch1
fmt.Println(ele)

ch1 <- 2
}

执行结果:

1
2
3
4
5
1
panic: send on closed channel

goroutine 1 [running]:
.....

如果通道关闭时,里面还有元素,进行接收操作时,返回的通道关闭标志仍然为true:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
package main

import "fmt"

func main() {
ch1 := make(chan int, 1)
ch1 <- 1
close(ch1)

ele1, statu1 := <-ch1
fmt.Println(ele1, statu1)
ele2, statu2 := <-ch1
fmt.Println(ele2, statu2)
}

执行结果:

1
2
1 true
0 false

由于通道的这种特性,可以让发送方来关闭通道。前面的例子可以这样写:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package main

import (
"fmt"
)

func main() {
str1 := []string{"hello","world", "!"}
ch1 := make(chan string, 0)

go func() {
for _, str := range str1 {
ch1 <- str
}
close(ch1)
}()

for i := 0; i < len(str1); i++ {
elem := <- ch1
fmt.Println(elem)
}

}

另外,不能对关闭的通道再次关闭:

1
2
3
4
5
6
7
8
9
10
11
package main

// import "fmt"

func main() {
ch1 := make(chan int, 1)
ch1 <- 1
close(ch1)
close(ch1)

}

执行结果:

1
panic: close of closed channel

select语句与通道

select语句通常与通道联用,它是专为通道而设计的。select语句执行时,一般只有一个case表达式或者default语句会被运行。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
package main

import "fmt"

func main() {
ch1 := make(chan int, 1)
num := 2

select {
case data := <-ch1:
fmt.Println("Read data: ", data)
case ch1 <- num:
fmt.Println("Write data: ", num)
default:
fmt.Println("No candidate case is selected!")
}
}

执行结果:

1
Write data:  2

需要注意的是,如果没有default默认分支,case表达式都没有满足条件,那么select语句就会被阻塞,直到至少有一个case表达式满足条件为止。

如果同时有多个分支满足条件,会随机选择一个分支执行

for语句与select语句联用时,分支中的break语句只能结束当前select语句的执行,而不会退出for循环。下面的代码永远不会退出循环:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
package main

import "fmt"

func main() {
ch1 := make(chan int, 1)
for {
select {
case ch1 <- 6:
fmt.Println("Write data: 6")
case data := <-ch1:
fmt.Println(data)
break
}
}
}

解决方案是使用goto语句和标签。

方法1:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
package main

import "fmt"

func main() {
ch1 := make(chan int, 1)
num := 6
for {
select {
case ch1 <- num:
fmt.Println("Write data: ", num)
case data := <-ch1:
fmt.Println("Read data: ", data)
goto loop
}
}
loop:
fmt.Println(ch1)
}

执行结果:

1
2
3
Write data:  6
Read data: 6
0xc00000e0e0

方法2:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
package main

import "fmt"

func main() {
ch1 := make(chan int, 1)
num := 6
loop:
for {
select {
case ch1 <- num:
fmt.Println("Write data: ", num)
case data := <-ch1:
fmt.Println("Read data: ", data)
break loop
}
}
fmt.Println(ch1)
}

执行结果:

1
2
3
Write data:  6
Read data: 6
0xc0000e4000

总结

本文主要介绍了通道的基本操作:初始化、发送、接收和关闭,要注意在什么情况下会引起通道阻塞。select语句通常与通道联用,介绍了分支的选择规则以及for语句与select语句联用时如何退出循环。

通道是 Go 语言并发编程的重要实现基础,还是有必要掌握的。

--THE END--

本文标题:Go语言中的通道

文章作者:hiyo

文章链接:https://hiyongz.github.io/posts/golang-notes-for-channel-sync/

许可协议:本博客文章除特别声明外,均采用CC BY-NC-ND 4.0 许可协议。转载请保留原文链接及作者。

关注微信公众号,及时接收最新技术文章!