美文网首页
Go - sync.Cond

Go - sync.Cond

作者: kyo1992 | 来源:发表于2021-05-16 09:35 被阅读0次

    设计目的

    为等待 / 通知场景下的并发问题提供支持。Cond 通常应用于等待某个条件的一组 goroutine,等条件变为 true 的时候,其中一个 goroutine 或者所有的 goroutine 都会被唤醒执行。

    顾名思义,Cond 是和某个条件相关,这个条件需要一组 goroutine 协作共同完成,在条件还没有满足的时候,所有等待这个条件的 goroutine 都会被阻塞住,只有这一组 goroutine 通过协作达到了这个条件,等待的 goroutine 才可能继续进行下去。

    基本用法

    标准库中的 Cond 并发原语初始化的时候,需要关联一个 Locker 接口的实例,一般我们使用 Mutex 或者 RWMutex。

    
    type Cond func NeWCond(l Locker) *Cond
    
    func (c *Cond) Broadcast()
    func (c *Cond) Signal()
    func (c *Cond) Wait()
    

    Cond 关联的 Locker 实例可以通过 c.L 访问,它内部维护着一个先入先出的等待队列。

    • Signal 方法,允许调用者 Caller 唤醒一个等待此 Cond 的 goroutine。如果此时没有等待的 goroutine,显然无需通知 waiter;如果 Cond 等待队列中有一个或者多个等待的 goroutine,则需要从等待队列中移除第一个 goroutine 并把它唤醒。 调用 Signal 方法时,不强求一定要持有 c.L 的锁。

    • Broadcast 方法,允许调用者 Caller 唤醒所有等待此 Cond 的 goroutine。如果此时没有等待的 goroutine,显然无需通知 waiter;如果 Cond 等待队列中有一个或者多个等待的 goroutine,则清空所有等待的 goroutine,并全部唤醒。 调用 Broadcast 方法时,也不强求一定持有 c.L 的锁。

    • Wait 方法,会把调用者 Caller 放入 Cond 的等待队列中并阻塞,直到被 Signal 或者 Broadcast 的方法从等待队列中移除并唤醒。调用 Wait 方法时必须要持有 c.L 的锁。

    例子

    
    func main() {
        c := sync.NewCond(&sync.Mutex{})
        var ready int
    
        for i := 0; i < 10; i++ {
            go func(i int) {
                time.Sleep(time.Duration(rand.Int63n(10)) * time.Second)
    
                // 加锁更改等待条件
                c.L.Lock()
                ready++
                c.L.Unlock()
    
                log.Printf("运动员#%d 已准备就绪\n", i)
                // 广播唤醒所有的等待者
                c.Broadcast()
            }(i)
        }
    
        c.L.Lock()
        for ready != 10 {
            c.Wait()
            log.Println("裁判员被唤醒一次")
        }
        c.L.Unlock()
    
        //所有的运动员是否就绪
        log.Println("所有运动员都准备就绪。比赛开始,3,2,1, ......")
    }
    

    它的复杂在于:一,这段代码有时候需要加锁,有时候可以不加;二,Wait 唤醒后需要检查条件;三,条件变量的更改,其实是需要原子操作或者互斥锁保护的。

    实现原理

    type Cond struct {
        noCopy noCopy
    
        // 当观察或者修改等待条件的时候需要加锁
        L Locker
    
        // 等待队列, 一个 Goroutine 的链表,它是实现同步机制的核心结构
        notify  notifyList
        checker copyChecker
    }
    
    func NewCond(l Locker) *Cond {
        return &Cond{L: l}
    }
    
    func (c *Cond) Wait() {
        c.checker.check()
        // 增加到等待队列中
        t := runtime_notifyListAdd(&c.notify)
        c.L.Unlock()
        // 阻塞休眠直到被唤醒
        runtime_notifyListWait(&c.notify, t)
        c.L.Lock()
    }
    
    func (c *Cond) Signal() {
        c.checker.check()
        runtime_notifyListNotifyOne(&c.notify)
    }
    
    func (c *Cond) Broadcast() {
        c.checker.check()
        runtime_notifyListNotifyAll(&c.notify)
    }
    
    type notifyList struct {
        wait uint32
        notify uint32
    
        lock mutex
        head *sudog
        tail *sudog
    }
    

    head 和 tail 分别指向的链表的头和尾,wait 和 notify 分别表示当前正在等待的和已经通知到的 Goroutine 的索引.

    Cond 的使用

    Cond 在实际项目中被使用的机会比较少,原因总结起来有两个。

    第一,同样的场景我们会使用其他的并发原语来替代。Go 特有的 Channel 类型,有一个应用很广泛的模式就是通知机制,这个模式使用起来也特别简单。所以很多情况下,我们会使用 Channel 而不是 Cond 实现 wait/notify 机制。

    第二,对于简单的 wait/notify 场景,比如等待一组 goroutine 完成之后继续执行余下的代码,我们会使用 WaitGroup 来实现。因为 WaitGroup 的使用方法更简单,而且不容易出错。比如,上面百米赛跑的问题,就可以很方便地使用 WaitGroup 来实现。

    总结

    • sync.Cond.Wait 在调用之前一定要使用获取互斥锁,否则会触发程序崩溃.
    • sync.Cond.Signal 唤醒的 Goroutine 都是队列最前面、等待最久的 Goroutine.
    • sync.Cond.Broadcast 会按照一定顺序广播通知等待的全部 Goroutine.
    • waiter goroutine 被唤醒不等于等待条件被满足

    相关文章

      网友评论

          本文标题:Go - sync.Cond

          本文链接:https://www.haomeiwen.com/subject/rxfcjltx.html