Golangchannel实现生产者消费者模式示例-Golang-PHP中文网

Golangchannel实现生产者消费者模式示例

P粉602998670

发布： 2025-10-19 14:40:01

原创

680人浏览过

Go语言中channel是实现生产者消费者模式的核心工具，通过goroutine与channel协作实现并发任务处理；示例中3个生产者向带缓冲channel发送数据，2个消费者从中接收并处理，使用WaitGroup同步并由单独goroutine在生产结束后关闭channel，消费者通过range自动退出，整个过程线程安全且无需显式锁，适用于任务分发等场景。

golangchannel实现生产者消费者模式示例

在Go语言中，channel 是实现生产者消费者模式的天然工具。它通过 goroutine 和 channel 的协作，能够简洁高效地完成并发任务处理。下面是一个典型的生产者消费者模式示例，展示如何使用 channel 在多个生产者和消费者之间传递数据。

基本结构说明

在这个模式中：

生产者（Producer）：向 channel 发送数据
消费者（Consumer）：从 channel 接收并处理数据
channel：作为协程间通信的管道

完整代码示例

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

// 启动3个生产者
for i := 1; i <= 3; i++ {
    wg.Add(1)
    go producer(i, dataChan, &wg)
}

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

// 等待所有生产者完成
go func() {
    wg.Wait()
    close(dataChan) // 所有生产者结束后关闭channel
}()

// 等待所有消费者完成（消费者会在channel关闭后自动退出）
wg.Wait()
fmt.Println("所有任务完成")

登录后复制

}