golang 重要知识：mutex_yue_xin_tech-CSDN博客 - 《golang 硬核学习》

摘要
mutex 初步认识
mutex 状态标志位
mutex 正常模式
mutex 饥饿模式
mutex 自旋
mutex 的 Lock() 过程
Unlock 过程
mutex 代码详解
- mutex Lock() 代码详解：
- mutex Unlock() 代码详解：

摘要

Go 号称是为了高并发而生的，在高并发场景下，势必会涉及到对公共资源的竞争。当对应场景发生时，我们经常会使用 mutex 的 Lock() 和 Unlock() 方法来占有或释放资源。虽然调用简单，但 mutex 的内部却涉及挺多的。今天，就让我们好好研究一下。

mutex 初步认识

mutex 的源码主要是在 src/sync/mutex.go文件里，它的结构体比较简单，如下:

type Mutex struct {
    state int32
    sema  uint32
}

我们可以看到有一个字段 sema，它表示信号量标记位。所谓的信号量是用于 Goroutine 之间阻塞或唤醒的。这有点像操作系统里的 PV 原语操作，我们先来认识下 PV 原语操作：

PV 原语解释：
通过操作信号量 S 来处理进程间的同步与互斥的问题。
S>0：表示有 S 个资源可用；S=0 表示无资源可用；S<0 绝对值表示等待队列或链表中的进程个数。信号量 S 的初值应大于等于 0。
P 原语：表示申请一个资源，对 S 原子性的减 1，若减 1 后仍 S>=0，则该进程继续执行；若减 1 后 S<0，表示已无资源可用，需要将自己阻塞起来，放到等待队列上。
V 原语：表示释放一个资源，对 S 原子性的加 1；若加 1 后 S>0，则该进程继续执行；若加 1 后 S<=0，表示等待队列上有等待进程，需要将第一个等待的进程唤醒。

通过上面的解释，mutex 就可以利用信号量来实现 goroutine 的阻塞和唤起了。

其实 mutex 本质上就是一个关于信号量的阻塞唤起操作。

当 goroutine 不能占有锁资源的时候会被阻塞挂起，此时不能继续执行后面的代码逻辑。

当 mutex 释放锁资源时，则会继续唤起之前的 goroutine 去抢占锁资源。

至于 mutex 的 state 状态字段则是用来做状态流转的，这些状态值涉及到了一些概念，下面我们具体来解释一番。

mutex 状态标志位

mutex 的 state 有 32 位，它的低 3 位分别表示 3 种状态：唤醒状态、上锁状态、饥饿状态，剩下的位数则表示当前阻塞等待的 goroutine 数量。

mutex 会根据当前的 state 状态来进入正常模式、饥饿模式或者是自旋。

mutex 正常模式

当 mutex 调用 Unlock() 方法释放锁资源时，如果发现有等待唤起的 Goroutine 队列时，则会将队头的 Goroutine 唤起。

队头的 goroutine 被唤起后，会调用 CAS 方法去尝试性的修改 state 状态，如果修改成功，则表示占有锁资源成功。

(注：CAS 在 Go 里用 atomic.CompareAndSwapInt32(addr *int32, old, new int32) 方法实现，CAS 类似于乐观锁作用，修改前会先判断地址值是否还是 old 值，只有还是 old 值，才会继续修改成 new 值，否则会返回 false 表示修改失败。)

mutex 饥饿模式

由于上面的 Goroutine 唤起后并不是直接的占用资源，还需要调用 CAS 方法去尝试性占有锁资源。如果此时有新来的 Goroutine，那么它也会调用 CAS 方法去尝试性的占有资源。

但对于 Go 的调度机制来讲，会比较偏向于 CPU 占有时间较短的 Goroutine 先运行，而这将造成一定的几率让新来的 Goroutine 一直获取到锁资源，此时队头的 Goroutine 将一直占用不到，导致饿死。

针对这种情况，Go 采用了饥饿模式。即通过判断队头 Goroutine 在超过一定时间后还是得不到资源时，会在 Unlock 释放锁资源时，直接将锁资源交给队头 Goroutine，并且将当前状态改为饥饿模式。

后面如果有新来的 Goroutine 发现是饥饿模式时，则会直接添加到等待队列的队尾。

mutex 自旋

如果 Goroutine 占用锁资源的时间比较短，那么每次都调用信号量来阻塞唤起 goroutine，将会很浪费资源。

因此在符合一定条件后，mutex 会让当前的 Goroutine 去空转 CPU，在空转完后再次调用 CAS 方法去尝试性的占有锁资源，直到不满足自旋条件，则最终会加入到等待队列里。

自旋的条件如下：

还没自旋超过 4 次
多核处理器
GOMAXPROCS > 1
p 上本地 Goroutine 队列为空

