X-Git-Url: http://git.bitcoin.ninja/index.cgi?p=rust-lightning;a=blobdiff_plain;f=lightning%2Fsrc%2Frouting%2Fnetwork_graph.rs;h=dd6d7d7ed734aadbd7f89f93308aefd0223a9f79;hp=faf2b5ff208ae1789df094e20d45f9e647fe84cd;hb=0e3bf19b664ad3f9dd252c9ed50ed365ff827524;hpb=5dc724f87532333059583c6b77ff90fae10cf45d

diff --git a/lightning/src/routing/network_graph.rs b/lightning/src/routing/network_graph.rs
index faf2b5ff..dd6d7d7e 100644
--- a/lightning/src/routing/network_graph.rs
+++ b/lightning/src/routing/network_graph.rs
@@ -67,25 +67,6 @@ impl NetGraphMsgHandler {
 			logger: logger.clone(),
 		}
 	}
-
-	/// Get network addresses by node id.
-	/// Returns None if the requested node is completely unknown,
-	/// or if node announcement for the node was never received.
-	pub fn get_addresses(&self, pubkey: &PublicKey) -> Option<Vec<NetAddress>> {
-		let network = self.network_graph.read().unwrap();
-		if let Some(node) = network.get_nodes().get(pubkey) {
-			if let Some(node_info) = node.announcement_info.as_ref() {
-				return Some(node_info.addresses.clone())
-			}
-		}
-		None
-	}
-
-	/// Dumps the entire network view of this NetGraphMsgHandler to the logger provided in the constructor at
-	/// level Trace
-	pub fn trace_state(&self) {
-		log_trace!(self, "{}", self.network_graph.read().unwrap());
-	}
 }
 
 
@@ -509,6 +490,18 @@ impl NetworkGraph {
 	/// Returns all known nodes' public keys along with announced node info.
 	pub fn get_nodes<'a>(&'a self) -> &'a BTreeMap<PublicKey, NodeInfo> { &self.nodes }
 
+	/// Get network addresses by node id.
+	/// Returns None if the requested node is completely unknown,
+	/// or if node announcement for the node was never received.
+	pub fn get_addresses<'a>(&'a self, pubkey: &PublicKey) -> Option<&'a Vec<NetAddress>> {
+		if let Some(node) = self.nodes.get(pubkey) {
+			if let Some(node_info) = node.announcement_info.as_ref() {
+				return Some(&node_info.addresses)
+			}
+		}
+		None
+	}
+
 	/// For an already known node (from channel announcements), update its stored properties from a given node announcement
 	/// Announcement signatures are checked here only if Secp256k1 object is provided.
 	fn update_node_from_announcement(&mut self, msg: &msgs::NodeAnnouncement, secp_ctx: Option<&Secp256k1<secp256k1::VerifyOnly>>) -> Result<bool, LightningError> {