在Go语言的并发编程中，通道是goroutine之间通信的重要载体，而for range是遍历通道的常用方式，这种方式可以自动从通道中接收值直到通道关闭。如果通道在被for range遍历时没有被正确关闭，程序就可能出现死锁，导致整个程序无法继续运行。
死锁产生的原理
for range遍历通道的逻辑是：不断从通道中接收值，每接收到一个值就执行一次循环体，直到通道被关闭才会退出循环。如果通道一直没有被关闭，for range就会一直阻塞在接收操作上，等待新的数据发送到通道中。
当没有其他goroutine往通道发送数据，并且通道也没有被关闭时，for range的接收操作会永久阻塞，而主goroutine或者其他相关goroutine如果也在等待这个循环结束，就会形成死锁。Go的运行时检测到所有goroutine都处于阻塞状态且没有任何可执行的任务时，就会抛出死锁错误。
下面是一段会触发死锁的示例代码：
package main
func main() {
ch := make(chan int)
// 启动一个goroutine发送数据，但发送完没有关闭通道
go func() {
ch <- 1
ch <- 2
// 这里没有调用close(ch)
}()
// for range遍历通道，会一直等待新数据
for num := range ch {
println(num)
}
}
上述代码中，子goroutine发送了两个数据后结束，没有关闭通道，主goroutine的for range会一直等待通道的新数据，而此时已经没有goroutine会往通道发送数据，也没有goroutine关闭通道，程序就会抛出死锁错误。
解决方案
1. 发送方主动关闭通道
这是最常见的解决方式，在发送数据的goroutine完成所有数据发送后，主动调用close函数关闭通道，for range遍历到通道关闭就会自动退出循环。
package main
func main() {
ch := make(chan int)
go func() {
ch <- 1
ch <- 2
// 发送完所有数据后关闭通道
close(ch)
}()
for num := range ch {
println(num)
}
println("遍历结束")
}
2. 使用带缓冲的通道并配合等待机制
如果可以提前确定发送的数据量，可以使用带缓冲的通道，发送完数据后关闭通道，或者使用sync.WaitGroup等待发送完成后再关闭通道。
package main
import "sync"
func main() {
ch := make(chan int, 2)
var wg sync.WaitGroup
wg.Add(1)
go func() {
defer wg.Done()
ch <- 1
ch <- 2
}()
// 等待发送goroutine完成后再关闭通道
go func() {
wg.Wait()
close(ch)
}()
for num := range ch {
println(num)
}
}
3. 使用context控制遍历退出
如果无法通过关闭通道的方式退出，可以使用context来传递退出信号，在循环中判断context的状态，主动退出循环。
package main
import (
"context"
"time"
)
func main() {
ch := make(chan int)
ctx, cancel := context.WithCancel(context.Background())
go func() {
for i := 0; i < 5; i++ {
select {
case <-ctx.Done():
return
case ch <- i:
time.Sleep(time.Millisecond * 100)
}
}
// 发送完数据关闭通道
close(ch)
}()
// 模拟3秒后取消遍历
go func() {
time.Sleep(time.Second * 3)
cancel()
}()
// 循环接收数据，同时判断context状态
for {
select {
case <-ctx.Done():
println("收到退出信号，结束遍历")
return
case num, ok := <-ch:
if !ok {
// 通道关闭，退出循环
println("通道已关闭，结束遍历")
return
}
println(num)
}
}
}
注意事项
- 只有发送数据的goroutine才应该关闭通道，接收方不应该关闭通道，否则可能会触发向已关闭通道发送数据的panic。
- 不要重复关闭同一个通道，重复关闭通道会直接触发panic。
- 如果通道被关闭，接收方从通道接收数据时可以通过第二个返回值判断通道是否已经关闭，例如
val, ok := <-ch，ok为false时表示通道已关闭。
理解for range遍历通道的机制，合理处理通道的关闭逻辑，就可以有效避免未关闭通道导致的死锁问题，让Go的并发程序更加稳定可靠。