unify lock logic

1 files changed, 2 insertions, 3 deletions

diff --git a/src/document/file.rs b/src/document/file.rs
index a773502..6e7a4b6 100644
--- a/src/document/file.rs
+++ b/src/document/file.rs
@@ -28,6 +28,7 @@ use tracing::{debug, error, warn};
 
 use crate::document::replica::Epoch;
 use crate::files::{FileKind, Files};
+use crate::locks::MutexExt;
 use crate::state::AppState;
 
 const MAGIC: &[u8] = b"continuum-document/1\n";
@@ -175,9 +176,7 @@ impl DocumentFile {
     }
 
     fn lock(&self) -> MutexGuard<'_, Option<Written>> {
-        self.written
-            .lock()
-            .unwrap_or_else(|poisoned| poisoned.into_inner())
+        self.written.lock_or_recover()
     }
 }