8.13. 互斥锁

8.13.1. 1.互斥锁

互斥锁是一种常用的控制共享资源访问的方法。在Go程序中的使用非常简单,参见下面的代码:

package main

import (
    "fmt"
    "sync"
)

var (
    // 逻辑中使用的某个变量
    count int
    // 与变量对应的使用互斥锁
    countGuard sync.Mutex
)


// 是一个获取count值的函数封装,通过这个函数可以并发安全的访问变量count。
func GetCount() int {
    // 锁定
    countGuard.Lock()

    // 使用defer将countGuard的解锁进行延迟调用,解锁操作将会发生在GetCount()函数返回时。
    defer countGuard.Unlock()
    return count
}

// 设置count值时,同样使用countGuard进行加锁、解锁操作,保证修改count值的过程是一个原子过程,不会发生并发访问冲突。
func SetCount(c int)  {
    countGuard.Lock()
    count = c
    countGuard.Unlock()
}

func main() {
    // 可以进行并发安全的设置
    SetCount(1)

    // 可以进行并发安全的获取
    fmt.Println(GetCount())
}

8.13.2. 2.读写互斥锁(sync.RWMutex)-在读比写多的环境下比互斥锁更高效

在读多写少的环境中,可以优先使用读写互斥锁,sync包中的RWMutex提供了读写互斥锁的封装。 读写互斥锁,参见如下代码

package main

import (
    "fmt"
    "sync"
)

var (
    // 逻辑中使用的某个变量
    count int
    //// 与变量对应的使用互斥锁
    //countGuard sync.Mutex
    // 与变量对应的使用互斥锁
    countGuard sync.RWMutex
)

//在声明 countGuard 时,从 sync.Mutex 互斥锁改为 sync.RWMutex 读写互斥锁。
func GetCount() int {
    //获取 count 的过程是一个读取 count 数据的过程,适用于读写互斥锁。在这一行,把 countGuard.Lock() 换做 countGuard.RLock(),将读写互斥锁标记为读状态。如果此时另外一个 goroutine 并发访问了 countGuard,同时也调用了 countGuard.RLock() 时,并不会发生阻塞。
    // 锁定
    countGuard.RLock()

    //与读模式加锁对应的,使用读模式解锁。
    // 在函数退出时解除锁定
    defer countGuard.RUnlock()
    return count
}

func SetCount(c int)  {
    countGuard.Lock()
    count = c
    countGuard.Unlock()
}

func main() {
    // 可以进行并发安全的设置
    SetCount(1)
    // 可以进行并发安全的获取
    fmt.Println(GetCount())
}