如何在Golang中实现生产者消费者模式-Golang-PHP中文网

如何在Golang中实现生产者消费者模式

P粉602998670

发布： 2025-10-15 12:15:01

原创

190人浏览过

使用goroutine和channel实现生产者消费者模式，通过缓冲channel解耦数据生成与处理，生产者发送任务到channel，消费者从中接收并处理。

如何在golang中实现生产者消费者模式

在Go语言中实现生产者消费者模式，最常用的方式是结合 goroutine 和 channel。这种模式能有效解耦数据生成和处理逻辑，适用于任务队列、数据流处理等场景。

使用Channel作为任务缓冲区

Go的channel天然适合实现生产者消费者模型。生产者将任务发送到channel，消费者从channel接收并处理。

定义一个缓冲channel作为任务队列，可以避免生产者被阻塞：

tasks := make(chan int, 100) // 缓冲大小为100的任务通道

登录后复制

生产者向channel发送数据：

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

go func() {
    for i := 0; i < 1000; i++ {
        tasks <- i
    }
    close(tasks) // 所有任务发送完成后关闭channel
}()

登录后复制

消费者从channel读取并处理数据：

go func() {
    for task := range tasks {
        fmt.Printf("处理任务: %d\n", task)
        // 模拟处理耗时
        time.Sleep(time.Millisecond * 10)
    }
}()
</font><H3>启动多个消费者提升处理能力</H3><p>为了提高并发处理能力，可以启动多个消费者goroutine同时消费任务。</p>
                    <div class="aritcle_card">
                        <a class="aritcle_card_img" href="/ai/1067">
                            <img src="https://img.php.cn/upload/ai_manual/001/503/042/68b6c96ce645f755.png" alt="Boomy">
                        </a>
                        <div class="aritcle_card_info">
                            <a href="/ai/1067">Boomy</a>
                            <p>AI音乐生成工具，创建生成音乐,与世界分享.</p>
                            <div class="">
                                <img src="/static/images/card_xiazai.png" alt="Boomy">
                                <span>272</span>
                            </div>
                        </div>
                        <a href="/ai/1067" class="aritcle_card_btn">
                            <span>查看详情</span>
                            <img src="/static/images/cardxiayige-3.png" alt="Boomy">
                        </a>
                    </div>
                <p>例如启动5个消费者：</p><font face="Courier New"><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for i := 0; i < 5; i++ {
    go func(workerID int) {
        for task := range tasks {
            fmt.Printf("Worker %d 处理任务: %d\n", workerID, task)
            time.Sleep(time.Millisecond * 10)
        }
    }(i)
}

登录后复制

所有消费者通过同一个channel读取任务，Go runtime会自动保证线程安全。

等待所有消费者完成

使用sync.WaitGroup确保主程序在所有消费者处理完毕后再退出。

var wg sync.WaitGroup
wg.Add(5) // 启动5个消费者
<p>for i := 0; i < 5; i++ {
go func(workerID int) {
defer wg.Done()
for task := range tasks {
fmt.Printf("Worker %d 处理任务: %d\n", workerID, task)
time.Sleep(time.Millisecond * 10)
}
}(i)
}</p><p>// 等待所有消费者完成
wg.Wait()

登录后复制

完整示例代码

以下是一个完整的生产者消费者实现：

package main
<p>import (
"fmt"
"sync"
"time"
)</p><p>func main() {
tasks := make(chan int, 100)
var wg sync.WaitGroup</p><pre class="brush:php;toolbar:false;"><code>// 生产者
go func() {
    for i := 0; i < 100; i++ {
        tasks <- i
    }
    close(tasks)
}()

// 消费者
for i := 0; i < 3; i++ {
    wg.Add(1)
    go func(workerID int) {
        defer wg.Done()
        for task := range tasks {
            fmt.Printf("Worker %d 处理任务 %d\n", workerID, task)
            time.Sleep(time.Millisecond * 50)
        }
    }(i)
}

wg.Wait()
fmt.Println("所有任务已完成")

登录后复制

}

基本上就这些。利用channel的并发安全特性和goroutine的轻量性，Go实现生产者消费者模式非常简洁高效。

以上就是如何在Golang中实现生产者消费者模式的详细内容，更多请关注php中文网其它相关文章！