Go语言编程:使用条件变量Cond和channel通道实现多个生产者和消费者模型

xuguiyi00 2020-03-04

如题,使用条件变量Cond和channel通道实现多个生产者和消费者模型。Go语言天生带有C语言的基因,很多东西和C与很像,但是用起来 绝对比C语言方便。今天用Go语言来实现下多消费者和生产者模型。如果对C语言的多生产者和消费者模型感兴趣的可以看Linux系统编程:使用mutex互斥锁和条件变量实现多个生成者和消费者模型

代码实现
代码实现用了Cond条件变量和channel通道。

package main

import (
"fmt"
"math/rand"
"sync"
"time"
)

var cond sync.Cond

//生产者
func produce(out chan<- int, nu int) {
for {
cond.L.Lock()
//产品区满 等待消费者消费
for len(out) == 3 {
cond.Wait()
}
num := rand.Intn(1000)
out <- num
fmt.Printf("%dth ***producer produce***,num = %d,len(chan) = %d\n", nu, num, len(out))
cond.L.Unlock()

//生产了产品唤醒 消费者线程
cond.Signal()
//生产完了歇一会,给其他协程机会
time.Sleep(time.Second)
}
}

//消费者
func consume(in <-chan int, nu int) {
for {
cond.L.Lock()
//产品区空 等待生产者生产
for len(in) == 0 {
cond.Wait()
}
num := <-in
fmt.Printf("%dth ###consumer consume###,num = %d,len(chan) = %d\n", nu, num, len(in))
cond.L.Unlock()
cond.Signal()

//消费完了歇一会,给其他协程机会
time.Sleep(time.Millisecond * 500)
}
}

func main() {
//设置随机数种子
rand.Seed(time.Now().UnixNano())
quit := make(chan bool)
//产品区 使用channel模拟
product := make(chan int, 3)

//创建互斥锁和条件变量
cond.L = new(sync.Mutex)

//5个消费者
for i := 0; i < 5; i++ {
go produce(product, i)
}
//3个生产者
for i := 0; i < 3; i++ {
go consume(product, i)
}

//主协程阻塞 不结束
<-quit
}

运行效果

Go语言编程:使用条件变量Cond和channel通道实现多个生产者和消费者模型

相关推荐