C++ inkrementacja std :: atomic_int if niezerowa

Question

Wdrażam mechanizm wskaźnika/wskaźnika słabego przy użyciu std::atomic s dla licznika odwołań (np. this). Do konwersji słaby wskaźnik do silnego jednym muszę atomowo

• sprawdzić, czy licznik silne odniesienia jest niezerowe
• jeśli tak, to przyrost
• wiem, czy coś się zmieniło.

Czy istnieje sposób, aby to zrobić za pomocą std::atomic_int? Myślę, że to musi być możliwe przy użyciu jednego z compare_exchange, ale nie mogę tego rozgryźć.

Answer 1

Biorąc pod uwagę definicję 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 odbędzie poprzednią wartość. Pamiętaj, że compare_exchange_weak zaktualizuje poprzednie, jeśli się nie powiedzie.

Answer 2

ten powinien zrobić:

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; 
}