unify lock logic

1 files changed, 2 insertions, 3 deletions

diff --git a/src/remake/local.rs b/src/remake/local.rs
index 517367b..8e9d8ba 100644
--- a/src/remake/local.rs
+++ b/src/remake/local.rs
@@ -16,6 +16,7 @@ use tokio::time::Instant;
 use super::records::Retention;
 use super::snapshot::Snapshot;
 use crate::document::replica::Epoch;
+use crate::locks::MutexExt;
 
 pub(crate) struct RemakeState {
     /// When this node started (Unix milliseconds): votes it cast before may be lost
@@ -185,9 +186,7 @@ impl RemakeState {
     }
 
     fn lock(&self) -> MutexGuard<'_, Inner> {
-        self.inner
-            .lock()
-            .unwrap_or_else(|poisoned| poisoned.into_inner())
+        self.inner.lock_or_recover()
     }
 
     pub(crate) fn store_prebuilt(&self, id: &str, snapshot: Snapshot) {