在 Go 语言中，可以使用 for ... range 循环来遍历通道（channel）。for ... range 循环会一直从通道中接收值，直到通道关闭并且所有值都被接收完毕。

使用 for ... range 遍历通道

示例代码

下面是一个使用 for ... range 遍历通道的示例：

package mainimport ("fmt""time"
)func main() {// 创建一个有缓冲的通道ch := make(chan int, 5)// 启动一个 goroutine 向通道发送值go func() {for i := 0; i < 5; i++ {ch <- itime.Sleep(100 * time.Millisecond) // 模拟耗时操作}close(ch) // 关闭通道}()// 使用 for ... range 遍历通道for value := range ch {fmt.Println("Received:", value)}
}

解释

1. 创建通道：

   ch := make(chan int, 5)
   

   • make(chan int, 5) 创建一个容量为 5 的有缓冲通道。

2. 启动 goroutine 发送值：

   go func() {for i := 0; i < 5; i++ {ch <- itime.Sleep(100 * time.Millisecond) // 模拟耗时操作}close(ch) // 关闭通道
   }()
   

   • 启动一个 goroutine，向通道发送 0 到 4 的整数。
   • time.Sleep(100 * time.Millisecond) 模拟耗时操作，使发送操作之间有时间间隔。
   • close(ch) 关闭通道，表示不再发送新的值。

3. 使用 for ... range 遍历通道：

   for value := range ch {fmt.Println("Received:", value)
   }
   

   • for value := range ch 循环会一直从通道中接收值，直到通道关闭并且所有值都被接收完毕。
   • 每次接收到一个值，都会打印出来。

注意事项

1. 关闭通道：

   • 必须在发送完所有值后关闭通道，否则 for ... range 循环会一直阻塞，等待新的值。
   • 关闭通道后，不能再向通道发送值，否则会引发 panic。

2. 多接收者：

   • 如果有多个 goroutine 同时从同一个通道接收值，可能会出现竞争条件。建议使用互斥锁或其他同步机制来确保线程安全。

3. 空通道：

   • 对于无缓冲通道，如果没有接收者，发送操作会阻塞，直到有接收者准备接收值。

示例：多接收者

下面是一个多接收者从同一个通道接收值的示例：

package mainimport ("fmt""sync""time"
)func main() {// 创建一个有缓冲的通道ch := make(chan int, 5)// 启动一个 goroutine 向通道发送值go func() {for i := 0; i < 5; i++ {ch <- itime.Sleep(100 * time.Millisecond) // 模拟耗时操作}close(ch) // 关闭通道}()// 使用 sync.WaitGroup 等待所有接收者完成var wg sync.WaitGroup// 启动多个接收者 goroutinefor i := 0; i < 3; i++ {wg.Add(1)go func(id int) {defer wg.Done()for value := range ch {fmt.Printf("Receiver %d received: %d\n", id, value)}}(i)}// 等待所有接收者完成wg.Wait()
}

解释

1. 启动多个接收者 goroutine：

   for i := 0; i < 3; i++ {wg.Add(1)go func(id int) {defer wg.Done()for value := range ch {fmt.Printf("Receiver %d received: %d\n", id, value)}}(i)
   }
   

   • 启动三个接收者 goroutine，每个 goroutine 都使用 for ... range 循环从通道中接收值。
   • wg.Add(1) 增加 WaitGroup 的计数。
   • defer wg.Done() 在 goroutine 结束时减少 WaitGroup 的计数。

2. 等待所有接收者完成：

   wg.Wait()
   

   • 使用 wg.Wait() 等待所有接收者 goroutine 完成。

通过使用 for ... range 循环，可以方便地遍历通道中的值，直到通道关闭并且所有值都被接收完毕。这对于处理并发任务和数据流非常有用。