use atomic::swap instead of atomic::fetch_update when cleaning

Arian8j2 · Sep 18, 2024 · e3dddff · e3dddff
1 parent a40c338
commit e3dddff
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/forwarder/src/lib.rs b/forwarder/src/lib.rs
@@ -162,17 +162,8 @@ fn try_cleanup(peer_manager: &RwLock<PeerManager>) {
     let mut peers = peer_manager.write();
     let mut used_client_count = 0;
     for peer in peers.get_all() {
-        let result = peer
-            .used
-            .fetch_update(Ordering::Relaxed, Ordering::Relaxed, |used| {
-                if used {
-                    Some(false)
-                } else {
-                    None
-                }
-            });
-        // `used` was false
-        if result.is_err() {
+        let used = peer.used.swap(false, Ordering::Relaxed);
+        if !used {
             let client_addr = peer.get_client_addr();
             log::info!("cleaning peer that handled '{client_addr}'");
             peers.remove_peer(client_addr, peer.get_token());