可以看出，自旋条件还是比较严格的，毕竟这会消耗 CPU 的运算能力。

mutex 的 Lock() 过程

首先，如果 mutex 的 state = 0，即没有谁在占有资源，也没有阻塞等待唤起的 goroutine。则会调用 CAS 方法去尝试性占有锁，不做其他动作。

如果不符合 m.state = 0，则进一步判断是否需要自旋。

当不需要自旋又或者自旋后还是得不到资源时，此时会调用 runtime_SemacquireMutex 信号量函数，将当前的 goroutine 阻塞并加入等待唤起队列里。

当有锁资源释放，mutex 在唤起了队头的 goroutine 后，队头 goroutine 会尝试性的占有锁资源，而此时也有可能会和新到来的 goroutine 一起竞争。

当队头 goroutine 一直得不到资源时，则会进入饥饿模式，直接将锁资源交给队头 goroutine，让新来的 goroutine 阻塞并加入到等待队列的队尾里。

对于饥饿模式将会持续到没有阻塞等待唤起的 goroutine 队列时，才会解除。

Unlock 过程

mutex 的 Unlock() 则相对简单。同样的，会先进行快速的解锁，即没有等待唤起的 goroutine，则不需要继续做其他动作。

如果当前是正常模式，则简单的唤起队头 Goroutine。如果是饥饿模式，则会直接将锁交给队头 Goroutine，然后唤起队头 Goroutine，让它继续运行。

mutex 代码详解

好了，上面大体流程讲完了，下面将会把详细的代码流程呈上，让大家能更详细的知道 mutex 的 Lock()、Unlock() 方法逻辑。

mutex Lock() 代码详解：


func (m *Mutex) Lock() {
    if atomic.CompareAndSwapInt32(&m.state, 0, mutexLocked) {
        if race.Enabled {
            race.Acquire(unsafe.Pointer(m))
        }
        return
    }
    m.lockSlow()
}
func (m *Mutex) lockSlow() {
  var waitStartTime int64  
  starving := false 
  awoke := false 
  iter := 0 
  old := m.state 
  for {
    if old&(mutexLocked|mutexStarving) == mutexLocked && runtime_canSpin(iter) {
      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 := old
    if old&mutexStarving == 0 {
      new |= mutexLocked
    }
    if old&(mutexLocked|mutexStarving) != 0 {
      new += 1 << mutexWaiterShift
    }
    if starving && old&mutexLocked != 0 {
      new |= mutexStarving
    }
    if awoke {
      if new&mutexWoken == 0 {
        throw("sync: inconsistent mutex state")
      }
      new &^= mutexWoken
    }
    if atomic.CompareAndSwapInt32(&m.state, old, new) {
      if old&(mutexLocked|mutexStarving) == 0 {
        break
      }
      queueLifo := waitStartTime != 0
      if waitStartTime == 0 {
        waitStartTime = runtime_nanotime()
      }
      runtime_SemacquireMutex(&m.sema, queueLifo, 1)
      starving = starving || runtime_nanotime()-waitStartTime > starvationThresholdNs
      old = m.state
      if old&mutexStarving != 0 {
        if old&(mutexLocked|mutexWoken) != 0 || old>>mutexWaiterShift == 0 {
          throw("sync: inconsistent mutex state")
        }
        delta := int32(mutexLocked - 1<<mutexWaiterShift)
        if !starving || old>>mutexWaiterShift == 1 {
          delta -= mutexStarving
        }
        atomic.AddInt32(&m.state, delta)
        break
      }
      awoke = true
      iter = 0
    } else {
      old = m.state
    }
  }
  if race.Enabled {
    race.Acquire(unsafe.Pointer(m))
  }
}

mutex Unlock() 代码详解：


func (m *Mutex) Unlock() {
    if race.Enabled {
        _ = m.state
        race.Release(unsafe.Pointer(m))
    }
    new := atomic.AddInt32(&m.state, -mutexLocked)
    if new != 0 {
        m.unlockSlow(new)
    }
}
func (m *Mutex) unlockSlow(new int32) {
  if (new+mutexLocked)&mutexLocked == 0 {
    throw("sync: unlock of unlocked mutex")
  }
  if new&mutexStarving == 0 {
    old := new
    for {
      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, 1)
        return
      }
      old = m.state
    }
  } else {
    runtime_Semrelease(&m.sema, true, 1)
  }
}

感兴趣的朋友可以搜一搜公众号「阅新技术」，关注更多的推送文章。
可以的话，就顺便点个赞、留个言、分享下，感谢各位支持！
阅新技术，阅读更多的新知识。
golang 重要知识：mutex_yue_xin_tech-CSDN博客 - 图1
https://blog.csdn.net/h_l_f/article/details/118462433