生产者消费者模式通过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)
}
}()
}
</font>使用for-range可以自动检测channel关闭,避免死锁。
生产者发送任务并关闭channel
生产者将任务写入channel。当所有任务发送完毕后,应关闭channel,通知消费者不再有新任务。
注意:只有生产者应关闭channel,多个关闭会引发panic。
go func() {
for i := 0; i < 10; i++ {
tasks <- i
}
close(tasks)
}()
</font>使用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)
}
}()
}
</font>主函数末尾调用wg.Wait()即可等待全部完成。
基本上就这些。利用channel的阻塞特性,Go能轻松实现线程安全的生产者消费者模型,无需手动加锁。关键是合理设置缓冲大小、正确关闭channel,并用WaitGroup协调生命周期。










