读写锁

读写锁的特点:读和读不互斥,读和写是互斥的,写和写也是互斥的
原理:
const rwmutexMaxReaders = 1 << 30 :获取读锁的协程的数量最多为2的30次方
type RWMutex struct {
w Mutex // 互斥锁,获取写锁的时候,写协程先获取互斥锁,防止其它协程抢夺,每次只允许一个协程获取写锁
写协程获取互斥锁后和读协程按照先后顺序运行
writerSem uint32 // 写协程的信号,一个写协程在上面等待
readerSem uint32 // 读协程的信号,多个读协程在上面等待
readerCount atomic.Int32 // 读的协程的数量,是一个32位的原子数,是一个正数,也可能是一个负数
获取读锁的协程,该协程称为为读协程,加读锁过程:先对readerCount加1,如果+1后结果小于0就说明有一个写协程对其减少了rwmutexMaxReaders,此时,该读协程将进入等待队列进行等待,直到写协程完成,写协程完成后会对其加rwmutexMaxReaders得到一个数r,如果r等于0,说明没有等待的读协程。如果大于0,说明有r个读协程在等待,需要一一唤醒,等唤醒完成后需要解除互斥锁,允许其它协程获取写锁.
读协程解锁的过程:先对readerCount减少一,如果减少一之后小于0,说明有一个写的协程在等待,这个时候需要将readerWait(正在运行的读协程的数量,也就是写协程等待前正在运行的读协程的数量)减少1,当最后一个正读协程解锁后,就会唤醒等待的写协程。
写协程的加锁过程:先获取互斥锁,只能允许一个协程进入,此时该协程称为写协程,写协程首先将readerCount减去rwmutexMaxReaders。获取一个新值z,将z与rwmutexMaxReaders得到数字m。
当m大于0时,说明在该写协程之前有m个读协程正在运行,需要等待这m个读协程运行完。此时该协程需要将readerWait设置为m,等待唤醒。唤醒过程就是读协程的解锁过程
当m等于0时,该写协程获取写锁成功
写协程的解锁过程:readerCount先加上rwmutexMaxReaders,如果返回的值r大于0,说明在写协程之后,有几个读协程正在等待,此时需要将其一一唤醒,然后再释放掉互斥锁
readerWait atomic.Int32 // 写协程需要等待这些读协程全部完成,当最后一个读协程完成之后,就会唤醒等待的写协程
}

读协程获取锁的过程:对readerCount加1,如果小于0,说明写协程在它之前运行,需要等待写协程运行完

func (rw *RWMutex) RLock() {
    if race.Enabled {
        _ = rw.w.state
        race.Disable()
    }
    if rw.readerCount.Add(1) < 0 {
        // A writer is pending, wait for it.
        runtime_SemacquireRWMutexR(&rw.readerSem, false, 0)
    }
    if race.Enabled {
        race.Enable()
        race.Acquire(unsafe.Pointer(&rw.readerSem))
    }
}

读协程解锁的过程:
将readerCount减少1,如果结果小于0,说明有一个写协程在等待,这个时候需要等最后一个读协程运行完,将写协程唤醒

func (rw *RWMutex) RUnlock() {
    if race.Enabled {
        _ = rw.w.state
        race.ReleaseMerge(unsafe.Pointer(&rw.writerSem))
        race.Disable()
    }
    if r := rw.readerCount.Add(-1); r < 0 {
        // Outlined slow-path to allow the fast-path to be inlined
        rw.rUnlockSlow(r)
    }
    if race.Enabled {
        race.Enable()
    }
}

func (rw *RWMutex) rUnlockSlow(r int32) {
    if r+1 == 0 || r+1 == -rwmutexMaxReaders {
        race.Enable()
        fatal("sync: RUnlock of unlocked RWMutex")
    }
    // A writer is pending.
    if rw.readerWait.Add(-1) == 0 {
        // The last reader unblocks the writer.
        runtime_Semrelease(&rw.writerSem, false, 1)
    }
}

写协程加锁的过程:先获取互斥锁,然后将readerCount减少rwmutexMaxReaders,再判断在它之前有多少个读协程正在运行,如果没有读协程正在运行,那么它将运行,否则将readerWait设置为需要等待的读协程数量,然后进入等待状态,等最后一个读协程运行完,将其唤醒

func (rw *RWMutex) Lock() {
    if race.Enabled {
        _ = rw.w.state
        race.Disable()
    }
    // First, resolve competition with other writers.
    rw.w.Lock()
    // Announce to readers there is a pending writer.
    r := rw.readerCount.Add(-rwmutexMaxReaders) + rwmutexMaxReaders
    // Wait for active readers.
    if r != 0 && rw.readerWait.Add(r) != 0 {
        runtime_SemacquireRWMutex(&rw.writerSem, false, 0)
    }
    if race.Enabled {
        race.Enable()
        race.Acquire(unsafe.Pointer(&rw.readerSem))
        race.Acquire(unsafe.Pointer(&rw.writerSem))
    }
}

写协程解锁的过程:先对readerCount加上rwmutexMaxReaders,如果结果大于或者等于rwmutexMaxReaders,就说明解锁了多次,需要panic。假如大于0小于 rwmutexMaxReaders,就说明有读协程在等待,需要一一唤醒,等唤醒后,需要释放互斥锁

func (rw *RWMutex) Unlock() {
    if race.Enabled {
        _ = rw.w.state
        race.Release(unsafe.Pointer(&rw.readerSem))
        race.Disable()
    }

    // Announce to readers there is no active writer.
    r := rw.readerCount.Add(rwmutexMaxReaders)
    if r >= rwmutexMaxReaders {
        race.Enable()
        fatal("sync: Unlock of unlocked RWMutex")
    }
    // Unblock blocked readers, if any.
    for i := 0; i < int(r); i++ {
        runtime_Semrelease(&rw.readerSem, false, 0)
    }
    // Allow other writers to proceed.
    rw.w.Unlock()
    if race.Enabled {
        race.Enable()
    }
}

注意:无论是读锁还是写锁只能释放一次,否则会发生panic。所以为了避免重复释放,最好使用defer进行释放

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,992评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,212评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,535评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,197评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,310评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,383评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,409评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,191评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,621评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,910评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,084评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,763评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,403评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,083评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,318评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,946评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,967评论 2 351

推荐阅读更多精彩内容