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

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

解释

1. 创建通道：

   ch := make(chan int, 5)
   

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

2. 启动 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) 关闭通道，表示不再发送新的值。

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

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

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

注意事项

1. 关闭通道：

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

2. 多接收者：

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

3. 空通道：

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

示例：多接收者

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

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

解释

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