unify lock logic

1 files changed, 2 insertions, 3 deletions

diff --git a/src/transport/sim.rs b/src/transport/sim.rs
index ce5977f..675f382 100644
--- a/src/transport/sim.rs
+++ b/src/transport/sim.rs
@@ -26,6 +26,7 @@ use tracing::{debug, warn};
 
 use super::mycelium::{decode_message, encode_message};
 use super::{EncodedMessage, MessageHandler, MessageKind, ParsedMessage, Transport};
+use crate::locks::MutexExt;
 use crate::testrand::Rng;
 
 /// How one direction of a link behaves.
@@ -102,9 +103,7 @@ impl SimNet {
     }
 
     fn lock(&self) -> MutexGuard<'_, NetInner> {
-        self.inner
-            .lock()
-            .unwrap_or_else(|poisoned| poisoned.into_inner())
+        self.inner.lock_or_recover()
     }
 
     /// A transport for a new node `id` on this net.