impl lockablescore for rwlock

Author: jbesraa

lightning/src/routing/scoring.rs

@@ -68,7 +68,7 @@ use core::convert::TryInto;
 use core::ops::{Deref, DerefMut};
 use core::time::Duration;
 use crate::io::{self, Read};
-use crate::sync::{Mutex, MutexGuard};
+use crate::sync::{Mutex, MutexGuard, RwLockWriteGuard, RwLock};
 
 /// We define Score ever-so-slightly differently based on whether we are being built for C bindings
 /// or not. For users, `LockableScore` must somehow be writeable to disk. For Rust users, this is
@@ -185,6 +185,19 @@ impl<'a, T: 'a + Score> LockableScore<'a> for Mutex<T> {
 	}
 }
 
+impl<'a, T: 'a + Score + Clone> LockableScore<'a> for RwLock<T> {
+    type WriteLocked = RwLockWriteGuard<'a, T>;
+    type ScoreLocked = T;
+
+    fn write_lock(&'a self) -> Self::WriteLocked {
+        self.write().unwrap()
+    }
+
+    fn score_lock(&'a self) -> Self::ScoreLocked {
+        self.read().unwrap().clone()
+    }
+}
+
 impl<'a, T: 'a + Score> LockableScore<'a> for RefCell<T> {
 	type ScoreLocked = RefMut<'a, T>;
 	type WriteLocked = RefMut<'a, T>;