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

如何使用Golang实现生产者消费者模式

P粉602998670

发布： 2025-10-11 18:53:01

原创

675人浏览过

生产者消费者模式通过goroutine和channel实现，生产者发送数据到缓冲channel，消费者并发接收处理；2. 使用make(chan int, 100)创建带缓冲的channel作为任务队列，避免生产者阻塞；3. 启动多个消费者goroutine从channel读取并处理任务，提升并发性能。

如何使用golang实现生产者消费者模式

在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关闭，避免死锁。

豆绘AI

豆绘AI是国内领先的AI绘图与设计平台，支持照片、设计、绘画的一键生成。

485

查看详情

生产者发送任务并关闭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协调生命周期。

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