在Golang的并发编程中,锁是保证共享资源一致性的常用手段,但如果锁的使用方式不当,就会引发严重的锁竞争,导致goroutine频繁阻塞等待,反而降低程序的整体并发性能。理解锁竞争的根源,掌握对应的优化方法,是写出高性能Golang并发程序的关键。

锁竞争产生的原因
锁竞争本质是多个goroutine同时尝试获取同一把锁,而锁同一时间只能被一个goroutine持有,其他goroutine只能进入等待队列。常见的引发锁竞争的原因包括:锁的覆盖范围过大,把不需要保护的代码也放在锁内部;频繁获取释放同一把锁;使用互斥锁保护读多写少的场景,读操作也会被阻塞。
减少锁竞争的具体方法
1. 缩小锁的粒度
尽量只把操作共享资源的必要代码放在锁内部,减少锁的持有时间,降低其他goroutine等待的概率。比如原本把整个函数逻辑都加锁,可以只给修改共享变量的几行代码加锁。
优化前的代码:
package main
import (
"sync"
"time"
)
var (
mu sync.Mutex
count int
)
func addWithBigLock() {
mu.Lock()
// 模拟非共享资源的耗时操作
time.Sleep(time.Millisecond)
count++
mu.Unlock()
}优化后的代码,把耗时操作移到锁外部:
func addWithSmallLock() {
// 耗时操作不需要保护,放在锁外
time.Sleep(time.Millisecond)
mu.Lock()
count++
mu.Unlock()
}2. 读多写少场景使用读写锁
如果共享资源的读操作远多于写操作,使用sync.RWMutex代替sync.Mutex,读操作可以并发执行,只有写操作需要互斥,能大幅提升并发性能。
package main
import (
"sync"
)
var (
rwMu sync.RWMutex
data map[string]string
)
// 读操作使用读锁
func getData(key string) string {
rwMu.RLock()
defer rwMu.RUnlock()
return data[key]
}
// 写操作使用写锁
func setData(key, value string) {
rwMu.Lock()
defer rwMu.Unlock()
data[key] = value
}3. 使用无锁数据结构或原子操作
对于简单的数值修改、状态标记等场景,优先使用sync/atomic包的原子操作,避免加锁。原子操作是底层硬件支持的不可中断操作,不存在锁竞争问题。
package main
import (
"sync/atomic"
)
var (
atomicCount int64
)
func addAtomic() {
// 原子自增,无需加锁
atomic.AddInt64(&atomicCount, 1)
}
func getAtomicCount() int64 {
return atomic.LoadInt64(&atomicCount)
}4. 使用sync包的其他工具类减少锁依赖
对于单次初始化、等待一组goroutine完成等场景,可以使用sync.Once、sync.WaitGroup等工具,避免自己实现加锁逻辑。比如sync.Once可以保证某个操作只执行一次,不需要额外加锁控制。
package main
import (
"sync"
)
var (
once sync.Once
instance *Singleton
)
type Singleton struct{}
func GetInstance() *Singleton {
// 保证instance只初始化一次,无需额外加锁
once.Do(func() {
instance = &Singleton{}
})
return instance
}5. 拆分锁,降低锁的冲突概率
如果多个独立的共享资源使用同一把锁,可以把锁拆分成多把,每个资源对应自己的锁,减少不同资源操作之间的锁竞争。比如用map存储多个资源的锁,每个key对应一把独立的锁。
package main
import (
"sync"
)
type ShardMap struct {
shards []*Shard
size int
}
type Shard struct {
mu sync.Mutex
data map[string]int
}
func NewShardMap(size int) *ShardMap {
sm := &ShardMap{
shards: make([]*Shard, size),
size: size,
}
for i := 0; i < size; i++ {
sm.shards[i] = &Shard{
data: make(map[string]int),
}
}
return sm
}
func (sm *ShardMap) Add(key string, value int) {
// 根据key哈希到对应的分片,只获取该分片的锁
shard := sm.shards[hash(key)%sm.size]
shard.mu.Lock()
shard.data[key] = value
shard.mu.Unlock()
}
func hash(key string) int {
h := 0
for _, c := range key {
h = h*31 + int(c)
}
return h
}优化效果的验证方法
可以使用Golang内置的go test -bench命令进行基准测试,对比优化前后的程序性能,也可以使用pprof工具分析锁竞争的情况,定位需要优化的锁热点。通过持续调整锁的使用方式,逐步降低锁竞争,提升程序的并发处理能力。