在 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
循环,可以方便地遍历通道中的值,直到通道关闭并且所有值都被接收完毕。这对于处理并发任务和数据流非常有用。