From de5ec6e1bcbb428ee672d7965c0656e628adde46 Mon Sep 17 00:00:00 2001
From: David Barsky <me@davidbarsky.com>
Date: Sat, 21 Dec 2019 11:30:00 -0500
Subject: dns: provide `lookup_host` function (#1870)

`ToSocketAddrs` is a sealed trait pending changes in Rust that will allow
defining async trait fns. Until then, `net::lookup_host` is provided as a way
to convert a `T: ToSocketAddrs` into `SocketAddr`s.
---
 tokio/src/net/lookup_host.rs   | 38 ++++++++++++++++++++++++++++++++++++++
 tokio/src/net/mod.rs           |  5 +++++
 tokio/tests/net_lookup_host.rs | 36 ++++++++++++++++++++++++++++++++++++
 3 files changed, 79 insertions(+)
 create mode 100644 tokio/src/net/lookup_host.rs
 create mode 100644 tokio/tests/net_lookup_host.rs

diff --git a/tokio/src/net/lookup_host.rs b/tokio/src/net/lookup_host.rs
new file mode 100644
index 00000000..3098b463
--- /dev/null
+++ b/tokio/src/net/lookup_host.rs
@@ -0,0 +1,38 @@
+cfg_dns! {
+    use crate::net::addr::ToSocketAddrs;
+
+    use std::io;
+    use std::net::SocketAddr;
+
+    /// Performs a DNS resolution.
+    ///
+    /// The returned iterator may not actually yield any values depending on the
+    /// outcome of any resolution performed.
+    ///
+    /// This API is not intended to cover all DNS use cases. Anything beyond the
+    /// basic use case should be done with a specialized library.
+    ///
+    /// # Examples
+    ///
+    /// To resolve a DNS entry:
+    ///
+    /// ```no_run
+    /// use tokio::net;
+    /// use std::io;
+    ///
+    /// #[tokio::main]
+    /// async fn main() -> io::Result<()> {
+    ///     for addr in net::lookup_host("localhost:3000").await? {
+    ///         println!("socket address is {}", addr);
+    ///     }
+    ///
+    ///     Ok(())
+    /// }
+    /// ```
+    pub async fn lookup_host<T>(host: T) -> io::Result<impl Iterator<Item = SocketAddr>>
+    where
+        T: ToSocketAddrs
+    {
+        host.to_socket_addrs().await
+    }
+}
diff --git a/tokio/src/net/mod.rs b/tokio/src/net/mod.rs
index 2be8f34e..83b12fd2 100644
--- a/tokio/src/net/mod.rs
+++ b/tokio/src/net/mod.rs
@@ -25,6 +25,11 @@
 mod addr;
 pub use addr::ToSocketAddrs;
 
+cfg_dns! {
+    mod lookup_host;
+    pub use lookup_host::lookup_host;
+}
+
 cfg_tcp! {
     pub mod tcp;
     pub use tcp::listener::TcpListener;
diff --git a/tokio/tests/net_lookup_host.rs b/tokio/tests/net_lookup_host.rs
new file mode 100644
index 00000000..4d064029
--- /dev/null
+++ b/tokio/tests/net_lookup_host.rs
@@ -0,0 +1,36 @@
+use tokio::net;
+use tokio_test::assert_ok;
+
+use std::io;
+use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr};
+
+#[tokio::test]
+async fn lookup_socket_addr() {
+    let addr: SocketAddr = "127.0.0.1:8000".parse().unwrap();
+
+    let actual = assert_ok!(net::lookup_host(addr).await).collect::<Vec<_>>();
+    assert_eq!(vec![addr], actual);
+}
+
+#[tokio::test]
+async fn lookup_str_socket_addr() {
+    let addr: SocketAddr = "127.0.0.1:8000".parse().unwrap();
+
+    let actual = assert_ok!(net::lookup_host("127.0.0.1:8000").await).collect::<Vec<_>>();
+    assert_eq!(vec![addr], actual);
+}
+
+#[tokio::test]
+async fn resolve_dns() -> io::Result<()> {
+    let mut hosts = net::lookup_host("localhost:3000").await?;
+    let host = hosts.next().unwrap();
+
+    let expected = if host.is_ipv4() {
+        SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 3000)
+    } else {
+        SocketAddr::new(IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)), 3000)
+    };
+    assert_eq!(host, expected);
+
+    Ok(())
+}
-- 
cgit v1.2.3