C++原子操作std::atomic用法详解

原子操作是不可中断的操作，std::atomic 提供对共享变量的原子访问，支持 load、store、fetch_add 等操作，默认使用 seq_cst 内存序，可用于实现无锁计数器或自旋锁，提升多线程程序性能与安全性。

在C++中，std::atomic 是实现无锁并发编程的核心工具之一。它保证对共享变量的读写操作是原子的，避免多个线程同时访问时产生数据竞争，从而提高程序性能和线程安全性。

什么是原子操作？

原子操作是指一个操作在执行过程中不会被其他线程中断，要么完全执行，要么不执行。在多线程环境中，对共享变量的普通读写可能引发竞态条件（race condition），而使用 std::atomic 可以确保操作的原子性，无需加锁也能安全访问。

基本用法：定义和初始化原子变量

std::atomic 可用于整型、指针等类型。常用的基本形式如下：

• std::atomic counter{0}; —— 原子整数，初始值为0
• std::atomic ready{false}; —— 原子布尔值
• std::atomic ptr{nullptr}; —— 原子指针

支持的操作包括 load（读）、store（写）、fetch_add、fetch_sub、exchange、compare_exchange_weak/strong 等。

常见原子操作函数说明

以下是常用的成员函数及其用途：

• load()：原子地读取当前值
• store(val)：原子地写入新值
• fetch_add(n)：原子增加n，返回旧值
• fetch_sub(n)：原子减少n，返回旧值
• exchange(val)：设置新值，并返回旧值
• compare_exchange_weak(expected, desired)：如果当前值等于 expected，则设为 desired，否则更新 expected
• compare_exchange_strong(...)：与 weak 类似，但不会因虚假失败而返回 false

这些操作默认使用最强内存序 std::memory_order_seq_cst，确保顺序一致性。

实际示例：线程安全计数器

下面是一个使用 std::atomic 实现无锁计数器的例子：

#include 
#include 
#include 
#include 

std::atomic count{0};

void increment() {
    for (int i = 0; i < 1000; ++i) {
        count.fetch_add(1);
    }
}

int main() {
    std::vector threads;
    for (int i = 0; i < 10; ++i) {
        threads.emplace_back(increment);
    }

    for (auto& t : threads) {
        t.join();
    }

    std::cout << "Final count: " << count.load() << '\n';
    return 0;
}

这个例子中，多个线程并发调用 fetch_add，由于原子性保障，最终结果一定是10000，不会出现数据竞争。

compare-and-swap 实现无锁逻辑

利用 compare_exchange_weak 可以实现更复杂的无锁结构，比如无锁栈或标志位控制：

std::atomic lock_flag{false};

bool try_lock() {
    bool expected = false;
    return lock_flag.compare_exchange_strong(expected, true);
}

void unlock() {
    lock_flag.store(false);
}

这段代码模拟了一个简单的无锁自旋锁。只有当 lock_flag 为 false 时，才能将其设为 true 并获得“锁”。

内存序（Memory Order）的选择

std::atomic 的操作可以指定内存序来优化性能。常见的选项有：

• std::memory_order_relaxed：只保证原子性，不提供同步或顺序约束
• std::memory_order_acquire：用于 load，保证之后的读写不会被重排到该操作之前
• std::memory_order_release：用于 store，保证之前的读写不会被重排到该操作之后
• std::memory_order_acq_rel：同时具有 acquire 和 release 语义
• std::memory_order_seq_cst：默认，最严格的顺序一致性

例如，计数器场景可使用 relaxed 模式提升性能：

count.fetch_add(1, std::memory_order_relaxed);

基本上就这些。合理使用 std::atomic 能写出高效、安全的无锁代码，但要注意不是所有类型都支持原子操作（如不能直接对自定义结构体使用 std::atomic，除非特化或使用 atomic_ref 等）。掌握原子变量和内存模型是深入并发编程的关键一步。