17370845950

c++的std::atomic如何实现原子操作 线程安全计数器示例【无锁编程】
std::atomic通过封装CPU原子指令、编译器内存序约束和缓存一致性协议实现无锁原子操作;其核心是硬件支持而非互斥锁,memory_order需按场景选择,且原子类型不保证复合操作的原子性。

std::atomic 本身不“实现”原子操作,而是封装并暴露底层硬件提供的原子指令(如 x86 的 LOCK XADD、ARM 的 LDXR/STXR),由编译器生成对应汇编,并配合内存序约束,确保读-改-写等操作不可分割、对其他线程可见。

原子变量的核心机制

它不是靠互斥锁,而是依赖三点:

  • CPU 提供的原子指令(如 incLOCK 前缀、cmpxchg
  • 编译器禁止对该变量进行重排序(通过 memory_order 控制)
  • 缓存一致性协议(如 MESI)保证多核间修改及时同步

线程安全计数器:无锁递增示例

以下是一个典型的无锁计数器,多个线程并发调用 increment() 不会丢失更新:

#include 
#include 
#include 
#include 

struct Counter {
    std::atomic value{0};

    void increment() {
        value.fetch_add(1, std::memory_order_relaxed);
    }

    long get() const {
        return value.load(std::memory_order_acquire);
    }
};

int main() {
    Counter c;
    std::vector threads;
    const int N = 100000;

    for (int i = 0; i < 4; ++i) {
        threads.emplace_back([&c, N]{
            for (int j = 0; j < N; ++j) {
                c.increment();
            }
        });
    }

    for (auto& t : threads) t.join();
    std::cout << "Final count: " << c.get() << "\n"; // 输出 400000
}

memory_order 的选择要点

不同场景需匹配合适内存序,影响性能与正确性:

  • memory_order_relaxed:仅保证该操作自身原子,无同步或顺序约束。适合计数器累加、引用计数等纯数值场景
  • memory_order_acquire / memory_order_release:用于构建“同步点”,如生产者-消费者中配对使用
  • memory_order_seq_cst(默认):最强一致性,所有线程看到相同操作顺序,但开销最大

注意:原子 ≠ 任意复合操作都安全

比如下面代码不是原子的:

if (counter.load() < 100) {
    counter.fetch_add(1); // ❌ 可能多个线程同时通过判断,再重复加一
}

若需条件更新,应改用 compare_exchange_weak 实现 CAS 循环:

long expected = counter.load();
while (expected < 100 && 
       !counter.compare_exchange_weak(expected, expected + 1)) {
    // expected 被自动更新为当前值,继续尝试
}