C++はゼロ以外の場合std :: atomic_intをインクリメントします

Question

std::atomicを参照カウンタに使用するポインタ/弱ポインタ​​機構を実装しています（thisなど）。強いものに弱いのポインタを変換するために、私は強い参照カウンタがそうであれば

ゼロでなければ

• チェックをアトミックにする必要があり、それは
• 何かが変更されたかどうかを知るインクリメント。

std::atomic_intを使用してこれを行う方法はありますか？私はそれがcompare_exchangeの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考える

Answer 1

は前の値を保持します。 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; 
}