Skip to content

Commit

Permalink
feat(dns): export client::connect::dns module, and
Browse files Browse the repository at this point in the history
`TokioThreadpoolGaiResolver` type.
  • Loading branch information
seanmonstar committed Oct 23, 2018
1 parent 1e8d643 commit 34d780a
Show file tree
Hide file tree
Showing 5 changed files with 60 additions and 41 deletions.
4 changes: 2 additions & 2 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,8 @@ tokio-executor = { version = "0.1.0", optional = true }
tokio-io = "0.1"
tokio-reactor = { version = "0.1", optional = true }
tokio-tcp = { version = "0.1", optional = true }
tokio-threadpool = { version = "0.1.3", optional = true }
tokio-timer = { version = "0.2", optional = true }
tokio-threadpool = { version = "0.1", optional = true }
want = "0.0.6"

[dev-dependencies]
Expand All @@ -62,8 +62,8 @@ runtime = [
"tokio-executor",
"tokio-reactor",
"tokio-tcp",
"tokio-timer",
"tokio-threadpool",
"tokio-timer",
]
nightly = []
__internal_flaky_tests = []
Expand Down
81 changes: 46 additions & 35 deletions src/client/connect/dns.rs
Original file line number Diff line number Diff line change
@@ -1,3 +1,11 @@
//! The `Resolve` trait, support types, and some basic implementations.
//!
//! This module contains:
//!
//! - A [`GaiResolver`](GaiResolver) that is the default resolver for the
//! `HttpConnector`.
//! - The [`Resolve`](Resolve) trait and related types to build a custom
//! resolver for use with the `HttpConnector`.
use std::{fmt, io, vec};
use std::net::{
IpAddr, Ipv4Addr, Ipv6Addr,
Expand All @@ -10,12 +18,10 @@ use futures::{Async, Future, Poll};
use futures::future::{Executor, ExecuteError};
use futures::sync::oneshot;
use futures_cpupool::{Builder as CpuPoolBuilder};
use tokio_threadpool;

use self::sealed::GaiTask;

#[cfg(feature = "runtime")]
pub use self::blocking::{TokioThreadpoolGaiFuture, TokioThreadpoolGaiResolver};

/// Resolve a hostname to a set of IP addresses.
pub trait Resolve {
/// The set of IP addresses to try to connect to.
Expand All @@ -37,10 +43,12 @@ pub struct GaiResolver {
executor: GaiExecutor,
}

/// An iterator of IP addresses returned from `getaddrinfo`.
pub struct GaiAddrs {
inner: IpAddrs,
}

/// A future to resole a name returned by `GaiResolver`.
pub struct GaiFuture {
rx: oneshot::SpawnHandle<IpAddrs, io::Error>,
}
Expand Down Expand Up @@ -242,46 +250,49 @@ pub(super) mod sealed {
}
}

#[cfg(feature = "runtime")]
mod blocking {
use futures::{Async, Future, Poll};
use std::io;
use std::net::ToSocketAddrs;
use tokio_threadpool;

use super::{Name, IpAddrs, GaiAddrs, Resolve};
/// A resolver using `getaddrinfo` calls via the `tokio_threadpool::blocking` API.
///
/// Unlike the `GaiResolver` this will not spawn dedicated threads, but only works when running on the
/// multi-threaded Tokio runtime.
#[derive(Clone, Debug)]
pub struct TokioThreadpoolGaiResolver(());

/// A resolver using `getaddrinfo` calls via the `tokio_threadpool::blocking` API.
///
/// Unlike the `GaiResolver` this will not spawn dedicated threads, but only works when running on the
/// multi-threaded Tokio runtime.
#[derive(Clone)]
pub struct TokioThreadpoolGaiResolver(());
/// The future returned by `TokioThreadpoolGaiResolver`.
#[derive(Debug)]
pub struct TokioThreadpoolGaiFuture {
name: Name,
}

pub struct TokioThreadpoolGaiFuture {
name: Name,
impl TokioThreadpoolGaiResolver {
/// Creates a new DNS resolver that will use tokio threadpool's blocking
/// feature.
///
/// **Requires** its futures to be run on the threadpool runtime.
pub fn new() -> Self {
TokioThreadpoolGaiResolver::new()
}
}

impl Resolve for TokioThreadpoolGaiResolver {
type Addrs = GaiAddrs;
type Future = TokioThreadpoolGaiFuture;
impl Resolve for TokioThreadpoolGaiResolver {
type Addrs = GaiAddrs;
type Future = TokioThreadpoolGaiFuture;

fn resolve(&self, name: Name) -> TokioThreadpoolGaiFuture {
TokioThreadpoolGaiFuture { name }
}
fn resolve(&self, name: Name) -> TokioThreadpoolGaiFuture {
TokioThreadpoolGaiFuture { name }
}
}

impl Future for TokioThreadpoolGaiFuture {
type Item = GaiAddrs;
type Error = io::Error;

impl Future for TokioThreadpoolGaiFuture {
type Item = GaiAddrs;
type Error = io::Error;

fn poll(&mut self) -> Poll<GaiAddrs, io::Error> {
match tokio_threadpool::blocking(|| (self.name.as_str(), 0).to_socket_addrs()) {
Ok(Async::Ready(Ok(iter))) => Ok(Async::Ready(GaiAddrs { inner: IpAddrs { iter } })),
Ok(Async::Ready(Err(e))) => Err(e),
Ok(Async::NotReady) => Ok(Async::NotReady),
Err(e) => Err(io::Error::new(io::ErrorKind::Other, e)),
}
fn poll(&mut self) -> Poll<GaiAddrs, io::Error> {
match tokio_threadpool::blocking(|| (self.name.as_str(), 0).to_socket_addrs()) {
Ok(Async::Ready(Ok(iter))) => Ok(Async::Ready(GaiAddrs { inner: IpAddrs { iter } })),
Ok(Async::Ready(Err(e))) => Err(e),
Ok(Async::NotReady) => Ok(Async::NotReady),
Err(e) => Err(io::Error::new(io::ErrorKind::Other, e)),
}
}
}
Expand Down
11 changes: 10 additions & 1 deletion src/client/connect/http.rs
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ use tokio_tcp::{TcpStream, ConnectFuture};
use tokio_timer::Delay;

use super::{Connect, Connected, Destination};
use super::dns::{self, GaiResolver, Resolve};
use super::dns::{self, GaiResolver, Resolve, TokioThreadpoolGaiResolver};

/// A connector for the `http` scheme.
///
Expand Down Expand Up @@ -99,6 +99,15 @@ impl HttpConnector {
}
}

impl HttpConnector<TokioThreadpoolGaiResolver> {
/// Construct a new HttpConnector using the `TokioThreadpoolGaiResolver`.
///
/// This resolver **requires** the threadpool runtime to be used.
pub fn new_with_tokio_threadpool_resolver() -> Self {
HttpConnector::new_with_resolver(TokioThreadpoolGaiResolver::new())
}
}


impl<R> HttpConnector<R> {
/// Construct a new HttpConnector.
Expand Down
3 changes: 1 addition & 2 deletions src/client/connect/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,8 @@ use futures::Future;
use http::{uri, Response, Uri};
use tokio_io::{AsyncRead, AsyncWrite};

#[cfg(feature = "runtime")] mod dns;
#[cfg(feature = "runtime")] pub mod dns;
#[cfg(feature = "runtime")] mod http;
#[cfg(feature = "runtime")] pub use self::dns::{GaiResolver, Name, Resolve};
#[cfg(feature = "runtime")] pub use self::http::{HttpConnector, HttpInfo};

/// Connect to a destination, returning an IO transport.
Expand Down
2 changes: 1 addition & 1 deletion src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@ extern crate time;
#[macro_use] extern crate tokio_io;
#[cfg(feature = "runtime")] extern crate tokio_reactor;
#[cfg(feature = "runtime")] extern crate tokio_tcp;
#[cfg(feature = "runtime")] extern crate tokio_timer;
#[cfg(feature = "runtime")] extern crate tokio_threadpool;
#[cfg(feature = "runtime")] extern crate tokio_timer;
extern crate want;

#[cfg(all(test, feature = "nightly"))]
Expand Down

0 comments on commit 34d780a

Please sign in to comment.