如何在Golang中实现生产者消费者模式_Golang生产者消费者模式实践汇总

Golang中生产者消费者模式依赖goroutine和channel实现，通过无缓冲或带缓冲channel进行同步通信，支持多消费者并行处理，可结合WaitGroup或done信道控制完成信号，合理关闭channel避免泄漏。

在Golang中实现生产者消费者模式，主要依赖于goroutine和channel。Go的并发模型天然适合这种场景：生产者将任务发送到channel，消费者从channel接收并处理。下面介绍几种常见的实践方式，帮助你在实际项目中灵活应用。

使用无缓冲channel实现基础模式

最简单的实现方式是使用无缓冲channel，生产者和消费者通过它同步通信。

特点：发送和接收必须同时就绪，否则阻塞。

package main
import (
"fmt"
"time"
)
func producer(ch chan<- int) {
for i := 1; i <= 5; i++ {
ch <- i
fmt.Printf("生产者发送: %d\n", i)
}
close(ch) // 关闭channel，通知消费者结束
}
func consumer(ch <-chan int, done chan<- bool) {
for data := range ch {
fmt.Printf("消费者处理: %d\n", data)
time.Sleep(100 * time.Millisecond) // 模拟处理耗时
}
done <- true
}
func main() {
ch := make(chan int)        // 无缓冲channel
done := make(chan bool)
go producer(ch)
go consumer(ch, done)

<-done // 等待消费者完成

}

使用带缓冲channel提升吞吐量
当生产速度偶尔快于消费速度时，可以使用带缓冲的channel避免阻塞。
立即学习“go语言免费学习笔记（深入）”；
适用场景：突发性任务、批量处理。
说明：缓冲区满了才会阻塞生产者。
示例修改：
ch := make(chan int, 10) // 缓冲大小为10

这样生产者可以在消费者未及时处理时继续发送最多10个任务，提高系统响应能力。
多个消费者提升处理能力
为了加快消费速度，可以启动多个消费者goroutine。

							

								
								

									剪映
									
一款全能易用的桌面端剪辑软件
								
								下载 
							
						
关键点：所有消费者应共享同一个输入channel，使用close通知结束。
实现方式：
const workerNum = 3
for i := 0; i < workerNum; i++ {
go consumer(ch, done)
}
// 等待所有消费者完成
for i := 0; i < workerNum; i++ {
<-done
}

注意：done channel需接收多次，对应每个worker一个信号。
控制生产或消费速率（可选）
在高负载场景下，可能需要限流。
可以通过time.Ticker或第三方库如golang.org/x/time/rate实现。
简单示例：生产者每200ms发送一次
tick := time.NewTicker(200 * time.Millisecond)
defer tick.Stop()
for i := 1; i <= 10; i++ {
<-tick.C
ch <- i
}

使用sync.WaitGroup替代done channel（更简洁）
当不需要返回值时，WaitGroup更直观。
var wg sync.WaitGroup
ch := make(chan int, 5)
// 启动消费者
wg.Add(1)
go func() {
defer wg.Done()
for data := range ch {
fmt.Println("处理:", data)
}
}()
// 生产
for i := 1; i <= 10; i++ {
ch <- i
}
close(ch)
wg.Wait() // 等待消费者完成

基本上就这些。Golang的channel和goroutine让生产者消费者模式变得简单高效。根据实际需求选择是否加缓冲、是否多消费者、是否限流。核心是合理关闭channel，避免panic或goroutine泄漏。不复杂但容易忽略细节。