Go语言中atomic包提供轻量级原子操作,适用于多协程下对基本类型的安全读写。1. 支持int32、int64、uint32、uint64、uintptr、unsafe.Pointer及布尔值的原子操作,常用函数有LoadXXX、StoreXXX、AddXXX和CompareAndSwapXXX。2. 示例中使用atomic.AddInt64和atomic.LoadInt64实现并发安全计数器,通过sync.WaitGroup协调1000个goroutine递增共享变量counter,最终输出正确的计数值1000,避免了传统锁的开销。

在Go语言中,atomic包提供了底层的原子操作支持,适用于多协程环境下对基本数据类型的读写保护。相比互斥锁(sync.Mutex),原子操作更轻量、高效,特别适合计数器、状态标志等简单共享变量的场景。
1. 原子操作的基本类型和函数
atomic包主要支持对以下类型的原子操作:
int32、int64 uint32、uint64、uintptr 指针(unsafe.Pointer) 布尔值(通过特定方式模拟)
常用函数包括:
atomic.LoadXXX:原子读取 atomic.StoreXXX:原子写入 atomic.AddXXX:原子增减 atomic.CompareAndSwapXXX:比较并交换(CAS)
2. 实际使用示例
下面是一个并发安全计数器的例子,展示如何用atomic.AddInt64和atomic.LoadInt64实现线程安全的计数:
立即学习“go语言免费学习笔记(深入)”;
package mainimport ( "fmt" "sync" "sync/atomic")func main() { var counter int64 var wg sync.WaitGroup for i := 0; i < 10; i++ { wg.Add(1) go func() { defer wg.Done() for j := 0; j < 1000; j++ { atomic.AddInt64(&counter, 1) } }() } wg.Wait() fmt.Println("最终计数:", atomic.LoadInt64(&counter))}
在这个例子中,多个goroutine同时对counter进行递增,使用atomic.AddInt64确保每次加1是原子的,避免了竞态条件。
3. 使用CompareAndSwap实现原子更新
CAS(Compare and Swap)是一种常见的无锁编程技术。可用于实现更复杂的原子逻辑,比如只设置一次的状态标志:
var status int32 // 0:未初始化, 1:已初始化func initialize() { if atomic.CompareAndSwapInt32(&status, 0, 1) { fmt.Println("执行初始化逻辑") // 实际初始化代码 } else { fmt.Println("已被其他协程初始化") }}
多个协程调用initialize时,只有第一个能成功将status从0改为1,其余会因当前值已不是0而返回false,从而保证初始化只执行一次。
4. 注意事项和限制
原子操作仅适用于基本类型,不能用于结构体或数组 必须传地址给atomic函数(如&counter) 不支持浮点型的原子操作,需用其他方式(如mutex)保护 atomic.Load/Store应成对使用,确保内存顺序一致性 复杂逻辑仍推荐使用mutex,避免过度依赖原子操作导致代码难懂
基本上就这些。atomic适合简单、高频的共享变量操作,理解其使用方式有助于写出更高效的并发程序。关键是根据场景选择合适的方法,不盲目追求性能而牺牲可维护性。
以上就是Golang如何使用atomic实现原子操作的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1417544.html
微信扫一扫
支付宝扫一扫