deboa-tokio 0.1.0

A friendly rest client on top of hyper.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use deboa::{
    dns::{DnsResolver, DnsResolverFuture},
    errors::{DeboaError::Dns, DnsError},
};
use rand::seq::SliceRandom;
use std::net::IpAddr;
use tokio::net::lookup_host;

/// Default DNS resolver implementation using tokio::net::lookup_host
#[derive(Default, Clone)]
pub struct DefaultDnsResolver;

impl DnsResolver for DefaultDnsResolver {
    fn resolve(&self, host: String, port: u16) -> DnsResolverFuture {
        let future = async move {
            let hostname = format!("{}:{}", host, port);
            let addrs = lookup_host(hostname).await;
            if let Err(e) = addrs {
                return Err(Dns(DnsError::Resolve { host, message: e.to_string() }));
            }

            let mut ips: Vec<IpAddr> = addrs
                .unwrap()
                .map(|addr| addr.ip())
                .collect();
            ips.shuffle(&mut rand::rng());
            Ok(ips)
        };
        Box::pin(future)
    }
}