unify lock logic

1 files changed, 2 insertions, 3 deletions

diff --git a/src/swarm/state_file.rs b/src/swarm/state_file.rs
index 8b883fc..ba9192e 100644
--- a/src/swarm/state_file.rs
+++ b/src/swarm/state_file.rs
@@ -15,6 +15,7 @@ use tokio::time::Instant;
 use tracing::{error, info, warn};
 
 use crate::files::{FileKind, Files};
+use crate::locks::MutexExt;
 use crate::state::AppState;
 use crate::swarm::members;
 
@@ -90,9 +91,7 @@ impl StateFile {
     }
 
     fn lock(&self) -> MutexGuard<'_, Inner> {
-        self.inner
-            .lock()
-            .unwrap_or_else(|poisoned| poisoned.into_inner())
+        self.inner.lock_or_recover()
     }
 
     fn write(&self, remembered: &Remembered) -> std::io::Result<()> {