Move connection logic to connection.rs

.. just a minor cleanup to further modularize the codebase. Also, we'll
be reusing these methods in `Event::ConnectionNeeded` soon.

Author: tnull

src/connection.rs

@@ -0,0 +1,66 @@
+use crate::logger::{log_error, log_info, FilesystemLogger, Logger};
+use crate::types::PeerManager;
+use crate::Error;
+
+use lightning::ln::msgs::SocketAddress;
+
+use bitcoin::secp256k1::PublicKey;
+
+use std::net::ToSocketAddrs;
+use std::sync::Arc;
+use std::time::Duration;
+
+pub(crate) async fn connect_peer_if_necessary(
+	node_id: PublicKey, addr: SocketAddress, peer_manager: Arc<PeerManager>,
+	logger: Arc<FilesystemLogger>,
+) -> Result<(), Error> {
+	for (pman_node_id, _pman_addr) in peer_manager.get_peer_node_ids() {
+		if node_id == pman_node_id {
+			return Ok(());
+		}
+	}
+
+	do_connect_peer(node_id, addr, peer_manager, logger).await
+}
+
+pub(crate) async fn do_connect_peer(
+	node_id: PublicKey, addr: SocketAddress, peer_manager: Arc<PeerManager>,
+	logger: Arc<FilesystemLogger>,
+) -> Result<(), Error> {
+	log_info!(logger, "Connecting to peer: {}@{}", node_id, addr);
+
+	let socket_addr = addr
+		.to_socket_addrs()
+		.map_err(|e| {
+			log_error!(logger, "Failed to resolve network address: {}", e);
+			Error::InvalidSocketAddress
+		})?
+		.next()
+		.ok_or(Error::ConnectionFailed)?;
+
+	match lightning_net_tokio::connect_outbound(Arc::clone(&peer_manager), node_id, socket_addr)
+		.await
+	{
+		Some(connection_closed_future) => {
+			let mut connection_closed_future = Box::pin(connection_closed_future);
+			loop {
+				match futures::poll!(&mut connection_closed_future) {
+					std::task::Poll::Ready(_) => {
+						log_info!(logger, "Peer connection closed: {}@{}", node_id, addr);
+						return Err(Error::ConnectionFailed);
+					}
+					std::task::Poll::Pending => {}
+				}
+				// Avoid blocking the tokio context by sleeping a bit
+				match peer_manager.get_peer_node_ids().iter().find(|(id, _addr)| *id == node_id) {
+					Some(_) => return Ok(()),
+					None => tokio::time::sleep(Duration::from_millis(10)).await,
+				}
+			}
+		}
+		None => {
+			log_error!(logger, "Failed to connect to peer: {}@{}", node_id, addr);
+			Err(Error::ConnectionFailed)
+		}
+	}
+}

src/lib.rs

@@ -76,6 +76,7 @@
 #![cfg_attr(docsrs, feature(doc_auto_cfg))]
 
 mod builder;
+mod connection;
 mod error;
 mod event;
 mod fee_estimator;
@@ -116,6 +117,7 @@ pub use builder::BuildError;
 #[cfg(not(feature = "uniffi"))]
 pub use builder::NodeBuilder as Builder;
 
+use connection::{connect_peer_if_necessary, do_connect_peer};
 use event::{EventHandler, EventQueue};
 use gossip::GossipSource;
 use liquidity::LiquiditySource;
@@ -1831,58 +1833,3 @@ impl Drop for Node {
 		let _ = self.stop();
 	}
 }
-
-async fn connect_peer_if_necessary(
-	node_id: PublicKey, addr: SocketAddress, peer_manager: Arc<PeerManager>,
-	logger: Arc<FilesystemLogger>,
-) -> Result<(), Error> {
-	for (pman_node_id, _pman_addr) in peer_manager.get_peer_node_ids() {
-		if node_id == pman_node_id {
-			return Ok(());
-		}
-	}
-
-	do_connect_peer(node_id, addr, peer_manager, logger).await
-}
-
-async fn do_connect_peer(
-	node_id: PublicKey, addr: SocketAddress, peer_manager: Arc<PeerManager>,
-	logger: Arc<FilesystemLogger>,
-) -> Result<(), Error> {
-	log_info!(logger, "Connecting to peer: {}@{}", node_id, addr);
-
-	let socket_addr = addr
-		.to_socket_addrs()
-		.map_err(|e| {
-			log_error!(logger, "Failed to resolve network address: {}", e);
-			Error::InvalidSocketAddress
-		})?
-		.next()
-		.ok_or(Error::ConnectionFailed)?;
-
-	match lightning_net_tokio::connect_outbound(Arc::clone(&peer_manager), node_id, socket_addr)
-		.await
-	{
-		Some(connection_closed_future) => {
-			let mut connection_closed_future = Box::pin(connection_closed_future);
-			loop {
-				match futures::poll!(&mut connection_closed_future) {
-					std::task::Poll::Ready(_) => {
-						log_info!(logger, "Peer connection closed: {}@{}", node_id, addr);
-						return Err(Error::ConnectionFailed);
-					}
-					std::task::Poll::Pending => {}
-				}
-				// Avoid blocking the tokio context by sleeping a bit
-				match peer_manager.get_peer_node_ids().iter().find(|(id, _addr)| *id == node_id) {
-					Some(_) => return Ok(()),
-					None => tokio::time::sleep(Duration::from_millis(10)).await,
-				}
-			}
-		}
-		None => {
-			log_error!(logger, "Failed to connect to peer: {}@{}", node_id, addr);
-			Err(Error::ConnectionFailed)
-		}
-	}
-}