From d33c3e9735fbbfb47953cfe154b8297613efc452 Mon Sep 17 00:00:00 2001
From: Sosthene00 <674694@protonmail.ch>
Date: Thu, 21 Mar 2024 18:06:17 +0100
Subject: [PATCH] Add MutexExt trait

---
 src/main.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index da7fedd..a55d23f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -58,6 +58,17 @@ type PeerMap = Arc<Mutex<HashMap<SocketAddr, Tx>>>;
 
 const FAUCET_AMT: Amount = Amount::from_sat(1000);
 
+pub(crate) trait MutexExt<T> {
+    fn lock_anyhow(&self) -> Result<MutexGuard<T>, Error>;
+}
+
+impl<T: Debug> MutexExt<T> for Mutex<T> {
+    fn lock_anyhow(&self) -> Result<MutexGuard<T>, Error> {
+        self.lock()
+            .map_err(|e| Error::msg(format!("Failed to lock: {}", e)))
+    }
+}
+
 enum BroadcastType {
     Sender(SocketAddr),
     ExcludeSender(SocketAddr),