C++ incremento std :: atomic_int se diverso da zero

Question

Sto implementando un meccanismo di puntatore/indicatore debole utilizzando std::atomic s per il contatore di riferimento (come this). Per la conversione di un puntatore debole per uno forte ho bisogno di atomicamente

• controllo se il contatore di riferimento forte è diverso da zero
• in caso affermativo, incrementarlo
• so se qualcosa è cambiato.

C'è un modo per farlo utilizzando std::atomic_int? Penso che sia possibile utilizzare uno degli compare_exchange, ma non riesco a capirlo.

Answer 1

data la definizione 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 conterrà il valore precedente. Si noti che compare_exchange_weak verrà aggiornato in caso di errore.

Answer 2

Questo dovrebbe farlo:

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