go条件变量和互斥锁结合channel使用

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\n", 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\n", 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

..................

posted @ 2021-01-21 12:07  大雾哥  阅读(228)  评论(0编辑  收藏  举报