golang源码学习之mutex

老实说呢,Mutex源码我看了好多遍,依旧没完全看懂。各种状态逻辑很难理解。(golang 1.12.7)
先来看看Mutex的核心注释
    // Mutex fairness.
    //
    // Mutex can be in 2 modes of operations: normal and starvation.
    // In normal mode waiters are queued in FIFO order, but a woken up waiter
    // does not own the mutex and competes with new arriving goroutines over
    // the ownership. New arriving goroutines have an advantage -- they are
    // already running on CPU and there can be lots of them, so a woken up
    // waiter has good chances of losing. In such case it is queued at front
    // of the wait queue. If a waiter fails to acquire the mutex for more than 1ms,
    // it switches mutex to the starvation mode.
    //
    // In starvation mode ownership of the mutex is directly handed off from
    // the unlocking goroutine to the waiter at the front of the queue.
    // New arriving goroutines don't try to acquire the mutex even if it appears
    // to be unlocked, and don't try to spin. Instead they queue themselves at
    // the tail of the wait queue.
    //
    // If a waiter receives ownership of the mutex and sees that either
    // (1) it is the last waiter in the queue, or (2) it waited for less than 1 ms,
    // it switches mutex back to normal operation mode.
    //
    // Normal mode has considerably better performance as a goroutine can acquire
    // a mutex several times in a row even if there are blocked waiters.
    // Starvation mode is important to prevent pathological cases of tail latency.
    
    Mutex是公平的。
    Mutex有两种模式:普通和饥饿模式。
    
    在普通模式下,等待者是按FIFO排队等待,但是被唤醒的等待者不是直接拥有锁,而是要和新来的goroutine
    竞争。由于新来的goroutine已经在CPU上了,所以唤醒的等待者很有可能获取不到锁。在这种情况下,被唤醒
    的等待者直接加入到队列首部。如果被唤醒的等待者超过1ms没获取到锁,那么mutex将切换到饥饿模式。
    
    在饥饿模式下,锁直接交给队列的第一个等待者。新来的goroutine不要去尝试获取锁,也不要自旋,即便是
    mutex处于unlock状态。直接将goroutine加到队列尾部。
    
    如果一个等待者获取了锁,在下面两种情况之一将转换为普通模式:1.等待者处于队列尾部;2.等待者少于1ms。
    
    普通模式具有良好的性能可以连续多次获取锁。
    饥饿模式解决了队尾一直无法获取锁的问题。
    

数据结构

/// src/sync/Mutex.go
type Mutex struct {
    state int32  // 状态
    sema  uint32 // 信号
}

Mutex的数据结构看似很简单就两个变量,但有时候看起来简单的东西其实并不简单。state是一个32bit的状态变量,它包含了4个部分。
第一位:锁标志。 0:未被锁; 1: 被锁
第二位:唤醒标志。 0:未被唤醒; 1:唤醒
第三位:饥饿标志。 0:普通模式; 1:饥饿模式
其余位:等待个数

常量

mutexLocked = 1 << iota //锁标志,对应state第一位
mutexWoken              //唤醒标志,对应state第二位
mutexStarving           //饥饿标志,对应state第三位
mutexWaiterShift = iota // 等待个数在state的位移
starvationThresholdNs = 1e6  // 1ms,用于判断被唤醒的等待者是否超时

Lock

