【加锁】通过go的 mutex和atomic完成全局变量加锁

hengbo.liu

于 2024-01-24 13:59:23 发布

阅读量374

点赞数 10

分类专栏： golang 文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/qq_41149621/article/details/135820008

版权

golang 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

未加锁示例

CPU执行大致流程(不代表CPU实际执行流程): 加载total变量 -> 增加或减少total变量 -> 写入total变量, 当这个流程被多个协程同时操作且没有进行加锁,时最终的数据就会出现问题,大致示意图如下:

var wg sync.WaitGroup
var total = 0
func add() {
	defer wg.Done()
	for i := 0; i < 10000; i++ {
		total++
	}
}
func sub() {
	defer wg.Done()
	for i := 0; i < 10000; i++ {
		total--
	}
}
func main() {
	wg.Add(2)
	go add()
	go sub()
	wg.Wait()
	fmt.Println(total)
}
// 输出:  -9902  (其实是随机变化的,无法预测的)

mux

适合相对复杂场景下的加锁,如结构体等

var wg sync.WaitGroup
var lock sync.Mutex
var total = 0
func add() {
	defer wg.Done()
	for i := 0; i < 10000; i++ {
		lock.Lock()
		total++
		lock.Unlock()
	}
}
func sub() {
	defer wg.Done()
	for i := 0; i < 10000; i++ {
		lock.Lock()
		total--
		lock.Unlock()
	}
}
func main() {
	wg.Add(2)
	go add()
	go sub()
	wg.Wait()
	fmt.Println(total)
}

atomic

性能更高, 适合相对简单的加锁, 如单个变量进行加锁的场景,
本示例中相当于把add()方法的操作,当做一个原子操作来执行

var wg sync.WaitGroup
var total int32
func add() {
	defer wg.Done()
	for i := 0; i < 10000; i++ {
		atomic.AddInt32(&total,1)
	}
}
func sub() {
	defer wg.Done()
	for i := 0; i < 10000; i++ {
		atomic.AddInt32(&total,-1)
	}
}
func main() {
	wg.Add(2)
	go add()
	go sub()
	wg.Wait()
	fmt.Println(total)
}