用Golang实现一个简单的生产者消费者并发模型-Golang-PHP中文网

用Golang实现一个简单的生产者消费者并发模型

P粉602998670

发布： 2025-08-30 08:40:01

原创

150人浏览过

Go语言通过goroutine和channel实现生产者消费者模型，生产者生成数据写入channel，消费者从channel读取处理，使用带缓冲channel和WaitGroup协调并发，确保线程安全与高效解耦。

用golang实现一个简单的生产者消费者并发模型

在Go语言中，通过goroutine和channel可以非常方便地实现生产者消费者模型。这个模型常用于解耦数据生成和处理过程，适用于任务队列、消息处理等场景。

模型基本结构

生产者负责生成数据并发送到channel，消费者从channel接收数据并处理。channel作为协程间通信的桥梁，保证线程安全。

下面是一个完整的示例，包含一个生产者、两个消费者，使用带缓冲的channel来提高效率：

立即学习“go语言免费学习笔记（深入）”；

燕雀Logo

为用户提供LOGO免费设计在线生成服务

101

查看详情

代码实现：

package main
<p>import (
"fmt"
"math/rand"
"sync"
"time"
)</p><p>// 生产者函数
func producer(ch chan<- int, wg <em>sync.WaitGroup) {
defer wg.Done()
for i := 0; i < 5; i++ {
num := rand.Intn(100)
ch <- num
fmt.Printf("生产者: 生成数字 %d\n", num)
time.Sleep(time.Millisecond </em> 100) // 模拟耗时
}
close(ch) // 生产完成，关闭channel
}</p><p>// 消费者函数
func consumer(ch <-chan int, id int, wg <em>sync.WaitGroup) {
defer wg.Done()
for num := range ch { // channel关闭后循环自动退出
fmt.Printf("消费者 %d: 处理数字 %d\n", id, num)
time.Sleep(time.Millisecond </em> 200) // 模拟处理时间
}
}</p><p>func main() {
// 创建带缓冲的channel，容量为3
ch := make(chan int, 3)</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">var wg sync.WaitGroup

// 启动生产者
wg.Add(1)
go producer(ch, &wg)

// 启动两个消费者
for i := 1; i <= 2; i++ {
    wg.Add(1)
    go consumer(ch, i, &wg)
}

// 等待所有goroutine完成
wg.Wait()
fmt.Println("所有任务完成")

登录后复制

}