c++ - 如果非零，C++ 增加 std::atomic_int

Question

我正在实现一个指针/弱指针机制，使用std::atomics 作为引用计数器（像这样）。为了将弱指针转换为强指针，我需要原子地

• 检查强引用计数器是否非零
• 如果是这样，增加它
• 知道某事是否发生了变化。

有没有办法做到这一点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如果失败，将更新以前的内容。

Answer 2

这应该这样做：

bool increment_if_non_zero(std::atomic<int>& i) {
    int expected = i.load();
    int to_be_loaded = expected;

    do {
        if(expected == 0) {
            to_be_loaded = expected;
        }
        else {
            to_be_loaded = expected + 1;
        }
    } while(!i.compare_exchange_weak(expected, to_be_loaded));

    return expected;
}