func (m *Mutex) Lock() {
    // 如果state处于空闲状态,那么直接获取锁。state=0说明state的各个部分都为0
    // 000...000 000
    if atomic.CompareAndSwapInt32(&m.state, 0, mutexLocked) {
        if race.Enabled {
            race.Acquire(unsafe.Pointer(m))
        }
        return
    }

    var waitStartTime int64 //goroutine的开始时间
    starving := false
    awoke := false
    iter := 0
    old := m.state
    for {
        // 如果state已锁且处于正常模式,且可自旋,那么就进入自旋。
        // runtime_canSpin(iter)可以看出能否可以自旋和iter有一定的关系。
        // 如果进入runtime_canSpin(runtime/lock_sema.go)可以知道iter>=4就不能自旋了。当然
        // 能否自旋还和cpu核数有关。
        if old&(mutexLocked|mutexStarving) == mutexLocked && runtime_canSpin(iter) {
            // 如果goroutine自身未唤醒 + state未唤醒 + 等待数>0, 那么将state设为唤醒  
            if !awoke && old&mutexWoken == 0 && old>>mutexWaiterShift != 0 &&
                atomic.CompareAndSwapInt32(&m.state, old, old|mutexWoken) {
                // 自身设为唤醒
                awoke = true
            }
            // 自旋
            runtime_doSpin()
            iter++
            old = m.state
            continue
        }
        // new 指的是期望设置的状态
        new := old
        // 如果state 正常模式,new设置为locked状态。
        // 反过来 如果是饥饿状态呢?饥饿状态的规则是把锁交给队列头的等待者。当前的goroutine是不会获取到锁的
        if old&mutexStarving == 0 {
            new |= mutexLocked
        }
        // 如果state为locked或者饥饿模式,那么new等待者+1
        // case1 locked: 说明没获取到锁,加入等待 
        // case2 饥饿: 饥饿模式锁只会交给等待队列头,所以也不会获取到锁,加入等待队列
        if old&(mutexLocked|mutexStarving) != 0 {
            new += 1 << mutexWaiterShift
        }

        // goroutine 饥饿 + state locked, new 设为饥饿
        if starving && old&mutexLocked != 0 {
            new |= mutexStarving
        }
        if awoke {
            // goroutine是唤醒的,那么old state必然也是唤醒的。见自旋那部分
            if new&mutexWoken == 0 {
                throw("sync: inconsistent mutex state")
            }
            // new 设为未唤醒状态
            new &^= mutexWoken
        }
        // new 替换 old
        if atomic.CompareAndSwapInt32(&m.state, old, new) {
            // 如果old为unlock + 普通模式,那么就直接获取锁
            if old&(mutexLocked|mutexStarving) == 0 {
                break // locked the mutex with CAS
            }
            // 如果是新来的goroutine,设置开始时间
            queueLifo := waitStartTime != 0
            if waitStartTime == 0 {
                waitStartTime = runtime_nanotime()
            }
            // 进入sleep, 等待唤醒
            // queueLifo = true, 说明是唤醒的goroutine,加入队列头
            // queueLifo = false, 说明是新来的goroutine,加入队列尾
            runtime_SemacquireMutex(&m.sema, queueLifo)
            // 如果唤醒时间 > 1ms,那么goroutine设为饥饿模式? 那如果 < 1ms 呢?
            starving = starving || runtime_nanotime()-waitStartTime > starvationThresholdNs
            old = m.state
            
            // 如果state是饥饿模式,那么说明唤醒的队列头
            // 为什么是队列头呢? 其实在unlock()部分定义的
            if old&mutexStarving != 0 {
            
                // state 检查
                // 如果是饥饿模式唤醒,那么state肯定是unlock + 未唤醒。为什么一定是未唤醒呢? 看unlock()。
                // 饥饿状态被唤醒,等待者肯定也是>0的,不然怎么把锁交给队列头
                if old&(mutexLocked|mutexWoken) != 0 || old>>mutexWaiterShift == 0 {
                    throw("sync: inconsistent mutex state")
                }
                
                // 等待者 - 1 
                delta := int32(mutexLocked - 1<<mutexWaiterShift)
                // goroutine普通模式  或者  等待者只剩一个,设置成普通模式
                if !starving || old>>mutexWaiterShift == 1 {
                    delta -= mutexStarving
                }
                // 更新state, 获取到锁
                atomic.AddInt32(&m.state, delta)
                break
            }
            
            // 普通模式, 随机唤醒队列
            
            // goroutine 唤醒, 置零自旋
            awoke = true
            iter = 0
        } else {
            old = m.state
        }
    }

    if race.Enabled {
        race.Acquire(unsafe.Pointer(m))
    }
}

反正呢代码很简单,但逻辑有点复杂,我也没理很顺。

Unlock

func (m *Mutex) Unlock() {

    // 设置unlock
    new := atomic.AddInt32(&m.state, -mutexLocked)
    // 先减一个mutexLocked,再加一个mutexLocked,等于原始状态。
    // 如果正在解锁一个处于unlock的锁, 非法。
    if (new+mutexLocked)&mutexLocked == 0 {
        throw("sync: unlock of unlocked mutex")
    }
    
    // 正常模式
    if new&mutexStarving == 0 {
        old := new
        for {
            // 等待者为0   或者   state为锁定、唤醒、饥饿
            if old>>mutexWaiterShift == 0 || old&(mutexLocked|mutexWoken|mutexStarving) != 0 {
                return
            }
            // 设置成唤醒
            new = (old - 1<<mutexWaiterShift) | mutexWoken
            // 设置新值
            if atomic.CompareAndSwapInt32(&m.state, old, new) {
                // 随机唤醒等待队列
                runtime_Semrelease(&m.sema, false)
                return
            }
            old = m.state
        }
    } else {
        // 唤醒队列头
        // 注意:这里没有设置unlock,在上面的Lock()过程中,饥饿模式下对于被唤醒的等待者不用关心是否被锁也可以获取锁
        // 那对于新来的goroutine呢,会获取锁吗?新来的goroutine只能在普通模式才能获取锁。
        runtime_Semrelease(&m.sema, true)
    }
}

解锁过程相对于加锁要简单很多。普通模式随机唤醒,饥饿模式唤醒队列头。

后记

加锁过程涉及到很多逻辑状态,一时半会还无法理解,有缘再琢磨琢磨。

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

推荐阅读更多精彩内容