Go语言包中的sync包提供了两种锁类型:sync.Mutex和sync.RWMutex。
Mutex是最简单的一种类型,同时也比较暴力,当一个goroutine获得Mutex后,其他goroutine就只能乖乖等待这个goroutine释放该Mutex.
RWMutex 相对友好些,是经典的单写多读模型。在读锁占用的情况下,会阻止写,但不阻止读,也就是多个 goroutine 可同时获取读锁(调用 RLock() 方法;而写锁(调用 Lock() 方法)会阻止任何其他 goroutine(无论读和写)进来,整个锁相当于由该 goroutine 独占。从 RWMutex 的实现看,RWMutex 类型其实组合了 Mutex:
type RWMutex struct {
w Mutex
writerSem uint32
readerSem uint32
readerCount int32
readerWait int32
}
对于这两个锁类型,任何一个Lock()和Rlock()均需要保证对应有Unlock()或者Runlock()调用与之对应,否则可能导致等待该锁的所有goroutine处于饥饿状态,甚至可能导致死锁。锁的典型使用模式如下:
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())
}
代码输出:
1
代码说明如下:
第10行是某个步骤中使用到的变量,无论是包级的变量还是结构体字段,都可以。
第13行,一般情况下,建议将互斥锁的粒度设置越小越好,降低因为共享访问时等待的时间。互斥锁的变量命名为以下格式:
变量名+Guard
以上互斥锁用于保护这个变量。
第16行是一个获取count值的函数封装,通过这个函数可以并发安全的访问变量count。
第19行,尝试对countGuard互斥量进行加锁。一旦coutGuard发生加锁,如果另外一个goroutine尝试继续加锁时将会发生阻塞,直到这个countGuard被解锁。
第22行使用defer将countGuard的解锁进行延迟调用,解锁操作将会发生在GetCount()函数返回时。
第27行在设置count值时,同样使用countGuard进行加锁、解锁操作,保证修改count值的过程是一个原子操作过程,不会发生并发冲突;
在读多写少的环境中,可以优先使用读写互斥锁(sync.RWMutex),它比互斥锁更加高效。sync包中RWMutex提供读写互斥锁的封装。
我们将互斥锁例子中的一部分代码修改为读写互斥锁。
var (
//逻辑使用的某个变量
count int
//与变量对应的使用互斥锁
countGuard sync.Mutex
)
func GetCount() int {
//锁定
countGuard.Lock()
//在函数退出时接触锁定
defer countGuard.Unlock()
return count
}
代码说明如下:
第6行,在声明countGuard时,从sync.Mutex互斥锁改为sync.RWMutex读写互斥锁。
第12行,获取count的过程是一个读取count数据的过程,适用于读写互斥锁。在这一行,把countGuard.Lock()换做countGuard.RLock(),将读写互斥锁标记为读状态。如果此时另外一个gorountine并发访问countGuard,同时也调用了countGuard.RLock() 时,并不会发生阻塞。
第15行,与读写模式加锁对应的,使用读写模式解锁。