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 包来处理计数器,以确保线程安全。最后,代码输出了发送和接收的数据量以及程序的运行时间。

最后修改于:2024年08月15日 14:03

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日