Golangchannel生产者消费者示例详解

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 &lt;= 3; i++ {
    wg.Add(1)
    go producer(i, dataChan, &amp;wg)
}

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

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

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

}