前言
Go
语言以 高并发 著称,其并发操作是重要特性之一。虽然并发可以提高程序性能和效率,但同时也可能带来 竞态条件 和 死锁 等问题。为了避免这些问题,Go
提供了许多 并发原语,例如 Mutex
、RWMutex
、WaitGroup
、Channel
等,用于实现同步、协调和通信等操作。
本文将着重介绍 Go
的 Mutex
并发原语,它是一种锁类型,用于实现共享资源互斥访问。
说明:本文使用的代码基于的 Go 版本:1.20.1
Mutex
基本概念
Mutex
是 Go
语言中互斥锁的实现,它是一种同步机制,用于控制多个 goroutine
之间的并发访问。当多个 goroutine
尝试同时访问同一个共享资源时,可能会导致数据竞争和其他并发问题,因此需要使用互斥锁来协调它们之间的访问。
在上述图片中,我们可以将绿色部分看作是临界区。当 g1
协程通过 mutex
对临界区进行加锁后,临界区将会被锁定。此时如果 g2
想要访问临界区,就会失败并进入阻塞状态,直到锁被释放,g2
才能拿到临界区的访问权。
结构体介绍
type Mutex struct { state int32 sema uint32 }
字段:
state
state
是一个 int32
类型的变量,它存储着 Mutex
的各种状态信息(未加锁、被加锁、唤醒状态、饥饿状态),不同状态通过位运算进行计算。
sema
sema
是一个信号量,用于实现 Mutex
的等待和唤醒机制。
方法:
Lock()
Lock()
方法用于获取 Mutex
的锁,如果 Mutex
已经被其他的 goroutine
锁定,则 Lock()
方法会一直阻塞,直到该 goroutine
获取到锁为止。
UnLock()
Unlock()
方法用于释放 Mutex
的锁,将 Mutex
的状态设置为未锁定的状态。
TryLock()
Go 1.18
版本以后,sync.Mutex
新增一个 TryLock()
方法,该方法为非阻塞式的加锁操作,如果加锁成功,返回 true
,否则返回 false
。
虽然 TryLock()
的用法确实存在,但由于其使用场景相对较少,因此在使用时应该格外谨慎。TryLock()
方法注释如下所示:
// Note that while correct uses of TryLock do exist, they are rare, // and use of TryLock is often a sign of a deeper problem // in a particular use of mutexes.
代码示例
我们先来看一个有并发安全问题的例子
package main import ( "fmt" "sync" ) var cnt int func main() { var wg sync.WaitGroup for i := 0; i < 10; i++ { wg.Add(1) go func() { defer wg.Done() for j := 0; j < 10000; j++ { cnt++ } }() } wg.Wait() fmt.Println(cnt) }
在这个例子中,预期的 cnt
结果为 10 * 10000 = 100000
。但是由于多个 goroutine
并发访问了共享变量 cnt
,并且没有进行任何同步操作,可能导致读写冲突(race condition
),从而影响 cnt
的值和输出结果的正确性。这种情况下,不能确定最终输出的 cnt
值是多少,每次执行程序得到的结果可能不同。
在这种情况下,可以使用互斥锁(sync.Mutex
)来保护共享变量的访问,保证只有一个 goroutine
能够同时访问 cnt
,从而避免竞态条件的问题。修改后的代码如下:
package main import ( "fmt" "sync" ) var cnt int var mu sync.Mutex func main() { var wg sync.WaitGroup for i := 0; i < 10; i++ { wg.Add(1) go func() { defer wg.Done() for j := 0; j < 10000; j++ { mu.Lock() cnt++ mu.Unlock() } }() } wg.Wait() fmt.Println(cnt) }
在这个修改后的版本中,使用互斥锁来保护共享变量 cnt
的访问,可以避免出现竞态条件的问题。具体而言,在 cnt++
操作前,先执行 Lock()
方法,以确保当前 goroutine
获取到了互斥锁并且独占了共享变量的访问权。在 cnt++
操作完成后,再执行 Unlock()
方法来释放互斥锁,从而允许其他 goroutine
获取互斥锁并访问共享变量。这样,只有一个 goroutine
能够同时访问 cnt
,从而确保了最终输出结果的正确性。
易错场景
忘记解锁
如果使用 Lock()
方法之后,没有调用 Unlock()
解锁,会导致其他 goroutine
被永久阻塞。例如:
package main import ( "fmt" "sync" "time" ) var mu sync.Mutex var cnt int func main() { go increase(1) go increase(2) time.Sleep(time.Second) fmt.Println(cnt) } func increase(delta int) { mu.Lock() cnt += delta }
在上述代码中,通常情况下,cnt
的结果应该为 3
。然而没有解锁操作,其中一个 goroutine
被阻塞,导致没有达到预期效果,最终输出的 cnt
可能只能为 1
或 2
。
正确的做法是使用 defer
语句在函数返回前释放锁。
func increase(delta int) { mu.Lock() defer mu.Unlock() // 通过 defer 语句在函数返回前释放锁 cnt += delta }
重复加锁
重复加锁操作被称为可重入操作。不同于其他一些编程语言的锁实现(例如 Java
的 ReentrantLock
),Go
的 mutex
并不支持可重入操作,如果发生了重复加锁操作,就会导致死锁。例如:
package main import ( "fmt" "sync" "time" ) var mu sync.Mutex var cnt int func main() { go increase(1) go increase(2) time.Sleep(time.Second) fmt.Println(cnt) } func increase(delta int) { mu.Lock() mu.Lock() cnt += delta mu.Unlock() }
在这个例子中,如果在 increase
函数中重复加锁,将会导致 mu
锁被第二次锁住,而其他 goroutine
将被永久阻塞,从而导致程序死锁。正确的做法是只对需要加锁的代码段进行加锁,避免重复加锁。
基于 Mutex 实现一个简单的线程安全的缓存
import "sync" type Cache struct { data map[string]any mu sync.Mutex } func (c *Cache) Get(key string) (any, bool) { c.mu.Lock() defer c.mu.Unlock() value, ok := c.data[key] return value, ok } func (c *Cache) Set(key string, value any) { c.mu.Lock() defer c.mu.Unlock() c.data[key] = value }
上述代码实现了一个简单的线程安全的缓存。使用 Mutex
可以保证同一时刻只有一个 goroutine
进行读写操作,避免多个 goroutine
并发读写同一数据时产生数据不一致性的问题。
对于缓存场景,读操作比写操作更频繁,因此使用 RWMutex
代替 Mutex
会更好,因为 RWMutex
允许多个 goroutine
同时进行读操作,只有在写操作时才会进行互斥锁定,从而减少了锁的竞争,提高了程序的并发性能。后续文章会对 RWMutex
进行介绍。
小结
本文主要介绍了 Go
语言中互斥锁 Mutex
的概念、对应的字段和方法、基本使用和易错场景,最后基于 Mutex
实现一个简单的线程安全的缓存。
Mutex
是保证共享资源数据一致性的重要手段,但使用不当会导致性能下降或死锁等问题。因此,在使用 Mutex
时需要仔细考虑代码的设计和并发场景,发挥 Mutex
的最大作用。
到此这篇关于详解Golang中的Mutex并发原语的文章就介绍到这了,更多相关Golang Mutex并发原语内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!