golang源码学习之mutex

· · 645 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

老实说呢,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)
    }
}

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

后记

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


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:

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

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

645 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传