C++ 递增 std::atomic_int 如果非零

Question

我正在为引用计数器(如 this)使用 std::atomic 实现指针/弱指针机制。为了将弱指针转换为强指针，我需要以原子方式进行

• 检查强引用计数器是否非零
• 如果是，增加它
• 了解是否发生了变化。

有没有办法使用 std::atomic_int 来做到这一点？我认为必须使用 compare_exchange 之一，但我想不通。

Answer 1

给定定义std::atomic<int> ref_count;

int previous = ref_count.load();
for (;;)
{
    if (previous == 0)
        break;
    if (ref_count.compare_exchange_weak(previous, previous + 1))
        break;
}

previous将保留以前的值。注意 compare_exchange_weak如果失败，将更新以前的内容。