fixup! Pass HRNResolver or DomainResolver into OnionMessenger

Author: chuksys

src/builder.rs

@@ -1531,6 +1531,8 @@ fn build_with_store_internal(
 	let om_resolver = match &config.hrn_config {
 		None => Arc::new(IgnoringMessageHandler {}),
 		Some(hrn_config) => {
+			let runtime_handle = runtime.handle();
+
 			let client_resolver: Arc<dyn DNSResolverMessageHandler + Send + Sync> =
 				match &hrn_config.client_resolution_config {
 					HRNResolverConfig::Blip32Onion => {
@@ -1550,15 +1552,17 @@ fn build_with_store_internal(
 							}
 						}));
 
-						hrn_res
+						hrn_res as Arc<dyn DNSResolverMessageHandler + Send + Sync>
 					},
 					HRNResolverConfig::LocalDns { dns_server_address } => {
 						let addr = dns_server_address
 							.parse()
 							.map_err(|_| BuildError::DNSResolverSetupFailed)?;
 						let hrn_res = Arc::new(DNSHrnResolver(addr));
 						hrn_resolver_out = Some(Arc::new(HRNResolver::Local(hrn_res)));
-						Arc::new(OMDomainResolver::ignoring_incoming_proofs(addr))
+						let resolver = Arc::new(OMDomainResolver::ignoring_incoming_proofs(addr));
+						resolver.set_runtime(runtime_handle.clone());
+						resolver as Arc<dyn DNSResolverMessageHandler + Send + Sync>
 					},
 				};
 
@@ -1578,7 +1582,7 @@ fn build_with_store_internal(
 					Arc::new(OMDomainResolver::with_runtime(
 						service_dns_addr,
 						Some(client_resolver),
-						Some(runtime.handle().clone()),
+						Some(runtime_handle.clone()),
 					))
 				} else {
 					log_error!(logger, "To act as an HRN resolution service, the DNS resolver must be configured to use a DNS server.");

src/lib.rs

@@ -1669,6 +1669,7 @@ impl Node {
 	/// # use ldk_node::entropy::{generate_entropy_mnemonic, NodeEntropy};
 	/// # use rand::distr::Alphanumeric;
 	/// # use rand::{rng, Rng};
+	/// # tokio::runtime::Runtime::new().unwrap().block_on(async {
 	/// # let mut config = Config::default();
 	/// # config.network = Network::Regtest;
 	/// # let mut temp_path = std::env::temp_dir();
@@ -1680,6 +1681,7 @@ impl Node {
 	/// # let node_entropy = NodeEntropy::from_bip39_mnemonic(mnemonic, None);
 	/// # let node = builder.build(node_entropy.into()).unwrap();
 	/// node.list_payments_with_filter(|p| p.direction == PaymentDirection::Outbound);
+	/// # });
 	/// ```
 	pub fn list_payments_with_filter<F: FnMut(&&PaymentDetails) -> bool>(
 		&self, f: F,