Golang深入浅出之-原子操作包（sync/atomic）在Go中的应用

原创

Jimaks 2024-05-09 13:53:11 博主文章分类：后端 ©著作权

文章标签 golang 开发语言后端原子操作数据 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者Jimaks的原创作品，请联系作者获取转载授权，否则将追究法律责任

Golang深入浅出之-原子操作包（sync/atomic）在Go中的应用

在Go语言的并发编程中，sync/atomic包提供了对整型值和指针进行原子操作的支持，确保这些操作在多线程环境中不会受到数据竞争的影响。本文将深入浅出地解析sync/atomic包的特性和用法，探讨常见问题、易错点及应对策略，并通过代码示例加深理解。

Golang深入浅出之-原子操作包（sync/atomic）在Go中的应用_后端

sync/atomic包简介

sync/atomic包主要包含以下几种原子操作：

原子整数操作：如AddInt32、CompareAndSwapInt32等，用于对32位或64位整型变量进行原子加减、交换、加载、存储等操作。
原子指针操作：如SwapPointer、StorePointer等，用于对指针进行原子交换、存储等操作。
原子标量函数：如LoadUint32、StoreUint32等，提供对各种宽度（32位、64位）和类型的标量值进行原子加载和存储。

import "sync/atomic"

var counter uint32

func increment() {
    atomic.AddUint32(&counter, 1)
}

func getCounter() uint32 {
    return atomic.LoadUint32(&counter)
}

常见问题与易错点

问题1：误用非原子操作

在并发环境下，直接对共享变量进行非原子操作可能导致数据竞争和竞态条件。

var counter uint32

func increment() {
    counter++ // 错误：非原子操作，可能导致数据竞争
}

解决办法：对共享变量的所有操作都应使用sync/atomic包提供的原子函数。

问题2：误解原子操作的语义

原子操作仅保证操作本身的原子性，但并不能替代互斥锁等同步原语来保证复杂的同步逻辑。例如，原子增加并不能保证计数的准确性，如果多个goroutine同时进行减法操作。

var counter uint32

func increment() {
    atomic.AddUint32(&counter, 1)
}

func decrement() {
    atomic.AddUint32(&counter, ^uint32(0)) // 错误：原子减法可能导致计数不准确
}

解决办法：对于需要保证复杂同步逻辑的场景，应结合使用原子操作与其他同步原语（如互斥锁、读写锁等）。在上述示例中，应使用AddUint32进行原子增加，用SubUint32进行原子减少。

问题3：忽略原子操作的内存排序约束

原子操作不仅保证操作本身的原子性，还隐含了特定的内存排序约束。如果不理解这些约束，可能导致意想不到的数据可见性问题。

var value uint32
var ready uint32

func producer() {
    value = 42
    atomic.StoreUint32(&ready, 1)
}

func consumer() {
    if atomic.LoadUint32(&ready) == 1 {
        fmt.Println(value) // 可能输出0，因为value的写入可能未对consumer可见
    }
}

解决办法：理解并遵循原子操作的内存排序约束。在上述示例中，可以使用AtomicStore的Release版本（如atomic.StoreUint32）确保value的写入对consumer可见。