在 Go 语言中，可以使用 for ... range 循环来遍历通道（channel）。for ... range 循环会一直从通道中接收值，直到通道关闭并且所有值都被接收完毕。
使用 for ... range 遍历通道
示例代码
下面是一个使用 for ... range 遍历通道的示例：
package main
import (
"fmt"
"time"
)
func main() {
// 创建一个有缓冲的通道
ch := make(chan int, 5)
// 启动一个 goroutine 向通道发送值
go func() {
for i := 0; i < 5; i++ {
ch <- i
time.Sleep(100 * time.Millisecond) // 模拟耗时操作
}
close(ch) // 关闭通道
}()
// 使用 for ... range 遍历通道
for value := range ch {
fmt.Println("Received:", value)
}
}
解释
-
创建通道：
ch := make(chan int, 5)make(chan int, 5)创建一个容量为 5 的有缓冲通道。
-
启动 goroutine 发送值：
go func() { for i := 0; i < 5; i++ { ch <- i time.Sleep(100 * time.Millisecond) // 模拟耗时操作 } close(ch) // 关闭通道 }()- 启动一个 goroutine，向通道发送 0 到 4 的整数。
time.Sleep(100 * time.Millisecond)模拟耗时操作，使发送操作之间有时间间隔。close(ch)关闭通道，表示不再发送新的值。
-
使用
for ... range遍历通道：for value := range ch { fmt.Println("Received:", value) }for value := range ch循环会一直从通道中接收值，直到通道关闭并且所有值都被接收完毕。- 每次接收到一个值，都会打印出来。
注意事项
-
关闭通道：
- 必须在发送完所有值后关闭通道，否则
for ... range循环会一直阻塞，等待新的值。 - 关闭通道后，不能再向通道发送值，否则会引发 panic。
- 必须在发送完所有值后关闭通道，否则
-
多接收者：
- 如果有多个 goroutine 同时从同一个通道接收值，可能会出现竞争条件。建议使用互斥锁或其他同步机制来确保线程安全。
-
空通道：
- 对于无缓冲通道，如果没有接收者，发送操作会阻塞，直到有接收者准备接收值。
示例：多接收者
下面是一个多接收者从同一个通道接收值的示例：
package main
import (
"fmt"
"sync"
"time"
)
func main() {
// 创建一个有缓冲的通道
ch := make(chan int, 5)
// 启动一个 goroutine 向通道发送值
go func() {
for i := 0; i < 5; i++ {
ch <- i
time.Sleep(100 * time.Millisecond) // 模拟耗时操作
}
close(ch) // 关闭通道
}()
// 使用 sync.WaitGroup 等待所有接收者完成
var wg sync.WaitGroup
// 启动多个接收者 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)
}
// 等待所有接收者完成
wg.Wait()
}
解释
-
启动多个接收者 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 的计数。
- 启动三个接收者 goroutine，每个 goroutine 都使用
-
等待所有接收者完成：
wg.Wait()- 使用
wg.Wait()等待所有接收者 goroutine 完成。
- 使用
通过使用 for ... range 循环，可以方便地遍历通道中的值，直到通道关闭并且所有值都被接收完毕。这对于处理并发任务和数据流非常有用。
