如题,使用条件变量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 }
运行效果
|
请发表评论