互斥锁
1、互斥锁概念
- 对于同一份资源来说,为了保证 在多个线程或者协程同时访问的时候,保证数据的读取不会出现错误。
- 互斥锁相对于 读写锁而言更加绝对一些,互斥锁不论是读写,加过锁后,都是互斥的,需要等待
2、Mutex 数据结构
type Mutex struct {
state int32 // 互斥锁状态
sema uint32 // 信号量
}
- sema 表示信号量,用来控制 goroutine 的 阻塞休眠和唤醒。协程阻塞时等待该信号量,协程解锁是释放该信号量并唤醒等待信号量的协程
- state 表示 互斥锁的状态,0表示 未上锁,1表示上锁状态
3、state 状态组成
state 一共32位,最低三位依次是 Locked、Woken、Starving,剩下的其他位表示 等待锁的goroutine
- Waiter 表示等待锁释放的goroutine数量
- starving 表示当前锁释放处于饥饿状态,0 不饿 1 饥饿。饥饿状态说是有协程阻塞超过了1ms
- woken 表示唤醒状态,0 没唤醒,1 唤醒状态。唤醒状态表示正在加锁
- locked 表述是否锁定 0 未锁定 1 已锁定
协程之间抢占锁实际就是看谁能给locked 赋值。赋值为1说明抢占成功,抢不到的话就是0.
抢不到的话就等待 sema 的信号量,一旦有协程解锁,等待的协程就会被依次唤醒
woken 和 starving 主要用户控制协程之间的抢锁过程
4、Mutex 的方法
Mutex 对外仅暴露两个方法,lock 和 unlock。加红锁的是私有的方法
Lock
加锁过程
1、无阻塞加锁图解
当前锁是空闲状态,且没有协程抢占。
加锁前判断Locked 是否是 0 , 如果是0 的话 改为 1 ,加锁成功。
仅 Locked 状态位变为1 ,其他位置不变化
2、阻塞加锁图解
协程B加锁的时候,发现该锁已经被协程A占用了
此时 Waiter 计数器会增加1,协程B 被阻塞,知道协程A释放锁。Locked 变为0 ,协程B才会被唤醒
加锁代码
// Lock locks m.
// If the lock is already in use, the calling goroutine
// blocks until the mutex is available.
func (m *Mutex) Lock() {
// Fast path: grab unlocked mutex.
if atomic.CompareAndSwapInt32(&m.state, 0, mutexLocked) {
if race.Enabled {
race.Acquire(unsafe.Pointer(m))
}
return
}
// Slow path (outlined so that the fast path can be inlined)
m.lockSlow()
}
- 首先使用 CAS 方法(CompareAndSwapInt32)判断是否可以获取资源。如果可以获取资源的话,就直接修改 Locked 的值,如果获取不到的话就 执行 lockSlow() 方法,自旋抢锁,直到拿到锁
1、CompareAndSwapInt32 方法介绍
-
CAS 是原子操作的一种,是在包 sync.atomic 下面
-
原子操作即是进行过程中不能被中断的操作。针对某个值的原子操作在被进行的过程中,CPU绝不会再去进行其他的针对该值的操作。
为了实现这样的严谨性,原子操作仅会由一个独立的CPU指令代表和完成
-
对应的其他方法还有 增或减(add)、比较并交换(CompareAndSwap)、载入(Load)、存储(Store)、交换(Swap)
-
对应的操作类型 :int32,int64,uint32,uint64,uintptr,unsafe.Pointer
我们互斥锁这里用到的是
func CompareAndSwapInt32(addr *int32, old, new int32) (swapped bool)
1、调用函数后,先判断 addr 指针指向的值和传入的 old 是否相同
2、如果相同,会用传入的 new 对addr的值进行替换。反之不替换
2、lockSlow 方法
加锁失败后就会进入 lockSlow 方法,自旋抢锁
func (m *Mutex) lockSlow() {
v