讲师中心 微信公众号
AI工具推荐 视频效率加速

Golang生产者消费者模型实践示例

千辰吖_2008

千辰吖_2008

发布时间:2026-01-10 01:35:20

|

672人浏览过

|

来源于php中文网

原创

最稳妥方式是带缓冲channel+sync.WaitGroup+单独goroutine关channel+for range自动退出;因直接close易致panic、漏数据或阻塞,须确保所有生产者完成后再由专用goroutine关闭。

golang生产者消费者模型实践示例

Go 语言中用 channel 实现生产者消费者模型,最稳妥、最常用的方式是:**带缓冲 channel + sync.WaitGroup + 单独 goroutine 关闭 channel + 消费者用 for range 自动退出**。其他方式(如无缓冲 channel、手动判断 ok)要么阻塞严重,要么容易 panic 或漏数据。

为什么不能只靠 close(ch) 和 range ch 就完事?

看似简单,但实际踩坑点密集:

  • 如果由某个生产者自己 close(ch),而其他生产者还在往里发数据,会 panic:send on closed channel
  • 如果主 goroutine 直接 close(ch),但此时还有生产者没结束,同样 panic
  • 如果没等所有生产者完成就关闭 channel,部分数据可能还没发出去就被截断
  • range ch 确实会在 channel 关闭后自动退出,但前提是它没被提前阻塞在发送端——所以必须确保“关 channel”这个动作发生在所有生产者 defer wg.Done() 之后

正确做法是:用一个额外 goroutine 监听 WaitGroup 完成后再 close(ch),这样既解耦又安全。

如何控制多个生产者、多个消费者并保证不漏不重?

关键不在 channel 类型,而在生命周期管理逻辑。多对多场景下,必须分离「生产者完成」和「消费者完成」两个信号:

立即学习“go语言免费学习笔记(深入)”;

  • 用一个 sync.WaitGroup 跟踪所有生产者(每个 go producer(...) 前 wg.Add(1))
  • 启动一个 goroutine:go func() { wg.Wait(); close(ch) }() —— 这是唯一合法的关 channel 时机
  • 消费者全部使用 for data := range ch,无需手动检查 ok,channel 关闭后自动退出循环
  • 若需等待消费者也全部退出,再起一个 consumerWg 管理它们(注意:不能复用同一个 wg,否则 wg.Wait() 可能提前返回)

缓冲区大小不是越大越好:make(chan int, 1000) 可能导致内存积压;make(chan int, 1) 则退化为同步模式,吞吐量低。一般按典型批次大小 × 并发数预估,比如每秒 20 个任务、处理延迟 100ms,缓冲 5–10 足够。

Golang Uber Dig
Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等

下载

带结构体的任务怎么传?要不要加超时或取消?

传结构体完全没问题,channel 支持任意可序列化类型(包括自定义 struct),但要注意值拷贝开销。如果结构体很大(比如含 []byte 或指针字段),建议传指针:chan *Task,避免复制成本。

真实项目中几乎都需要超时或取消支持:

  • 生产者发数据前加 select + time.After,防止卡死在满 channel 上
  • 消费者处理任务时用 context.WithTimeout 控制单次处理时长
  • 整个流程用 context.WithCancel 统一中断(比如服务 shutdown)

但初学阶段先跑通基础模型更重要——先确保 ch 关得对、range 退得稳、WaitGroup 数得准。

package main
<p>import (
"fmt"
"math/rand"
"sync"
"time"
)</p><p>type Task struct {
ID   int
Data string
}</p><p>func producer(id int, ch chan<- Task, wg <em>sync.WaitGroup) {
defer wg.Done()
for i := 0; i < 3; i++ {
task := Task{
ID:   i,
Data: fmt.Sprintf("p%d-task%d", id, i),
}
time.Sleep(time.Duration(rand.Intn(300)) </em> time.Millisecond)
ch <- task
fmt.Printf("✅ 生产者 %d 发送: %+v\n", id, task)
}
}</p><p>func consumer(id int, ch <-chan Task, wg <em>sync.WaitGroup) {
defer wg.Done()
for task := range ch {
fmt.Printf("? 消费者 %d 处理: %+v\n", id, task)
time.Sleep(time.Duration(rand.Intn(500)) </em> time.Millisecond)
}
fmt.Printf("? 消费者 %d 退出\n", id)
}</p><p>func main() {
const numProducers = 2
const numConsumers = 3
const bufferSize = 5</p><pre class='brush:php;toolbar:false;'>ch := make(chan Task, bufferSize)
var producerWg sync.WaitGroup

// 启动生产者
for i := 0; i < numProducers; i++ {
    producerWg.Add(1)
    go producer(i, ch, &producerWg)
}

// 启动消费者(用独立 WaitGroup)
var consumerWg sync.WaitGroup
for i := 0; i < numConsumers; i++ {
    consumerWg.Add(1)
    go consumer(i, ch, &consumerWg)
}

// 生产者全部结束后关闭 channel
go func() {
    producerWg.Wait()
    close(ch)
}()

// 等待所有消费者完成
consumerWg.Wait()
fmt.Println("? 全部任务处理完毕")

}

最后一句提醒:别在消费者里直接 close(ch),也别让多个 goroutine 竞争关同一个 channel——Go 的 channel 关闭是一次性操作,重复关闭 panic,且只能由生产方逻辑决定何时终止投递。

热门AI工具

更多
超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

519

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

616

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

1176

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

791

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

1746

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

4934

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

2274

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

4146

2025.06.17

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

100

2026.10.22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn