package main import ( "fmt" "math/rand" "sync" "time" ) var cond sync.Cond // 定义全局条件变量 func product(out chan<- int, index int) { for { // 先加锁 cond.L.Lock() // 判断缓冲区是否满 for len(out) == 5 { // a)阻塞等待条件变量满足 // b)释放已掌握的互斥锁相当于cond.L.Unlock()。 注意:两步为一个原子操作。 // c)当被唤醒,Wait()函数返回时,解除阻塞并重新获取互斥锁。相当于cond.L.Lock() cond.Wait() } num := rand.Intn(1000) out <- num fmt.Printf("生产者%dth,生产:%d ", index, num) // 访问公共区结束,并且打印结束,解锁 cond.L.Unlock() // 唤醒阻塞在条件变量上的 消费者 cond.Signal() time.Sleep(time.Millisecond * 300) } } func consumer(in <-chan int, index int) { for { // 先加锁 cond.L.Lock() // 判断 缓冲区是否为空 for len(in) == 0 { // a)阻塞等待条件变量满足 // b)释放已掌握的互斥锁相当于cond.L.Unlock()。 注意:两步为一个原子操作。 // c)当被唤醒,Wait()函数返回时,解除阻塞并重新获取互斥锁。相当于cond.L.Lock() cond.Wait() } num := <-in fmt.Printf("----消费者%dth,消费:%d ", index, num) // 访问公共区结束后,解锁 cond.L.Unlock() // 唤醒 阻塞在条件变量上的 生产者 cond.Signal() time.Sleep(time.Millisecond * 200) } } func main() { ch := make(chan int, 5) quit := make(chan bool) rand.Seed(time.Now().UnixNano()) // 指定条件变量 使用的锁 cond.L = new(sync.Mutex) for i := 0; i < 5; i++ { go product(ch, i+1) } for i := 0; i < 5; i++ { go consumer(ch, i+1) } <-quit }
打印:
生产者1th,生产:386
----消费者5th,消费:386
生产者3th,生产:975
----消费者1th,消费:975
生产者2th,生产:615
----消费者3th,消费:615
生产者4th,生产:207
生产者5th,生产:140
----消费者2th,消费:207
----消费者4th,消费:140
..................