attempt!

Author: Hywan

crates/matrix-sdk-common/src/cross_process_lock.rs

@@ -117,6 +117,12 @@ pub struct CrossProcessLockGuard {
     num_holders: Arc<AtomicU32>,
 }
 
+impl CrossProcessLockGuard {
+    fn new(num_holders: Arc<AtomicU32>) -> Self {
+        Self { num_holders }
+    }
+}
+
 impl Drop for CrossProcessLockGuard {
     fn drop(&mut self) {
         self.num_holders.fetch_sub(1, Ordering::SeqCst);
@@ -242,7 +248,9 @@ where
 
         // The lock is already poisoned? Let's stop here.
         if self.is_poisoned() {
-            return Err(CrossProcessLockError::LockPoisoned);
+            return Err(CrossProcessLockError::LockPoisoned(CrossProcessLockGuard::new(
+                self.num_holders.clone(),
+            )));
         }
 
         // If another thread obtained the lock, make sure to only superficially increase
@@ -280,7 +288,9 @@ where
                     );
                     self.is_poisoned.store(true, Ordering::SeqCst);
 
-                    return Err(CrossProcessLockError::LockPoisoned);
+                    return Err(CrossProcessLockError::LockPoisoned(CrossProcessLockGuard::new(
+                        self.num_holders.clone(),
+                    )));
                 }
 
                 // This was the same generation, no problem.
@@ -404,8 +414,7 @@ where
 
         self.num_holders.fetch_add(1, Ordering::SeqCst);
 
-        let guard = CrossProcessLockGuard { num_holders: self.num_holders.clone() };
-        Ok(Some(guard))
+        Ok(Some(CrossProcessLockGuard::new(self.num_holders.clone())))
     }
 
     /// Attempt to take the lock, with exponential backoff if the lock has
@@ -474,7 +483,7 @@ pub enum CrossProcessLockError {
     /// The lock has been poisoned, it means an holder took it before the lock
     /// was acquired or renewed.
     #[error("a lock has been poisoned")]
-    LockPoisoned,
+    LockPoisoned(CrossProcessLockGuard),
 
     #[error(transparent)]
     #[cfg(not(target_family = "wasm"))]