ns-router 0.1.6

A configurable name resolver for abstract-ns
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

extern crate abstract_ns;
extern crate futures;
extern crate ns_router;
extern crate tokio_core;

use std::net::{IpAddr};
use std::time::Duration;
use std::sync::atomic::{AtomicUsize, Ordering};

use futures::{Stream, lazy};
use futures::future::{FutureResult, ok};
use abstract_ns::{HostSubscribe, Subscribe, Name, Address, IpList, Error};
use abstract_ns::{Resolve, HostResolve};
use ns_router::{Config, Router, SubscribeExt};

#[derive(Debug)]
struct IncrMock(AtomicUsize);

impl HostResolve for IncrMock {
    type HostFuture = FutureResult<IpList, Error>;
    fn resolve_host(&self, _name: &Name) -> Self::HostFuture {
        let n = self.0.fetch_add(1, Ordering::SeqCst);
        ok(vec![format!("127.0.0.{}", n).parse().unwrap()].into())
    }
}

impl Resolve for IncrMock {
    type Future = FutureResult<Address, Error>;
    fn resolve(&self, _name: &Name) -> Self::Future {
        let n = self.0.fetch_add(1, Ordering::SeqCst);
        ok([format!("127.0.0.{}:443", n).parse().unwrap()][..].into())
    }
}

#[test]
fn test_host() {
    let mut core = tokio_core::reactor::Core::new().unwrap();
    let handle = core.handle();

    let cfg = Config::new()
        .set_fallthrough(IncrMock(AtomicUsize::new(1))
            .interval_subscriber(Duration::from_millis(100), &handle))
        .done();
    let router = Router::from_config(&cfg, &handle);

    let res = core.run(lazy(|| {
        router.subscribe_host(&"localhost".parse().unwrap()).into_future()
    })).unwrap();
    assert_eq!(res.0,
        Some(vec!["127.0.0.1".parse::<IpAddr>().unwrap()].into()));

    let res = core.run(lazy(move || res.1.into_future())).unwrap();
    assert_eq!(res.0,
        Some(vec!["127.0.0.2".parse::<IpAddr>().unwrap()].into()));
}

#[test]
fn test_service() {
    let mut core = tokio_core::reactor::Core::new().unwrap();
    let handle = core.handle();

    let cfg = Config::new()
        .set_fallthrough(IncrMock(AtomicUsize::new(1))
            .interval_subscriber(Duration::from_millis(100), &handle))
        .done();
    let router = Router::from_config(&cfg, &handle);

    let res = core.run(lazy(|| {
        router.subscribe(&"localhost".parse().unwrap()).into_future()
    })).unwrap();
    assert_eq!(res.0,
        Some(Address::parse_list(&["127.0.0.1:443"]).unwrap()));

    println!("First time");

    let res = core.run(lazy(move || res.1.into_future())).unwrap();
    assert_eq!(res.0,
        Some(Address::parse_list(&["127.0.0.2:443"]).unwrap()));
}