c++中的std::atomic_flag有什么用_c++最简单的原子类型与自旋锁【并发】

std::atomic_flag是最轻量的无锁原子类型,专为自旋锁设计,仅支持test_and_set()和clear()两个原子操作,不支持load/store、拷贝或赋值,必须显式初始化,保证lock-free实现。

std::atomic_flag 是 C++ 中最轻量、最底层的原子类型,专为实现自旋锁(spinlock)这类无锁同步原语而设计。它只支持两个操作:test_and_set()(置位并返回旧值)和clear()(清零),且保证这两个操作是原子的、不可中断的。它不提供 load/store 语义,也不支持拷贝或赋值——天生就是“开关型”工具。

为什么需要 atomic_flag?

在多线程环境下,有时需要极简、低开销的互斥控制,比如保护一小段关键代码、实现自定义锁、或构建更复杂的原子结构。mutex 太重(涉及系统调用、上下文切换),而 std::atomic 虽简单,但无法保证“测试+设置”是单个原子指令(可能被编译器或 CPU 拆分)。atomic_flag 则由标准强制要求必须以**无锁方式实现**(lock-free),通常直接映射到 CPU 的 test-and-setexchangecompare-and-swap 指令,真正做到了最小延迟。

怎么用它写一个自旋锁?

一个典型的、可复用的自旋锁实现如下:

(注意:实际项目中建议优先使用 std::mutex;自旋锁仅适用于临界区极短、且线程数 ≤ CPU 核心数的场景)

  • ATOMIC_FLAG_INIT 静态初始化(C++17 起可直接用 {} 初始化)
  • 构造时调用 test_and_set(std::memory_order_acquire) 尝试上锁;若返回 false,说明之前是未设置状态,成功获得锁
  • 循环重试直到成功(即“自旋”),每次用 std::this_thread::yield() 提示调度器让出时间片,避免空转霸占 CPU
  • 析构或解锁时调用 clear(std::memory_order_release),确保释放前的写操作对其他线程可见

atomic_flag 和 atomic 的关键区别

  • atomic_flag 一定是 lock-free 的;atomic 可能退化为内部加锁(罕见,但标准允许)
  • atomic_flag 不支持读取当前值(没有 load()),只能通过 test_and_set() “消耗性读取”
  • atomic_flag 不可拷贝、不可赋值、无默认构造函数(必须显式初始化)
  • 语义更纯粹:它不是“一个布尔原子变量”,而是“一个只能开关一次再清零的原子门闩”

一个最小可行示例

下面代码演示如何用 atomic_flag 保护一个共享计数器递增:

std::atomic_flag lock = ATOMIC_FLAG_INIT;
int counter = 0;

void increment() {
    while (lock.test_and_set(std::memory_order_acquire)) {
        std::this_thread::yield(); // 让出 CPU,减少忙等开销
    }
    ++counter;
    lock.clear(std::memory_order_release);
}

多个线程并发调用 increment() 时,只会有一个线程能“穿过”这个锁,其余线程在 test_and_set() 返回 true 后持续自旋等待,直到锁被释放。

基本上就这些。它不复杂,但容易忽略它的设计初衷——不是用来存状态,而是用来建同步机制的“砖块”。