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

相关文章
|
1天前
|
Java 编译器 Go
探索Go语言的性能优化技巧
在本文中,我们将深入探讨Go语言的底层机制,以及如何通过代码层面的优化来提升程序性能。我们将讨论内存管理、并发控制以及编译器优化等关键领域,为你提供一系列实用的技巧和最佳实践。
|
1天前
|
Cloud Native Go API
Go语言在微服务架构中的创新应用与实践
本文深入探讨了Go语言在构建高效、可扩展的微服务架构中的应用。Go语言以其轻量级协程(goroutine)和强大的并发处理能力,成为微服务开发的首选语言之一。通过实际案例分析,本文展示了如何利用Go语言的特性优化微服务的设计与实现,提高系统的响应速度和稳定性。文章还讨论了Go语言在微服务生态中的角色,以及面临的挑战和未来发展趋势。
|
1天前
|
安全 Go 调度
探索Go语言的并发模式:协程与通道的协同作用
Go语言以其并发能力闻名于世,而协程(goroutine)和通道(channel)是实现并发的两大利器。本文将深入了解Go语言中协程的轻量级特性,探讨如何利用通道进行协程间的安全通信,并通过实际案例演示如何将这两者结合起来,构建高效且可靠的并发系统。
|
1天前
|
安全 Go 开发者
破译Go语言中的并发模式:从入门到精通
在这篇技术性文章中,我们将跳过常规的摘要模式,直接带你进入Go语言的并发世界。你将不会看到枯燥的介绍,而是一段代码的旅程,从Go的并发基础构建块(goroutine和channel)开始,到高级模式的实践应用,我们共同探索如何高效地使用Go来处理并发任务。准备好,让Go带你飞。
|
2天前
|
运维 Go 开发者
Go语言在微服务架构中的应用与优势
本文深入探讨了Go语言在构建微服务架构中的独特优势和实际应用。通过分析Go语言的核心特性,如简洁的语法、高效的并发处理能力以及强大的标准库支持,我们揭示了为何Go成为开发高性能微服务的首选语言。文章还详细介绍了Go语言在微服务架构中的几个关键应用场景,包括服务间通信、容器化部署和自动化运维等,旨在为读者提供实用的技术指导和启发。
|
2天前
|
安全 Go 调度
探索Go语言的并发之美:goroutine与channel
在这个快节奏的技术时代,Go语言以其简洁的语法和强大的并发能力脱颖而出。本文将带你深入Go语言的并发机制,探索goroutine的轻量级特性和channel的同步通信能力,让你在高并发场景下也能游刃有余。
|
3天前
|
Go 开发者
Go语言中的并发编程:从基础到实践
在当今的软件开发中,并发编程已经成为了一项不可或缺的技能。Go语言以其简洁的语法和强大的并发支持,成为了开发者们的首选。本文将带你深入了解Go语言中的并发编程,从基础概念到实际应用,帮助你掌握这一重要的编程技能。
|
4天前
|
Go
使用go语言将A助手加入项目中
使用go语言将A助手加入项目中
12 2
|
3天前
|
安全 Go 调度
探索Go语言的并发模型:Goroutine与Channel的魔力
本文深入探讨了Go语言的并发模型,不仅解释了Goroutine的概念和特性,还详细讲解了Channel的用法和它们在并发编程中的重要性。通过实际代码示例,揭示了Go语言如何通过轻量级线程和通信机制来实现高效的并发处理。
|
3天前
|
存储 安全 Go
Go语言切片:从入门到精通的深度探索###
本文深入浅出地剖析了Go语言中切片(Slice)这一核心概念,从其定义、内部结构、基本操作到高级特性与最佳实践,为读者提供了一个全面而深入的理解。通过对比数组,揭示切片的灵活性与高效性,并探讨其在并发编程中的应用优势。本文旨在帮助开发者更好地掌握切片,提升Go语言编程技能。 ###