Go通道与死锁概述
go语言的并发模型基于goroutine和通道（channel），通道是goroutine之间通信的强大工具。然而，如果不正确地使用通道，尤其是未关闭的通道与for-range循环结合时，很容易导致程序陷入死锁。死锁在go中通常表现为“all goroutines are asleep - deadlock!”错误，这意味着所有goroutine都在等待某个事件（通常是通道操作），但这个事件永远不会发生。这种错误通常是程序逻辑上的缺陷，而非可捕获的运行时异常。
考虑以下一个简单的树遍历示例，其中Walk函数将树中的值发送到一个通道，而main函数尝试使用for-range从通道接收并打印这些值：
package main
import (
"fmt"
"code.google.com/p/go-tour/tree" // 假设这是一个提供树结构的库
)
// Walk 遍历树t，将所有值发送到通道ch
func Walk(t *tree.Tree, ch chan int) {
if t != nil {
Walk(t.Left, ch)
ch <- t.Value
Walk(t.Right, ch)
}
}
func main() {
var ch chan int = make(chan int)
go Walk(tree.New(1), ch)
for c := range ch { // 此处会发生死锁
fmt.Printf("%d ", c)
}
}运行上述代码会遇到死锁错误。原因在于main函数中的for-range ch循环会持续等待通道ch中是否有新的值。当Walk goroutine完成树的遍历并将所有值发送完毕后，它会退出。此时，通道ch中不再有新的值会被发送，但它也没有被关闭。for-range循环会认为通道可能还会接收到数据，因此会无限期地等待下去，导致所有goroutine（包括main goroutine）都进入休眠状态，最终Go运行时检测到死锁并报错。
解决方案一：显式关闭通道
解决上述死锁问题的核心在于，当所有数据都已发送到通道后，需要显式地关闭通道。close(ch)操作会通知所有接收者，通道不会再有新的值发送。当for-range循环在一个已关闭的通道上尝试接收时，它会依次接收完所有剩余的值，然后优雅地退出循环。
正确的做法是将close(ch)放在发送goroutine中，确保在所有值发送完毕后执行：
立即学习“go语言免费学习笔记（深入）”；
package main
import (
"fmt"
"code.google.com/p/go-tour/tree"
)
// Walk 遍历树t，将所有值发送到通道ch
func Walk(t *tree.Tree, ch chan int) {
if t != nil {
Walk(t.Left, ch)
ch <- t.Value
Walk(t.Right, ch)
}
}
func main() {
var ch chan int = make(chan int)
go func() { // 使用匿名goroutine封装Walk和close操作
Walk(tree.New(1), ch)
close(ch) // 在所有值发送完毕后关闭通道
}()
for c := range ch {
fmt.Printf("%d ", c)
}
fmt.Println("\nTraversal complete.") // 循环结束后打印
}通过将Walk函数调用和close(ch)操作封装在一个新的goroutine中，我们确保了通道在所有数据发送完毕后被关闭。这样，main函数中的for-range循环就能正常接收所有值并最终退出，避免了死锁。
注意事项：
- 发送端关闭，接收端不关闭： 只有发送方才能关闭通道。尝试关闭一个已关闭的通道或由接收方关闭通道会导致运行时错误（panic）。
- 关闭通道的意义： 关闭通道是向接收者发出信号，表示不会再有数据发送过来。接收者仍然可以从已关闭的通道中读取所有已发送但未接收的数据，直到通道为空。之后，从已关闭的通道接收将立即返回零值，并且第二个返回值（ok）为false。
解决方案二：并发遍历与sync.WaitGroup协调
在更复杂的并发场景中，例如树的并行遍历，可能有多个goroutine向同一个通道发送数据。在这种情况下，简单地在单个发送goroutine结束后关闭通道是不够的，因为其他并行发送的goroutine可能尚未完成。为了确保在所有发送goroutine都完成后再关闭通道，我们需要使用sync.WaitGroup来协调。
sync.WaitGroup用于等待一组goroutine完成。它有三个主要方法：
- Add(delta int)：增加等待的goroutine计数。
- Done()：递减等待的goroutine计数，通常在goroutine完成工作后调用。
- Wait()：阻塞直到等待的goroutine计数为零。
以下是如何使用sync.WaitGroup实现并行树遍历并安全关闭通道的示例：
package main
import (
"fmt"
"sync"
"code.google.com/p/go-tour/tree"
)
// Walk 并行遍历树t，将所有值发送到通道ch。
// done用于协调goroutine的完成。
func Walk(t *tree.Tree, ch chan int, done *sync.WaitGroup) {
defer done.Done() // 确保无论如何，当前goroutine完成时都会调用Done()
if t != nil {
done.Add(2) // 为左右子树的并行遍历各增加一个计数
go Walk(t.Left, ch, done)
go Walk(t.Right, ch, done)
ch <- t.Value // 将当前节点值发送到通道
}
}
func main() {
// 使用带缓冲的通道，以减少并行发送时的阻塞，提高效率
// 缓冲大小可以根据树的深度或预期并发量来调整
var ch chan int = make(chan int, 64)
go func() {
done := new(sync.WaitGroup)
done.Add(1) // 为初始的Walk调用增加一个计数
// 启动并行遍历
Walk(tree.New(1), ch, done)
done.Wait() // 等待所有Walk goroutine完成
close(ch) // 所有发送者都完成后，安全关闭通道
}()
// 主goroutine从通道接收并打印值
for c := range ch {
fmt.Printf("%d ", c)
}
fmt.Println("\nParallel traversal complete.")
}在这个并行遍历的例子中：
- main函数启动一个匿名goroutine来管理整个遍历过程和通道的关闭。
- done := new(sync.WaitGroup) 创建了一个WaitGroup实例。
- done.Add(1) 为最初调用的Walk(tree.New(1), ch, done) 增加一个计数。
- 在Walk函数内部，defer done.Done() 确保无论函数如何退出，都会递减WaitGroup计数。
- 当Walk函数递归地启动左右子树的遍历时，它会调用done.Add(2)来为这两个新的goroutine增加计数。
- done.Wait() 会阻塞，直到所有由Add增加的计数都被Done递减为零，这意味着所有相关的Walk goroutine都已完成。
- 一旦done.Wait()返回，我们就可以安全地调用close(ch)，因为此时所有数据都已经发送完毕。
- main goroutine的for-range循环会接收所有数据，并在通道关闭后正常退出。
关键点：
- defer done.Done()： 这是确保Done()被调用的惯用模式，即使在函数内部发生错误或提前返回，也能保证计数正确递减。
- 缓冲通道： 在并行发送数据的场景中，使用带缓冲的通道（make(chan int, 64)）可以避免发送goroutine因接收方来不及处理而阻塞，从而提高并发效率。
- 计数管理： Add和Done的调用必须配对。Add通常在启动新的goroutine之前调用，Done则在goroutine完成其工作后调用。
总结
Go语言中的通道死锁通常是由于for-range循环等待一个永远不会关闭的通道而导致的程序逻辑错误。解决死锁的关键在于确保通道在所有数据发送完毕后被显式关闭。对于简单的单发送者场景，直接在发送goroutine中调用close()即可。对于复杂的并发场景，如多个goroutine并行向同一通道发送数据，应使用sync.WaitGroup来协调所有发送goroutine的完成，确保在所有发送者都完成后再关闭通道，以避免死锁并保证程序的正确性。理解并正确管理通道的生命周期是编写健壮Go并发程序的基石。
