生产者消费者模式通过goroutine和channel实现,生产者发送数据到缓冲channel,消费者并发接收处理;2. 使用make(chan int, 100)创建带缓冲的channel作为任务队列,避免生产者阻塞;3. 启动多个消费者goroutine从channel读取并处理任务,提升并发性能。

在Go语言中,生产者消费者模式可以通过goroutine和channel高效实现。这种模式的核心是让生产者将任务或数据发送到通道,消费者从通道接收并处理,从而解耦生产和消费过程,提升程序并发性能。
使用channel作为任务队列
Go的channel天然适合做生产者消费者之间的通信桥梁。定义一个缓冲channel来存放任务,生产者向其中发送数据,消费者通过for-range或select监听该channel。
例如,创建一个整数类型的缓冲channel:
tasks := make(chan int, 100)
这个channel可以缓存最多100个任务,避免生产者被阻塞(除非队列满)。
立即学习“go语言免费学习笔记(深入)”;
启动多个消费者goroutine
消费者通常以多个goroutine形式运行,提高处理能力。每个消费者从channel读取数据,处理完成后标记完成。
示例代码:
for i := 0; i < 3; i++ {
go func() {
for task := range tasks {
fmt.Printf("处理任务: %d\n", task)
// 模拟处理耗时
time.Sleep(time.Millisecond * 100)
}
}()
}
使用for-range可以自动检测channel关闭,避免死锁。
在现实生活中的购物过程,购物者需要先到商场,找到指定的产品柜台下,查看产品实体以及标价信息,如果产品合适,就将该产品放到购物车中,到收款处付款结算。电子商务网站通过虚拟网页的形式在计算机上摸拟了整个过程,首先电子商务设计人员将产品信息分类显示在网页上,用户查看网页上的产品信息,当用户看到了中意的产品后,可以将该产品添加到购物车,最后使用网上支付工具进行结算,而货物将由公司通过快递等方式发送给购物者
生产者发送任务并关闭channel
生产者将任务写入channel。当所有任务发送完毕后,应关闭channel,通知消费者不再有新任务。
注意:只有生产者应关闭channel,多个关闭会引发panic。
go func() {
for i := 0; i < 10; i++ {
tasks <- i
}
close(tasks)
}()
使用WaitGroup等待消费者完成
如果需要等待所有消费者处理完才退出主程序,可以用sync.WaitGroup配合关闭机制。
改进后的消费者:
var wg sync.WaitGroup
for i := 0; i < 3; i++ {
wg.Add(1)
go func() {
defer wg.Done()
for task := range tasks {
fmt.Printf("处理任务: %d\n", task)
time.Sleep(time.Millisecond * 100)
}
}()
}
主函数末尾调用wg.Wait()即可等待全部完成。
基本上就这些。利用channel的阻塞特性,Go能轻松实现线程安全的生产者消费者模型,无需手动加锁。关键是合理设置缓冲大小、正确关闭channel,并用WaitGroup协调生命周期。









