Replace manual iteration with for loops in outbound gossip sync

Author: TheBlueMatt

lightning/src/routing/gossip.rs

@@ -320,28 +320,24 @@ where C::Target: chain::Access, L::Target: Logger
 
 	fn get_next_channel_announcement(&self, starting_point: u64) -> Option<(ChannelAnnouncement, Option<ChannelUpdate>, Option<ChannelUpdate>)> {
 		let channels = self.network_graph.channels.read().unwrap();
-		let mut iter = channels.range(starting_point..);
-		loop {
-			if let Some((_, ref chan)) = iter.next() {
-				if chan.announcement_message.is_some() {
-					let chan_announcement = chan.announcement_message.clone().unwrap();
-					let mut one_to_two_announcement: Option<msgs::ChannelUpdate> = None;
-					let mut two_to_one_announcement: Option<msgs::ChannelUpdate> = None;
-					if let Some(one_to_two) = chan.one_to_two.as_ref() {
-						one_to_two_announcement = one_to_two.last_update_message.clone();
-					}
-					if let Some(two_to_one) = chan.two_to_one.as_ref() {
-						two_to_one_announcement = two_to_one.last_update_message.clone();
-					}
-					return Some((chan_announcement, one_to_two_announcement, two_to_one_announcement));
-				} else {
-					// TODO: We may end up sending un-announced channel_updates if we are sending
-					// initial sync data while receiving announce/updates for this channel.
+		for (_, ref chan) in channels.range(starting_point..) {
+			if chan.announcement_message.is_some() {
+				let chan_announcement = chan.announcement_message.clone().unwrap();
+				let mut one_to_two_announcement: Option<msgs::ChannelUpdate> = None;
+				let mut two_to_one_announcement: Option<msgs::ChannelUpdate> = None;
+				if let Some(one_to_two) = chan.one_to_two.as_ref() {
+					one_to_two_announcement = one_to_two.last_update_message.clone();
 				}
+				if let Some(two_to_one) = chan.two_to_one.as_ref() {
+					two_to_one_announcement = two_to_one.last_update_message.clone();
+				}
+				return Some((chan_announcement, one_to_two_announcement, two_to_one_announcement));
 			} else {
-				return None;
+				// TODO: We may end up sending un-announced channel_updates if we are sending
+				// initial sync data while receiving announce/updates for this channel.
 			}
 		}
+		None
 	}
 
 	fn get_next_node_announcement(&self, starting_point: Option<&PublicKey>) -> Option<NodeAnnouncement> {
@@ -353,17 +349,14 @@ where C::Target: chain::Access, L::Target: Logger
 			} else {
 				nodes.range::<NodeId, _>(..)
 			};
-		loop {
-			if let Some((_, ref node)) = iter.next() {
-				if let Some(node_info) = node.announcement_info.as_ref() {
-					if let Some(msg) = node_info.announcement_message.clone() {
-						return Some(msg);
-					}
+		for (_, ref node) in iter {
+			if let Some(node_info) = node.announcement_info.as_ref() {
+				if let Some(msg) = node_info.announcement_message.clone() {
+					return Some(msg);
 				}
-			} else {
-				return None;
 			}
 		}
+		None
 	}
 
 	/// Initiates a stateless sync of routing gossip information with a peer