golang源码学习之RWMutex

读写锁(RWMutex)是在互斥锁(Mutex)的基础上构建的。Mutex笔记见另一篇

那么为什么需要读写锁呢?试想一下,在读多写少的场景下,mutex只允许一个goroutine进行读操作,而读操作往往是不会修改目标数据的。那我们为什么不允许多个goroutine并发执行读操作以提高效率呢,所以就有了读写锁。读锁和写锁互斥,但读锁和读锁不互斥。

下面是读锁和写锁的兼容性,是不是很像innoDB的排他锁和共享锁呢

数据结构

type RWMutex struct {
 w Mutex // held if there are pending writers
 writerSem uint32 // 唤醒写锁的信号
 readerSem uint32 // 唤醒读锁的信号
 readerCount int32 // 当前正在进行读操作的个数, 已经获取读锁 + 还没获取读锁
 readerWait int32 // 获取写锁前需要等待多少个读锁解锁
}

RLock

func (rw *RWMutex) RLock() {
 .....
 // 读锁个数 + 1, 在什么情况下会<0呢?在有写锁被获取的时候
 if atomic.AddInt32(&rw.readerCount, 1) < 0 {
 // sleep 等待写锁释放时唤醒
 runtime_SemacquireMutex(&rw.readerSem, false)
 }
 
 .....
}

Lock

func (rw *RWMutex) Lock() {
 .....
 // 写锁用到了互斥锁,所以同时只会有一把写锁
 rw.w.Lock()
 // 读锁个数(readerCount)瞬间减去一个非常大的数(const rwmutexMaxReaders = 1 << 30),为什么需要这样做呢?
 // 其实在上面RLock()中是通过判断readerCount来决定是否可以获取读锁的,当readerCount变成一个非常小的负数时会导致无法获取到读锁
 r := atomic.AddInt32(&rw.readerCount, -rwmutexMaxReaders) + rwmutexMaxReaders
 // 这里的r其实就是原来的readerCount。r!=0 说明还有读锁未释放
 if r != 0 && atomic.AddInt32(&rw.readerWait, r) != 0 {
 // sleep 等待读锁释放唤醒
 runtime_SemacquireMutex(&rw.writerSem, false)
 }
 ......
}

RUnlock

func (rw *RWMutex) RUnlock() {
 ....
 
 // readerCount - 1, 什么时候 r < 0呢? 解锁一个未加锁的锁
 if r := atomic.AddInt32(&rw.readerCount, -1); r < 0 {
 // r+1 == 0 解锁了一个空闲的锁。r+1 == -rwmutexMaxReaders 解锁了一个写锁
 if r+1 == 0 || r+1 == -rwmutexMaxReaders {
 race.Enable()
 throw("sync: RUnlock of unlocked RWMutex")
 }
 // 写锁前面等待释放的读锁释放完了,唤醒写锁
 if atomic.AddInt32(&rw.readerWait, -1) == 0 {
 runtime_Semrelease(&rw.writerSem, false)
 }
 }
 
 .....
}

从上面可以看锁其实也是有先后顺序的。虽然读锁和读锁兼容,但并不是一直存在读操作写锁就没机会执行了。

Unlock

func (rw *RWMutex) Unlock() {
 
 .....
 // lock()的时候减了一个非常大的数,现在加回去
 r := atomic.AddInt32(&rw.readerCount, rwmutexMaxReaders)
 // 正常情况下r应该远小于rwmutexMaxReaders。这里说明解锁了一个未锁的锁
 if r >= rwmutexMaxReaders {
 race.Enable()
 throw("sync: Unlock of unlocked RWMutex")
 }
 // 循环唤醒读锁
 for i := 0; i < int(r); i++ {
 runtime_Semrelease(&rw.readerSem, false)
 }
 // 解锁互斥锁
 rw.w.Unlock()
 ......
}

后记

读写锁整体来说还是比较简单的,没有啥繁琐的逻辑。

本文来自:简书

感谢作者:

查看原文:golang源码学习之RWMutex

• 编辑
• 预览