Go语言互斥锁(sync.Mutex)和读写互斥锁(sync.RWMutex)

参考:
http://c.biancheng.net/view/107.html

关键点

通过关键词汇,实现快速理解,记忆目的

1、Go语言包中的 sync 包提供了两种类型:

  • sync.Mutex

    • 比较暴力,一个协程住后,其他协程只能等待;

    • 不管你的代码操作还是操作,锁住后,其他协程只能等待

  • sync.RWMutex (一句话总结锁和互斥,读的时候不能写,写的时候,不能读;可以同时获取多个读

    • 读锁占用的情况下,会阻止写,但不阻止读;即多个 goroutine 可同时获取读锁(调用 RLock() 方法),会阻止写;

    • 一个协程的写锁占用时,会阻止其他协程的读锁写锁
      2、读多写少的场景,推荐使用sync.RWMutex

1、sync.Mutex

Mutex 是最简单的一种类型,同时也比较暴力,当一个 goroutine 获得了 Mutex 后,其他 goroutine 就只能乖乖等到这个 goroutine 释放该 Mutex。

2、sync.RWMutex

RWMutex 相对友好些,是经典的单写多读模型。

从 RWMutex 的实现看,RWMutex 类型其实组合了 Mutex:

type RWMutex struct {
    w Mutex
    writerSem uint32
    readerSem uint32
    readerCount int32
    readerWait int32
}

对于这两种锁类型,任何一个 Lock() 或 RLock() 均需要保证对应有 Unlock() 或 RUnlock() 调用与之对应,否则可能导致等待该锁的所有 goroutine 处于饥饿状态,甚至可能导致死锁。

3、sync.Mutex 例子

锁的典型使用模式如下:

package main
import (
    "fmt"
    "sync"
)
var (
    // 逻辑中使用的某个变量
    count int
    // 与变量对应的使用互斥锁
    countGuard sync.Mutex
)
func GetCount() int {
    // 锁定
    countGuard.Lock()
    // 在函数退出时解除锁定
    defer countGuard.Unlock()
    return count
}
func SetCount(c int) {
    countGuard.Lock()
    count = c
    countGuard.Unlock()
}
func main() {
    // 可以进行并发安全的设置
    SetCount(1)
    // 可以进行并发安全的获取
    fmt.Println(GetCount())
}

代码说明如下:

  • 第 10 行是某个逻辑步骤中使用到的变量,无论是包级的变量还是结构体成员字段,都可以。

  • 第 13 行,一般情况下,建议将互斥锁的粒度设置得越小越好,降低因为共享访问时等待的时间。这里笔者习惯性地将互斥锁的变量命名为以下格式:
    变量名+Guard
    以表示这个互斥锁用于保护这个变量。

  • 第 16 行是一个获取 count 值的函数封装,通过这个函数可以并发安全的访问变量 count。

  • 第 19 行,尝试对 countGuard 互斥量进行加锁。一旦 countGuard 发生加锁,如果另外一个 goroutine 尝试继续加锁时将会发生阻塞,直到这个 countGuard 被解锁。

  • 第 22 行使用 defer 将 countGuard 的解锁进行延迟调用,解锁操作将会发生在 GetCount() 函数返回时。

  • 第 27 行在设置 count 值时,同样使用 countGuard 进行加锁、解锁操作,保证修改 count 值的过程是一个原子过程,不会发生并发访问冲突。

image
image
image

4、sync.RWMutex 例子

读多写少的环境中,可以优先使用读写互斥锁(sync.RWMutex),它比互斥锁更加高效。

sync 包中的 RWMutex 提供了读写互斥锁的封装

我们将互斥锁例子中的一部分代码修改为读写互斥锁,参见下面代码:

var (
    // 逻辑中使用的某个变量
    count int
    // 与变量对应的使用互斥锁
    countGuard sync.RWMutex
)
func GetCount() int {
    // 锁定
    countGuard.RLock()
    // 在函数退出时解除锁定
    defer countGuard.RUnlock()
    return count
}

代码说明如下:

  • 第 6 行,在声明 countGuard 时,从 sync.Mutex 互斥锁改为 sync.RWMutex 读写互斥锁。

  • 第 12 行,获取 count 的过程是一个读取 count 数据的过程,适用于读写互斥锁。在这一行,把 countGuard.Lock() 换做 countGuard.RLock(),将读写互斥锁标记为读状态。如果此时另外一个 goroutine 并发访问了 countGuard,同时也调用了 countGuard.RLock() 时,并不会发生阻塞。

  • 第 15 行,与读模式加锁对应的,使用读模式解锁。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

友情链接更多精彩内容