go语言中的Atomic操作与sema锁

简介: 在并发编程中,确保数据一致性和程序正确性是关键挑战。Go语言通过协程和通道提供强大支持,但在需精细控制资源访问时,Atomic操作和sema锁变得至关重要。Atomic操作确保多协程环境下对共享资源的访问是不可分割的,如`sync/atomic`包中的`AddInt32`等函数,底层利用硬件锁机制实现。sema锁(信号量锁)控制并发协程数量,其核心是一个uint32值,当大于零时通过CAS操作实现锁的获取与释放;当为零时,sema锁管理协程休眠队列。这两种机制共同保障了Go语言并发环境下的数据完整性和程序稳定性。

前言

在并发编程中,确保数据的一致性和程序的正确性始终是开发者面临的重大挑战。Go语言以其独特的并发模型——协程(Goroutine)和通道(Channel),为开发者提供了强大的并发编程工具。

然而,仅仅依靠协程和通道还不足以解决所有并发问题,特别是在需要精细控制资源访问和数据一致性的场景下。这时,Atomic操作和sema锁(信号量锁)便成为了Go并发编程中不可或缺的一部分。

例如,在使用Go语言开发业务的过程中,我们常常会使用到sync包下面的MutexRWMutex来对我们的业务逻辑加锁。实际上,互斥锁和读写锁底层都依赖于我们下面要讲的两项技术,即Atomic操作和sema锁。

Atomic操作

Atomic操作,顾名思义,是指在执行过程中不可分割的操作。在多线程或多协程环境下,对共享资源的访问必须确保原子性,以避免数据竞争和不一致性问题。Go语言通过标准库中的sync/atomic包提供了一系列原子操作函数,这些函数能够确保在并发环境中对特定变量的读取和写入是安全的。

最典型的案例就是开多个协程并发多一个全局变量自增,结果是永远是达不到预期效果的,原因是这些对全局资源的操作都不是原子性操作,往往要经过内存读取、操作、写回内存这三个步骤。如果我们要让效果达到预期,就需要使用atomic包下的方法。

atomic包提供了许多好用的Api,这里我们介绍常见的几个:

  • 变量原子性增减:可以使用如atomic.AddInt32(指针,要添加的值)这个Api进行操作。
  • CAS操作:CAS操作即compareAndSwap,可以传入old value和new value,如果值等于old value,就设置为new value。
  • LoadInt64:加载内存的时候加锁,防止变量的二进制位被其它协程修改。

Atomic的底层原理:硬件锁

Go语言中,Atomic操作的本质是一种硬件层面加锁的机制,可以保证操作一个变量的时候,其它协程和线程是无法访问的。不过,这种原子操作只能用于简单变量的简单操作。

我们可以查看一下atomic.AddInt32的底层实现,发现Go代码只有声明,说明底层应该是使用汇编来实现的。

使用Golandctrl+shift+f,找到对应的汇编,比如AMD64的实现,我们发现底层在操作变量前,使用了LOCK汇编指令,而LOCK是CPU级别的锁,这说明atomic使用的是硬件锁。

sema锁

sema锁,也叫做信号量锁(semaphore)。在Go语言中,它的核心是一个uint32的值,含义是同时可并发的协程数量。

Go语言中,每个sema锁是一个uint32值,每个值都对应一个SemaRoot结构体。查看底层代码我们可以知道,每一个锁背后都对应一个AVL树用于协程排队。

runtime/sema.go里面,我们可以找到SemaRoot结构体:

  • sudog:平衡二叉树(AVL Tree)的根节点,用于协程排队
  • nwait:排队的协程数量

go

代码解读

复制代码

type semaRoot struct {
	lock  mutex
	treap *sudog        // root of balanced tree of unique waiters.
	nwait atomic.Uint32 // Number of waiters. Read w/o the lock.
}

sudog结构体在runtime/runtime2.go里面:每个sudog中含有一个协程

go

代码解读

复制代码

type sudog struct {  
    g *g  
  
    next *sudog  
    prev *sudog  
    elem unsafe.Pointer 
    
    acquiretime int64  
    releasetime int64  
    ticket      uint32  
  
    parent   *sudog // semaRoot binary tree
    waitlink *sudog // g.waiting list or semaRoot  
    waittail *sudog // semaRoot  
    c        *hchan // channel  
}

可见,sema在内存中的大致结构如下:

当uint32 > 0时的sema操作

当uint32 > 0时的sema操作:表示可以并发获取协程的个数。

  • 获取锁:使用CAS将uint32值减少1成功,则获取锁成功。
  • 释放锁:使用atomic将uint32的值添加1,表示释放锁成功。

我们可以查看一下Go的源码:

sema.go/semacquire(addr):传入sema的uint32的值

  • 调用cansemacquire(addr)
  • 如果uint32的值=0,return false。
  • 使用cas减1,如果成功 return true

semarelease()方法:

  • 使用atomic.Xad(addr,1)给信号量新增1,表示释放成功。

当uint32 = 0时候的sema操作

当uint32 = 0时候的sema操作:sema锁退化为一个协程休眠队列。

  • 获取锁:协程休眠,进入到堆树去等待。
  • 释放锁:从堆树中取出一个协程,唤醒。

semacquire

  • 使用root.queue(),将协程放到AVL树里面排队
  • 使用goparkunlock(),休眠当前协程。

semarelease

  • 如果等待的协程数量是0(atomic.Load(&root.nwait) == 0),返回。
  • 如果不是,root.dequeue()释放一个协程,唤醒协程。

小结

通过本文的探讨,我们深入了解了Go语言中Atomic操作和sema锁在并发编程中的重要作用和实际应用。

Atomic操作通过提供不可分割的读写操作,确保了并发环境下数据的一致性和安全性;而sema锁则通过控制对共享资源的访问数量,实现了更为灵活的并发控制策略。在实际开发中,我们往往会使用更上层的锁,即Go语言封装的sync.Mutexsync.RWMutex,另外Atomic操作也是很常用的。


转载来源:https://juejin.cn/post/7400969682390614027

相关文章
|
17天前
|
存储 Go 索引
go语言中数组和切片
go语言中数组和切片
27 7
|
17天前
|
Go 开发工具
百炼-千问模型通过openai接口构建assistant 等 go语言
由于阿里百炼平台通义千问大模型没有完善的go语言兼容openapi示例,并且官方答复assistant是不兼容openapi sdk的。 实际使用中发现是能够支持的,所以自己写了一个demo test示例,给大家做一个参考。
|
17天前
|
程序员 Go
go语言中结构体(Struct)
go语言中结构体(Struct)
92 71
|
16天前
|
存储 Go 索引
go语言中的数组(Array)
go语言中的数组(Array)
100 67
|
17天前
|
存储 Go
go语言中映射
go语言中映射
32 11
|
19天前
|
Go
go语言for遍历映射(map)
go语言for遍历映射(map)
29 12
|
18天前
|
Go 索引
go语言使用索引遍历
go语言使用索引遍历
26 9
|
18天前
|
Go 索引
go语言使用range关键字
go语言使用range关键字
24 7
|
18天前
|
Go 索引
go语言修改元素
go语言修改元素
25 6
|
9天前
|
Go 数据安全/隐私保护 UED
优化Go语言中的网络连接:设置代理超时参数
优化Go语言中的网络连接:设置代理超时参数