研究Golang的锁实现方式

type Mutex struct {
    waiting   int32 // 等待队列，记录等待获取锁的goroutine数量
    isLocked  int32 // 锁的状态标志，0代表未锁住，1代表已锁住
}

func (m *Mutex) Lock() {
    for !atomic.CompareAndSwapint32(&m.isLocked, 0, 1) { // 自旋等待获取锁
        runtime.Gosched()
    }
}

func (m *Mutex) Unlock() {
    atomic.StoreInt32(&m.isLocked, 0) // 释放锁，将状态标志设置为未锁住
}

type RWMutex struct {
    writerSem uint32    // 写入信号量，用于限制只能有一个goroutine写入
    readerSem uint32    // 读取信号量，用于限制多个goroutine同时读取
    readerCount int32   // 读取计数，记录当前同时读取的goroutine数量
    readerWait  int32   // 当前等待读取的goroutine数量
}

func (rw *RWMutex) Lock() {
    rw.lockWhile(func() {atomic.LoadUint32(&rw.readerSem) != 0 || atomic.LoadUint32(&rw.writerSem) != 0})
    atomic.AddUint32(&rw.writerSem, 1) // 获取写锁，递增写入信号量
}

func (rw *RWMutex) Unlock() {
    atomic.AddUint32(&rw.writerSem, ^uint32(0)) // 释放写锁，递减写入信号量
    rw.unlockWhile(func() {atomic.LoadInt32(&rw.readerCount) != 0}) // 释放读锁，根据读取计数判断是否需要唤醒等待读取的goroutine
}

func (rw *RWMutex) RLock() {
    rw.lockWhile(func() {atomic.LoadUint32(&rw.writerSem) != 0}) // 当有 goroutine 持有写锁时，自旋等待
    atomic.AddInt32(&rw.readerCount, 1) // 递增读取计数
}

func (rw *RWMutex) RUnlock() {
    atomic.AddInt32(&rw.readerCount, -1) // 递减读取计数
    rw.unlockWhile(func() {atomic.LoadInt32(&rw.readerCount) != 0}) // 根据读取计数判断是否需要唤醒等待读取的goroutine
}