add functionality for BlockNotifier to unregister a previously registered listener, in order to no longer receive events

Author: rloomba

lightning/src/chain/chaininterface.rs

@@ -19,6 +19,7 @@ use std::sync::atomic::{AtomicUsize, Ordering};
 use std::collections::HashSet;
 use std::ops::Deref;
 use std::marker::PhantomData;
+use std::ptr::eq as ptr_eq;
 
 /// Used to give chain error details upstream
 pub enum ChainError {
@@ -252,12 +253,21 @@ impl<'a, CL: Deref<Target = ChainListener + 'a> + 'a> BlockNotifier<'a, CL> {
 	}
 
 	/// Register the given listener to receive events.
-	// TODO: unregister
 	pub fn register_listener(&self, listener: CL) {
 		let mut vec = self.listeners.lock().unwrap();
 		vec.push(listener);
 	}
 
+    /// Unregister the given listener to no longer
+    /// receive events.
+    ///
+    /// If the same listener is registered multiple times, unregistering
+    /// will remove ALL occurrences of that listener.
+    pub fn unregister_listener(&self, listener: CL) {
+        let mut vec = self.listeners.lock().unwrap();
+        vec.retain(|item| !ptr_eq(&(**item) as *const _, &(*listener) as *const _));
+    }
+
 	/// Notify listeners that a block was connected given a full, unfiltered block.
 	///
 	/// Handles re-scanning the block and calling block_connected again if listeners register new
@@ -376,3 +386,59 @@ impl ChainWatchInterfaceUtil {
 		watched.does_match_tx(tx)
 	}
 }
+
+#[cfg(test)]
+mod tests {
+	use ln::functional_test_utils::{create_node_cfgs};
+	use super::{BlockNotifier, ChainListener};
+	use std::ptr::eq as ptr_eq;
+
+	#[test]
+	fn register_listener_test() {
+		let node_cfgs = create_node_cfgs(1);
+		let block_notifier = BlockNotifier::new(node_cfgs[0].chain_monitor.clone());
+		let listener = &node_cfgs[0].chan_monitor.simple_monitor as &ChainListener;
+		block_notifier.register_listener(listener);
+		let vec = block_notifier.listeners.lock().unwrap();
+		assert_eq!(vec.len(), 1);
+		let item = vec.first().clone().unwrap();
+		assert!(ptr_eq(&(**item) as *const _, &(*listener) as *const _));
+	}
+
+	#[test]
+	fn unregister_single_listener_test() {
+		let node_cfgs = create_node_cfgs(2);
+		let block_notifier = BlockNotifier::new(node_cfgs[0].chain_monitor.clone());
+		let listener1 = &node_cfgs[0].chan_monitor.simple_monitor as &ChainListener;
+		let listener2 = &node_cfgs[1].chan_monitor.simple_monitor as &ChainListener;
+		block_notifier.register_listener(listener1);
+		block_notifier.register_listener(listener2);
+		let vec = block_notifier.listeners.lock().unwrap();
+		assert_eq!(vec.len(), 2);
+		drop(vec);
+		block_notifier.unregister_listener(listener1);
+		let vec = block_notifier.listeners.lock().unwrap();
+		assert_eq!(vec.len(), 1);
+		let item = vec.first().clone().unwrap();
+		assert!(ptr_eq(&(**item) as *const _, &(*listener2) as *const _));
+	}
+
+	#[test]
+	fn unregister_multiple_of_the_same_listeners_test() {
+		let node_cfgs = create_node_cfgs(2);
+		let block_notifier = BlockNotifier::new(node_cfgs[0].chain_monitor.clone());
+		let listener1 = &node_cfgs[0].chan_monitor.simple_monitor as &ChainListener;
+		let listener2 = &node_cfgs[1].chan_monitor.simple_monitor as &ChainListener;
+		block_notifier.register_listener(listener1);
+		block_notifier.register_listener(listener1);
+		block_notifier.register_listener(listener2);
+		let vec = block_notifier.listeners.lock().unwrap();
+		assert_eq!(vec.len(), 3);
+		drop(vec);
+		block_notifier.unregister_listener(listener1);
+		let vec = block_notifier.listeners.lock().unwrap();
+		assert_eq!(vec.len(), 1);
+		let item = vec.first().clone().unwrap();
+		assert!(ptr_eq(&(**item) as *const _, &(*listener2) as *const _));
+	}
+}

lightning/src/ln/mod.rs

@@ -27,7 +27,8 @@ mod onion_utils;
 mod wire;
 
 #[cfg(test)]
-#[macro_use] mod functional_test_utils;
+#[macro_use]
+pub(crate) mod functional_test_utils;
 #[cfg(test)]
 mod functional_tests;
 #[cfg(test)]