Go语言并发秘籍:精通Channel发送与接收技巧
// 并发地使用 channel 发送和接收数据
package main
import (
"fmt"
"sync"
"sync/atomic"
"time"
)
func main() {
// 创建一个 channel,容量为 10
channel := make(chan int, 10)
var wg sync.WaitGroup
var sent, received int32
// 添加等待组的计数
wg.Add(1)
// 启动一个 goroutine 来接收数据
go func() {
defer wg.Done()
for num := range channel {
// 接收到数据后,原子性地增加接收计数
atomic.AddInt32(&received, 1)
fmt.Printf("接收到数据: %d\n", num)
}
}()
// 模拟发送数据
start := time.Now()
for i := 0; i < 10; i++ {
// 将数据发送到 channel
channel <- i
// 原子性地增加发送计数
atomic.AddInt32(&sent, 1)
}
// 关闭 channel,通知接收方结束循环
close(channel)
// 等待 goroutine 完成
wg.Wait()
// 输出发送和接收的数量
fmt.Printf("发送的数量: %d\n", sent)
fmt.Printf("接收的数量: %d\n", received)
fmt.Printf("程序运行时间: %v\n", time.Since(start))
}
这段代码创建了一个有缓冲的 channel,并启动了一个 goroutine 来接收数据。主线程模拟发送数据,并在发送完所有数据后关闭 channel,通知接收方结束接收过程。代码使用了 sync.WaitGroup
来等待 goroutine 完成,使用了 atomic
包来处理计数器,以确保线程安全。最后,代码输出了发送和接收的数据量以及程序的运行时间。
评论已关闭