channel 是怎么走上死锁这条路的

---

本篇文章接着 hello world 的并发实现一文介绍 Go 的 channel 类型，同时进一步介绍 channel 的几种死锁情况，这些都是代码中很容易遇到的，要重点摘出来讲，防止一不留神程序就“死”了。

1. 为什么需要 channel?

channel 是一种通道类型，它通过发送和接收需要共享的资源，实现共享资源在 goroutine 之间的同步。说起来有点枯燥，来看一段代码:

func speak(words *string) {
	*words = "hello world"
}

func listen(words string) {
	fmt.Printf("listen %s\n", words)
}

func main() {
	runtime.GOMAXPROCS(2)

	var words string
	go speak(&words)
	go listen(words)
}

这段代码的初衷是开启两个 goroutine，其中 speak goroutine 写字符 "hello world" 给 words，而 listen goroutine 打印字符。执行代码发现什么都没打印，这是因为主main goroutine 还没等两个 goroutine 运行就提前结束了。那么加上 sync 等待每个 goroutine 运行，继续改写代码:

func speak(words *string) {
	defer wg.Done()
	*words = "hello world"
}

func listen(words string) {
	defer wg.Done()
	fmt.Printf("listen %s\n", words)
}

func main() {
	runtime.GOMAXPROCS(2)

	wg.Add(2)
	var words string
	go speak(&words)
	go listen(words)
	wg.Wait()
}

执行结果:

listen

可以看到两个 goroutine 都执行了，不过由于是并发执行，并没有实现我们想要的打印 hello world 效果。

因此一个问题就冒出来了，各个 goroutine 间怎么知道彼此是否执行完毕了呢？或者换个问法，各个 goroutine 间怎么实现相互同步呢？
这时候令人欢呼雀跃，让人疯狂的 channel 迈着矫健的步伐闪亮登场了，它彷佛在说快用我，快用我。没错，使用 channel 可以很好的解决这个问题。改下代码如下:

func speak(words chan string) {
	defer wg.Done()
	words <- "hello world"
}

func listen(words chan string) {
	defer wg.Done()
	value := <-words
	fmt.Printf("listen %s", value)
}

func main() {
	runtime.GOMAXPROCS(2)

	wg.Add(2)
	words := make(chan string)
	go speak(words)
	go listen(words)
	wg.Wait()
}

执行结果:

listen hello world

可以看到，引入 channel 很容易的打印出想要的效果。
值得注意的是: 引入 channel，这里的 sync 还是必须的，如果不要的话 main goroutine 还是会不等两个 goroutine 运行完而提前走人。

关于 channel 的使用，分类，关闭和遍历等内容不在本篇文章的讨论范围之内。接下来直接看使用 channel 会遇到的几种死锁情况。

2. channel 死锁

2.1 死锁名场面1

代码示例:

func speak(words chan string) {jing
	defer wg.Done()
	words <- "hello world"
}

func main() {
	runtime.GOMAXPROCS(2)

	wg.Add(2)
	words := make(chan string)
	go speak(words)
	wg.Wait()
}

上述代码往 channel 中写入字符串，但是没有 goroutine 接收 channel 中的字符串导致程序阻塞在传值，引发死锁。

func listen(words chan string) {
	defer wg.Done()
	value := <-words
	fmt.Printf("listen %s", value)
}

func main() {
	runtime.GOMAXPROCS(2)

	wg.Add(2)
	words := make(chan string)
	go listen(words)
	wg.Wait()
}

与前面死锁情况类似，这里没有 goroutine 往 channel 内写字符，导致程序阻塞在取值，引发死锁。

2.2 死锁名场面2

func speak(words chan string) {
	defer wg.Done()
	words <- "hello world"
}

func listen(words chan string) {
	defer wg.Done()

	disturber := make(chan string)
	disturber <- "hi"

	value := <-words
	fmt.Printf("listen %s", value)
}

func main() {
	runtime.GOMAXPROCS(2)

	wg.Add(2)
	words := make(chan string)
	go speak(words)
	go listen(words)
	wg.Wait()
}

进一步的，这里添加了另一个 channel disturber，它将接收字符串。
程序依然会出现死锁。因为两个 goroutine 都在等着通道 words 和 disturber 的值被接收，即便 value 会取 words 的值，但由于 disturber 的值没有接收，程序会一直阻塞在 disturber 传值这里，导致死锁。

2.3 死锁名场面3

改写代码如下:

func cal_hello_num_(word string, ch chan int) {
	defer wg.Done()

	/* 通道是收方双方的，如果收和方是一对多的关系则需对方发进行限制，防止竞争抢占 */
	mutex.Lock()
	number++
	ch <- number
	fmt.Printf("say: %s, %d\n", word, number)
	mutex.Unlock()
}

func show_hello_num_(ch chan int) {
	defer wg.Done()
	for {
		num, ok := <-ch
		if !ok {
			fmt.Printf("\nnum: %d", num)
			break
		}
	}
}

func num_say_hello_channel(ch chan int) {
	go cal_hello_num_("w", ch)
	go cal_hello_num_("o", ch)
	go cal_hello_num_("r", ch)
	go cal_hello_num_("l", ch)
	go cal_hello_num_("d", ch)
	go show_hello_num_(ch)
}

func main() {
	runtime.GOMAXPROCS(2)
	wg.Add(6)
	ch := make(chan int)
	num_say_hello_channel(ch)
	wg.Wait()
}

代码很简单这里就不介绍了，查看执行结果:

say: w, 1
say: r, 2
say: o, 3
say: l, 4
say: d, 5
fatal error: all goroutines are asleep - deadlock!

报死锁了，为什么呢？
细看之下发现问题出在 for 循环这里，for 循环持续从通道 ch 读数据，当通道中无数据可读的时候相当于阻塞在通道取值，从而引发死锁。这是对于无缓冲通道的取值而言，对于有缓冲的通道也是适用的，有缓冲的通道数据读完了也相当于无缓冲的通道。

知道了哪里错了在解决起来就不难了，在最后一个 goroutine 传通道值后将通道 close 掉，后面使用 for 读取已经关闭的通道将输出 ok 为 false:

func cal_hello_num_(word string, ch chan int) {
	defer wg.Done()

	/* 经典案例: 通道是收方双方的，如果收和方是一对多的关系则需对方发进行限制，防止竞争抢占 */
	mutex.Lock()
	number++
	ch <- number
	fmt.Printf("say: %s, %d\n", word, number)

	if number == 5 {
		fmt.Printf("close channel")
		close(ch)
	}

	mutex.Unlock()
}

执行结果:

say: w, 1
say: o, 2
say: r, 3
say: l, 4
say: d, 5
close channel
num: 0