携手共进，深入剖析sync.Cond

2023-11-25 17:49:07

sync.Cond 是 Go 并发编程中的一项重要工具，它允许一个或多个 goroutine 在满足某个条件之前进入等待状态，并在条件满足时被唤醒。sync.Cond 通常与锁（如 Mutex 或 RWMutex）配合使用，以确保对共享数据的并发访问是安全的。

sync.Cond 的工作原理

sync.Cond 的核心是一个等待队列，goroutine 可以通过调用 Wait 方法进入等待队列。当条件满足时，调用 Signal 或 Broadcast 方法可以唤醒等待队列中的 goroutine。

sync.Cond 的 Wait 方法会将当前 goroutine 放入等待队列，并释放锁（如果当前 goroutine 持有锁）。当条件满足时，调用 Signal 或 Broadcast 方法会将等待队列中的 goroutine 唤醒，并重新获取锁。

sync.Cond 的使用示例

以下是一个使用 sync.Cond 实现 FIFO 队列的示例：

import (
    "sync"
)

type FIFO struct {
    cond *sync.Cond
    queue []interface{}
}

func NewFIFO() *FIFO {
    return &FIFO{
        cond: sync.NewCond(&sync.Mutex{}),
        queue: make([]interface{}, 0),
    }
}

func (f *FIFO) Enqueue(v interface{}) {
    f.cond.L.Lock()
    defer f.cond.L.Unlock()

    f.queue = append(f.queue, v)
    f.cond.Signal()
}

func (f *FIFO) Dequeue() interface{} {
    f.cond.L.Lock()
    defer f.cond.L.Unlock()

    for len(f.queue) == 0 {
        f.cond.Wait()
    }

    v := f.queue[0]
    f.queue = f.queue[1:]
    return v
}