C++的std::atomic_flag怎么用_C++中最简单的原子类型实现自旋锁

穿越時空 2025-12-23 00:00:00 次阅读

首先给出明确答案，std::atomic_flag 是C++中用于实现自旋锁的轻量级原子布尔类型，仅支持test_and_set和clear操作，初始为清除状态，不可拷贝，确保线程安全。

std::atomic_flag 是 C++ 中最轻量的原子类型，它只有两个状态：置位（true）和清除（false），初始时通常处于清除状态。它不支持拷贝构造和赋值，保证了操作的原子性，非常适合用来实现自旋锁（spinlock）。

基本用法：test_and_set 与 clear

std::atomic_flag 提供两个主要操作：

test_and_set()：原子地将标志设为 true，并返回它之前的值。
clear()：原子地将标志设为 false。

这两个操作是实现自旋锁的核心。

用 atomic_flag 实现自旋锁

自旋锁是一种简单的同步机制：当一个线程持有锁时，其他尝试获取锁的线程会不断循环（“自旋”）等待，直到锁被释放。

以下是一个基于 std::atomic_flag 的自旋锁实现：

#include 
#include 
#include 

class spinlock {
    std::atomic_flag flag = ATOMIC_FLAG_INIT;

public:
    void lock() {
        while (flag.test_and_set()) { 
            // 自旋等待
        }
    }

    void unlock() {
        flag.clear();
    }
};

spinlock sp;

void worker(int id) {
    for (int i = 0; i < 2; ++i) {
        sp.lock();
        std::cout << "Worker " << id << " in critical section.\n";
        sp.unlock();
        // 模拟一些工作
        std::this_thread::yield();
    }
}

多个线程调用 worker 函数时，会互斥地进入临界区，输出不会交错。