如何用Golang实现生产者消费者模型_Golang 生产者消费者实践-Golang-PHP中文网

如何用Golang实现生产者消费者模型_Golang 生产者消费者实践

P粉602998670

发布： 2025-11-17 14:02:02

原创

947人浏览过

使用Golang的goroutine和channel可高效实现生产者消费者模型；2. 定义Task结构体并通过缓冲channel传递任务；3. 启动多个消费者协程从channel接收并处理任务，实现解耦与并发。

如何用golang实现生产者消费者模型_golang 生产者消费者实践

生产者消费者模型是并发编程中的经典问题，用于解决数据生成与处理速度不匹配的场景。在 Golang 中，通过 goroutine 和 channel 可以非常简洁高效地实现这一模型。下面介绍一种实用且易于扩展的实现方式。

使用 Channel 构建基本模型

Go 的 channel 天然适合解耦生产者和消费者。生产者将任务发送到 channel，消费者从 channel 接收并处理。

定义一个任务结构体，便于传递不同类型的数据：

type Task struct {
    ID   int
    Data string
}

登录后复制

创建缓冲 channel 作为任务队列：

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

tasks := make(chan Task, 100)

登录后复制

启动多个消费者监听该 channel：

for i := 0; i < 3; i++ {
    go func(workerID int) {
        for task := range tasks {
            fmt.Printf("Worker %d processing task %d: %s\n", workerID, task.ID, task.Data)
            time.Sleep(time.Second) // 模拟处理耗时
        }
    }(i)
}
</font><p>生产者向 channel 发送任务：</p>
                    <div class="aritcle_card">
                        <a class="aritcle_card_img" href="/ai/909">
                            <img src="https://img.php.cn/upload/ai_manual/000/000/000/175679993814027.png" alt="歌者PPT">
                        </a>
                        <div class="aritcle_card_info">
                            <a href="/ai/909">歌者PPT</a>
                            <p>歌者PPT，AI 写 PPT 永久免费</p>
                            <div class="">
                                <img src="/static/images/card_xiazai.png" alt="歌者PPT">
                                <span>197</span>
                            </div>
                        </div>
                        <a href="/ai/909" class="aritcle_card_btn">
                            <span>查看详情</span>
                            <img src="/static/images/cardxiayige-3.png" alt="歌者PPT">
                        </a>
                    </div>
                <font face="Courier New"><pre class="brush:php;toolbar:false;">
for i := 0; i < 5; i++ {
    tasks <- Task{ID: i, Data: fmt.Sprintf("data-%d", i)}
}
close(tasks) // 所有任务发送完成后关闭 channel
</font><H3>控制并发与优雅退出</H3><p>实际应用中，需要控制消费者数量，并确保程序能正确退出。可以使用 sync.WaitGroup 等待所有消费者完成。</p><font face="Courier New"><pre class="brush:php;toolbar:false;">
var wg sync.WaitGroup
<p>// 启动消费者
for i := 0; i < 3; i++ {
wg.Add(1)
go func(workerID int) {
defer wg.Done()
for task := range tasks {
fmt.Printf("Worker %d handling task %d\n", workerID, task.ID)
time.Sleep(500 * time.Millisecond)
}
}(i)
}</p><p>// 生产者
for i := 0; i < 10; i++ {
tasks <- Task{ID: i, Data: fmt.Sprintf("payload-%d", i)}
}
close(tasks)</p><p>// 等待所有消费者结束
wg.Wait()
</font></p><H3>加入错误处理与超时机制</H3><p>增强健壮性，可为每个任务添加结果反馈通道，或使用 context 控制执行时间。</p><p>例如，使用 context.WithTimeout 防止任务卡死：</p><font face="Courier New"><pre class="brush:php;toolbar:false;">
go func() {
    ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
    defer cancel()
<pre class='brush:php;toolbar:false;'>select {
case tasks <- Task{ID: 99, Data: "urgent"}:
    fmt.Println("Urgent task sent")
case <-ctx.Done():
    fmt.Println("Failed to send urgent task:", ctx.Err())
}

登录后复制

}()