当前位置：首页 > article >正文

go语言range的高级用法-使用range来接收通道里面的数据

article 2024/11/26 21:21:36

在 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)
}