tdns-cli 0.0.5

DNS update client (RFC 2136)
Documentation 
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
32
33
34
35
36
37
38
39
40
41
42
43
44

/// An abstraction over different ways to do DNS queries.
use std::net::SocketAddr;

use tokio_tcp::TcpStream;
use tokio_udp::UdpSocket;
use trust_dns::{
    client::{BasicClientHandle, ClientFuture, ClientHandle},
    proto::{udp::UdpResponse, xfer::dns_multiplexer::DnsMultiplexerSerialResponse},
    tcp::TcpClientStream,
    udp::UdpClientStream,
};

pub type RuntimeHandle = tokio::runtime::current_thread::Handle;

pub trait DnsOpen: Clone {
    type Client: ClientHandle;
    fn open(&mut self, runtime: RuntimeHandle, addr: SocketAddr) -> Self::Client;
}

#[derive(Debug, Clone)]
pub struct TcpOpen;

impl DnsOpen for TcpOpen {
    type Client = BasicClientHandle<DnsMultiplexerSerialResponse>;
    fn open(&mut self, runtime: RuntimeHandle, addr: SocketAddr) -> Self::Client {
        let (connect, handle) = TcpClientStream::<TcpStream>::new(addr);
        let (bg, client) = ClientFuture::new(connect, handle, None);
        runtime.spawn(bg).unwrap();
        client
    }
}

#[derive(Debug, Clone)]
pub struct UdpOpen;

impl DnsOpen for UdpOpen {
    type Client = BasicClientHandle<UdpResponse<tokio_udp::UdpSocket>>;
    fn open(&mut self, runtime: RuntimeHandle, addr: SocketAddr) -> Self::Client {
        let stream = UdpClientStream::<UdpSocket>::new(addr);
        let (bg, client) = ClientFuture::connect(stream);
        runtime.spawn(bg).unwrap();
        client
    }
}