compare_exchange_weak 的用法

compare_exchange_weak是 C++ 原子库中用于执行比较并交换（CAS）操作的函数，它原子地比较原子变量的当前值与期望值，如果相等则更新为目标值，否则更新期望值为当前值。

bool compare_exchange_weak(T& expected, T desired, std::memory_order success = std::memory_order_seq_cst, std::memory_order failure = std::memory_order_seq_cst) noexcept;

返回值：

• true：交换成功，原子变量被更新为desired
• false：交换失败，expected被更新为原子变量的当前值

关键点：整个操作是原子的，不可中断。

核心行为（伪代码）

bool compare_exchange_weak(T& expected, T desired) { if (atomic_value == expected) { atomic_value = desired; // 成功：更新原子变量 return true; } else { expected = atomic_value; // 失败：更新期望值为当前值 return false; } }

典型用法：循环内重试

由于weak可能虚假失败，必须配合循环使用：

std::atomic<int> val(10); // 正确用法：do-while 循环 int expected = val.load(); do { int desired = expected * 2; // 计算新值 } while (!val.compare_exchange_weak(expected, desired)); // 或者使用 while 循环 int expected = val.load(); while (!val.compare_exchange_weak(expected, expected * 2)) { // expected 已被更新为最新值，继续重试 }

实际应用场景

1. 原子计数器

std::atomic<int> counter(0); void increment() { int expected = counter.load(); while (!counter.compare_exchange_weak(expected, expected + 1)) { // expected 自动更新为最新值，继续重试 } }

2. 自旋锁

std::atomic<bool> lock(false); void acquire_lock() { bool expected = false; while (!lock.compare_exchange_weak(expected, true)) { expected = false; // 重置 expected，准备下次尝试 } } void release_lock() { lock.store(false); }

3. 无锁栈的 push 操作

struct Node { int value; Node* next; }; std::atomic<Node*> head(nullptr); void push(int val) { Node* new_node = new Node{val, nullptr}; Node* expected = head.load(); do { new_node->next = expected; } while (!head.compare_exchange_weak(expected, new_node)); }

常见错误

错误1：不使用循环

// ❌ 错误：可能虚假失败 int expected = val.load(); val.compare_exchange_weak(expected, desired); // 可能无声失败

错误2：循环内不更新expected

// ❌ 错误：expected 没有重置 while (!val.compare_exchange_weak(expected, desired)) { // expected 已被更新，但这里没有重新计算 desired // 如果 desired 依赖于 expected，需要重新计算 }

错误3：混淆weak和strong

// ❌ 错误：单次尝试用 weak if (flag.compare_exchange_weak(expected, true)) { // 可能虚假失败 // 获取锁 } // ✅ 正确：单次尝试用 strong if (flag.compare_exchange_strong(expected, true)) { // 获取锁 }

总结

最佳实践：在循环中优先使用weak，非循环场景使用strong。