diff options
Diffstat (limited to 'src/util.rs')
| -rw-r--r-- | src/util.rs | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/src/util.rs b/src/util.rs index 425d26f..4676e64 100644 --- a/src/util.rs +++ b/src/util.rs @@ -1,5 +1,4 @@ use std::net::SocketAddr; -use std::net::ToSocketAddrs; use log::info; use serde::Serialize; @@ -71,6 +70,8 @@ pub fn parse_peer_addr(peer: &str) -> Option<(NodeID, SocketAddr)> { /// Parse and resolve a peer's address including public key, written in the format: /// `<public key hex>@<ip or hostname>:<port>` pub fn parse_and_resolve_peer_addr(peer: &str) -> Option<(NodeID, Vec<SocketAddr>)> { + use std::net::ToSocketAddrs; + let delim = peer.find('@')?; let (key, host) = peer.split_at(delim); let pubkey = NodeID::from_slice(&hex::decode(&key).ok()?)?; @@ -80,3 +81,18 @@ pub fn parse_and_resolve_peer_addr(peer: &str) -> Option<(NodeID, Vec<SocketAddr } Some((pubkey, hosts)) } + +/// async version of parse_and_resolve_peer_addr +pub async fn parse_and_resolve_peer_addr_async(peer: &str) -> Option<(NodeID, Vec<SocketAddr>)> { + let delim = peer.find('@')?; + let (key, host) = peer.split_at(delim); + let pubkey = NodeID::from_slice(&hex::decode(&key).ok()?)?; + let hosts = tokio::net::lookup_host(&host[1..]) + .await + .ok()? + .collect::<Vec<_>>(); + if hosts.is_empty() { + return None; + } + Some((pubkey, hosts)) +} |