unify lock logic

1 files changed, 2 insertions, 2 deletions

diff --git a/src/swarm/peer.rs b/src/swarm/peer.rs
index 651e067..ec6543b 100644
--- a/src/swarm/peer.rs
+++ b/src/swarm/peer.rs
@@ -14,6 +14,7 @@ use tracing::{debug, info, instrument, trace, warn};
 use crate::document::meta;
 use crate::document::replica::{Epoch, EpochRelation};
 use crate::document::store::{self, StoreError};
+use crate::locks::MutexExt;
 use crate::state::{AppState, HelloState, Liveness, PeerState, Standing};
 use crate::swarm::sync::{SyncEvent, handle_sync};
 use crate::swarm::topology;
@@ -543,8 +544,7 @@ pub(crate) async fn run_peer_worker(state: AppState) {
                 state.traffic.retain(|peer| known.contains_key(peer));
                 state
                     .throughput
-                    .lock()
-                    .unwrap_or_else(|poisoned| poisoned.into_inner())
+                    .lock_or_recover()
                     .retain(|peer| known.contains_key(peer));
             }
             // Fallback for what doesn't notify (`topology::run_worker`).