一、互斥锁
先看一个并发情况,同时操作一个全局变量,如果没有锁会怎么样
假设有1000个goroutines并发进行银行余额的扣除,每次都扣除10元,起始的总余额是10000,理论上并发执行完应该是0对不对,但实际却不是
import ("fmt""sync"
)var balance uint = 10000func withdraw(amount uint) {balance = balance - amountfmt.Println("剩余存款:", balance)defer wg.Done()
}func getBalance() uint {return balance
}func main() {for i := 1; i <= 1000; i++ {wg.Add(1)go withdraw(10)}wg.Wait()fmt.Println(getBalance())
}
得到的结果如下:
这时候就需要引入一个锁来保护balance全局变量,在并发goroutines执行的情况下,防止其他goroutines获取到balance变量进行操作(其他goroutines会处于等待锁的状态)
import ("fmt""sync"
)var balance uint = 10000
var mu sync.Mutex
var wg sync.WaitGroupfunc withdraw(amount uint) {//如果其它的goroutine已经获得了这个锁的话,这个操作会被阻塞直到其它goroutine调用了Unlock使该锁变回可用状态。mutex会保护共享变量。mu.Lock() ******************* 增加锁 **********************balance = balance - amountfmt.Println("剩余存款:", balance)mu.Unlock() ******************* 释放锁 **********************defer wg.Done()
}func getBalance() uint {return balance
}func main() {for i := 1; i <= 1000; i++ {wg.Add(1)go withdraw(10)}wg.Wait()fmt.Println(getBalance())
}
得到的结果如下:
二、读写锁
RWMutex,即读写互斥锁,适用于读操作远远多于写操作的场景,当多个goroutine需要频繁地读取某个共享资源,而写入操作相对较少时,使用RWMutex可以提高并发性能。
func getBalance() uint {rwMu.RLock() ********* 加锁读取 ********defer rwMu.RUnlock() ********* 返回释放锁 ********return balance
}
原理:RWMutex允许多个goroutine同时获取读锁(RLock),进行并发读操作,而写锁(Lock)则是互斥的,同一时间只允许一个goroutine进行写操作。当一个goroutine获取读锁时,其他goroutine也可以继续获取读锁来读取数据,而不需要等待。只有当有goroutine尝试获取写锁时,读锁才会被阻塞,这样可以有效地提高并发读取的效率。
sync.RWMutex锁的设计就是为了优化读操作频繁的场景。它能够区分读锁(RLock)和写锁(Lock)
读锁(RLock):
当一个goroutine需要读取共享资源时,它会尝试获取读锁。如果当前没有其他goroutine持有写锁,那么它可以成功获取读锁,并且可以并发地读取数据。如果有其他goroutine已经持有读锁,新的读锁请求也会被允许,从而允许多个goroutine同时读取。
写锁(Lock):
当一个goroutine需要写入共享资源时,它会尝试获取写锁。在写锁被获取之前,所有的读锁和写锁请求都会被阻塞。这意味着,一旦有goroutine开始写操作,其他想要读取或写入的goroutine都必须等待,直到写操作完成并且写锁被释放。
锁的释放:
当一个goroutine完成读取或写入操作后,它会释放相应的锁。如果是读操作完成,它会释放读锁,这样其他正在等待的goroutine就可以获取读锁并开始读取。如果是写操作完成,它会释放写锁,这样其他正在等待的读锁或写锁请求就可以继续进行。
RWMutex的这种设计使得在读操作远多于写操作的情况下,可以提高程序的并发性能。因为多个goroutine可以同时读取,而不需要等待写操作完成,这样就减少了锁竞争,提高了整体的吞吐量。然而,需要注意的是,RWMutex并不是在所有情况下都能提高性能。如果写操作非常频繁,或者读写操作几乎同等频繁,使用RWMutex可能会导致性能下降,因为写操作会阻塞所有读锁和写锁的请求。在这种情况下,使用Mutex可能会更加合适。