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
use std::env;
use failure::Error;
use futures_cpupool::CpuPool;
use tokio_core::reactor::Handle;
use ns_std_threaded::ThreadedResolver;
use Router;
use router;
use config::Config;
use ns::HostResolve;
use router::subscribe_ext::SubscribeExt;
use void::unreachable;
pub fn init(h: &Handle) -> Result<Router, Error> {
init_default(h, &Config::default())
}
pub fn init_default(h: &Handle, config: &Config) -> Result<Router, Error> {
if let Ok(val) = env::var("RUST_NS") {
if val.len() > 0 {
if let Ok(cfg) = val.parse() {
return force_config(h, &cfg)
}
}
}
force_config(h, config)
}
pub fn force_config(h: &Handle, config: &Config) -> Result<Router, Error> {
use config::Route;
let mut cfg = router::Config::new();
match config.fallthrough {
Route::Std(ref std) => {
let pool = CpuPool::new(std.threads as usize);
cfg.set_fallthrough(ThreadedResolver::use_pool(pool)
.null_service_resolver()
.interval_subscriber(std.poll_ivl, &h));
}
Route::__Nonexhaustive(void) => unreachable(void),
}
let ns = Router::from_config(&cfg.done(), &h);
Ok(ns)
}