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
|
use std::env;
use std::net::SocketAddrV4;
use igd::aio::search_gateway;
use igd::PortMappingProtocol;
use std::collections::HashMap;
struct DiplonatConfig {
private_ip: String,
consul_node_name: String,
refresh_time: u32,
expiration_time: u32
}
fn fetch_configuration() -> Result<DiplonatConfig, String> {
let env_private_ip = "DIPLONAT_PRIVATE_IP";
let private_ip = match env::var(env_private_ip) {
Ok(val) => val,
Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_private_ip, e)),
};
let env_refresh_time = "DIPLONAT_REFRESH_TIME";
let refresh_time: u32 = match env::var(env_refresh_time) {
Ok(val) => val.parse().unwrap(),
Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_refresh_time,e))
};
let env_expiration_time = "DIPLONAT_EXPIRATION_TIME";
let expiration_time: u32 = match env::var(env_expiration_time) {
Ok(val) => val.parse().unwrap(),
Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_expiration_time,e))
};
let env_consul_node_name = "DIPLONAT_CONSUL_NODE_NAME";
let consul_node_name = match env::var(env_consul_node_name) {
Ok(val) => val,
Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_consul_node_name,e))
};
if refresh_time * 2 > expiration_time {
return Err(format!("Expiration time (currently: {}s) must be twice bigger than refresh time (currently: {}s)", expiration_time, refresh_time))
}
let config = DiplonatConfig { private_ip: private_ip, refresh_time: refresh_time, expiration_time: expiration_time, consul_node_name: consul_node_name };
println!("\tConsul node name: {}", config.consul_node_name);
println!("\tPrivate IP address: {}", config.private_ip);
println!("\tRefresh time: {} seconds", config.refresh_time);
println!("\tExpiration time: {} seconds", config.expiration_time);
return Ok(config);
}
#[tokio::main]
async fn main() {
let config = match fetch_configuration() {
Ok(val) => val,
Err(e) => return println!("unable to build configuration: {}", e),
};
let resp = reqwest::get("http://127.0.0.1:8500")
.await
.unwrap()
.json::<HashMap<String, String>>()
.await.unwrap();
println!("{:#?}", resp);
let gateway = match search_gateway(Default::default()).await {
Ok(g) => g,
Err(err) => return println!("Faild to find IGD: {}", err),
};
let service = format!("{}:{}", config.private_ip, 1234);
let service: SocketAddrV4 = service.parse().expect("Invalid socket address");
match gateway.add_port(PortMappingProtocol::TCP, 1234, service, config.expiration_time, "diplonat").await {
Ok(_) => (),
Err(e) => return println!("Unable to insert port 1234: {}", e),
};
}
|
