0321ea043d
These types were missing a short docstring, leaving the comment space to their right blank on the crate page. Also, this takes advantage of intra-rustdoc links to link terms: https://github.com/rust-lang/rfcs/blob/master/text/1946-intra-rustdoc-links.md
55 lines
1.2 KiB
Rust
55 lines
1.2 KiB
Rust
use std::fmt;
|
|
use std::io::Result as IoResult;
|
|
use std::net::{SocketAddr, ToSocketAddrs};
|
|
use std::sync::Arc;
|
|
|
|
/// A custom resolver to override the default DNS behavior.
|
|
pub trait Resolver: Send + Sync {
|
|
fn resolve(&self, netloc: &str) -> IoResult<Vec<SocketAddr>>;
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub(crate) struct StdResolver;
|
|
|
|
impl Resolver for StdResolver {
|
|
fn resolve(&self, netloc: &str) -> IoResult<Vec<SocketAddr>> {
|
|
ToSocketAddrs::to_socket_addrs(netloc).map(|iter| iter.collect())
|
|
}
|
|
}
|
|
|
|
impl<F> Resolver for F
|
|
where
|
|
F: Fn(&str) -> IoResult<Vec<SocketAddr>>,
|
|
F: Send + Sync,
|
|
{
|
|
fn resolve(&self, netloc: &str) -> IoResult<Vec<SocketAddr>> {
|
|
self(netloc)
|
|
}
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub(crate) struct ArcResolver(Arc<dyn Resolver>);
|
|
|
|
impl<R> From<R> for ArcResolver
|
|
where
|
|
R: Resolver + 'static,
|
|
{
|
|
fn from(r: R) -> Self {
|
|
Self(Arc::new(r))
|
|
}
|
|
}
|
|
|
|
impl fmt::Debug for ArcResolver {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
write!(f, "ArcResolver(...)")
|
|
}
|
|
}
|
|
|
|
impl std::ops::Deref for ArcResolver {
|
|
type Target = dyn Resolver;
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
self.0.as_ref()
|
|
}
|
|
}
|