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

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行，与读写模式加锁对应的，使用读写模式解锁。