如何在Golang中使用atomic原子操作_Golang atomic原子操作实践

P粉602998670 2025-11-24 00:00:00 次阅读

atomic提供高效无锁同步，适用于计数器、状态标志等场景；通过Load、Store、Add和CAS等操作实现并发安全，如原子增减和单例模式初始化，避免数据竞争且性能优于互斥锁。

在高并发编程中，多个goroutine同时访问共享变量容易引发数据竞争问题。Golang的sync/atomic包提供了原子操作，能有效避免锁的开销，提升性能。合理使用atomic可以实现轻量级的同步控制。

理解atomic原子操作的作用

原子操作保证一组操作在执行过程中不会被其他goroutine中断。atomic包支持对int32、int64、uint32、uint64、uintptr和unsafe.Pointer等类型的值进行读取、写入、增加、比较并交换等操作。这些操作由底层硬件直接支持，效率高于互斥锁。

常见场景包括计数器、状态标志、单例初始化等不需要复杂逻辑的共享变量操作。例如，一个服务需要统计请求数，使用atomic.AddInt64比加锁更高效。

常用atomic函数及用法示例

atomic包中最常用的几个函数包括Load、Store、Add、Swap和CompareAndSwap（简称CAS）。

atomic.LoadInt64(&value)：安全读取int64类型变量的当前值
atomic.StoreInt64(&value, newVal)：安全写入新值
atomic.AddInt64(&value, delta)：对变量进行原子加法，返回新值
atomic.CompareAndSwapInt64(&value, old, new)：如果当前值等于old，则设置为new，返回是否成功

示例代码：

var counter int64

go func() {
    for i := 0; i < 1000; i++ {
        atomic.AddInt64(&counter, 1)
    }
}()

// 主线程等待后读取最终结果
time.Sleep(time.Second)
fmt.Println("Counter:", atomic.LoadInt64(&counter))

使用atomic实现并发安全的单例模式

通过atomic.CompareAndSwapPointer可以实现无锁的单例初始化，避免重复创建对象。

定义一个指针和状态标识，利用CAS判断是否已初始化实例。只有第一个成功的goroutine会真正创建对象，其余直接返回已有实例。

示例：

var instance *Service
var initialized uintptr

type Service struct{}

func GetService() *Service {
    if atomic.LoadUintptr(&initialized) == 1 {
        return instance
    }
    // 尝试设置初始化状态
    if atomic.CompareAndSwapUintptr(&initialized, 0, 1) {
        instance = &Service{}
    }
    return instance
}