Rename STUN actor to autodiscovery actor
This commit is contained in:
parent
6147bb0b8c
commit
6614dfaeb6
5 changed files with 28 additions and 21 deletions
|
@ -5,7 +5,7 @@ use anyhow::{anyhow, bail, Result};
|
||||||
use log::*;
|
use log::*;
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
use crate::config::{RuntimeConfigConsul, RuntimeConfigStun};
|
use crate::config::{RuntimeConfigAutoDiscovery, RuntimeConfigConsul};
|
||||||
use crate::consul;
|
use crate::consul;
|
||||||
|
|
||||||
/// If autodiscovery returns None but an address was obtained less than
|
/// If autodiscovery returns None but an address was obtained less than
|
||||||
|
@ -13,7 +13,7 @@ use crate::consul;
|
||||||
/// in the Consul db instead of insterting a None.
|
/// in the Consul db instead of insterting a None.
|
||||||
const PERSIST_SOME_RESULT_DURATION_SECS: u64 = 900;
|
const PERSIST_SOME_RESULT_DURATION_SECS: u64 = 900;
|
||||||
|
|
||||||
pub struct StunActor {
|
pub struct AutoDiscoveryActor {
|
||||||
consul: consul::Consul,
|
consul: consul::Consul,
|
||||||
refresh_time: Duration,
|
refresh_time: Duration,
|
||||||
|
|
||||||
|
@ -39,22 +39,22 @@ pub struct AutodiscoverResult {
|
||||||
pub address: Option<IpAddr>,
|
pub address: Option<IpAddr>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl StunActor {
|
impl AutoDiscoveryActor {
|
||||||
pub fn new(
|
pub fn new(
|
||||||
consul_config: &RuntimeConfigConsul,
|
consul_config: &RuntimeConfigConsul,
|
||||||
stun_config: &RuntimeConfigStun,
|
autodiscovery_config: &RuntimeConfigAutoDiscovery,
|
||||||
node: &str,
|
node: &str,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
assert!(stun_config
|
assert!(autodiscovery_config
|
||||||
.stun_server_v4
|
.stun_server_v4
|
||||||
.map(|x| x.is_ipv4())
|
.map(|x| x.is_ipv4())
|
||||||
.unwrap_or(true));
|
.unwrap_or(true));
|
||||||
assert!(stun_config.stun_server_v6.is_ipv6());
|
assert!(autodiscovery_config.stun_server_v6.is_ipv6());
|
||||||
|
|
||||||
let autodiscovery_v4 = StunAutodiscovery {
|
let autodiscovery_v4 = StunAutodiscovery {
|
||||||
consul_key: format!("diplonat/autodiscovery/ipv4/{}", node),
|
consul_key: format!("diplonat/autodiscovery/ipv4/{}", node),
|
||||||
is_ipv4: true,
|
is_ipv4: true,
|
||||||
stun_server: stun_config.stun_server_v4,
|
stun_server: autodiscovery_config.stun_server_v4,
|
||||||
last_result: None,
|
last_result: None,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -67,7 +67,7 @@ impl StunActor {
|
||||||
consul: consul::Consul::new(consul_config),
|
consul: consul::Consul::new(consul_config),
|
||||||
autodiscovery_v4,
|
autodiscovery_v4,
|
||||||
autodiscovery_v6,
|
autodiscovery_v6,
|
||||||
refresh_time: stun_config.refresh_time,
|
refresh_time: autodiscovery_config.refresh_time,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,7 +5,8 @@ mod runtime;
|
||||||
|
|
||||||
pub use options::{ConfigOpts, ConfigOptsBase, ConfigOptsConsul};
|
pub use options::{ConfigOpts, ConfigOptsBase, ConfigOptsConsul};
|
||||||
pub use runtime::{
|
pub use runtime::{
|
||||||
RuntimeConfig, RuntimeConfigConsul, RuntimeConfigFirewall, RuntimeConfigIgd, RuntimeConfigStun,
|
RuntimeConfig, RuntimeConfigAutoDiscovery, RuntimeConfigConsul, RuntimeConfigFirewall,
|
||||||
|
RuntimeConfigIgd,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub const EXPIRATION_TIME: u16 = 300;
|
pub const EXPIRATION_TIME: u16 = 300;
|
||||||
|
|
|
@ -33,7 +33,7 @@ pub struct RuntimeConfigIgd {
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct RuntimeConfigStun {
|
pub struct RuntimeConfigAutoDiscovery {
|
||||||
pub stun_server_v4: Option<SocketAddr>,
|
pub stun_server_v4: Option<SocketAddr>,
|
||||||
pub stun_server_v6: SocketAddr,
|
pub stun_server_v6: SocketAddr,
|
||||||
pub refresh_time: Duration,
|
pub refresh_time: Duration,
|
||||||
|
@ -44,7 +44,7 @@ pub struct RuntimeConfig {
|
||||||
pub consul: RuntimeConfigConsul,
|
pub consul: RuntimeConfigConsul,
|
||||||
pub firewall: RuntimeConfigFirewall,
|
pub firewall: RuntimeConfigFirewall,
|
||||||
pub igd: Option<RuntimeConfigIgd>,
|
pub igd: Option<RuntimeConfigIgd>,
|
||||||
pub stun: RuntimeConfigStun,
|
pub autodiscovery: RuntimeConfigAutoDiscovery,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl RuntimeConfig {
|
impl RuntimeConfig {
|
||||||
|
@ -55,13 +55,13 @@ impl RuntimeConfig {
|
||||||
false => Some(RuntimeConfigIgd::new(&opts.base)?),
|
false => Some(RuntimeConfigIgd::new(&opts.base)?),
|
||||||
true => None,
|
true => None,
|
||||||
};
|
};
|
||||||
let stun = RuntimeConfigStun::new(&opts.base)?;
|
let autodiscovery = RuntimeConfigAutoDiscovery::new(&opts.base)?;
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
consul,
|
consul,
|
||||||
firewall,
|
firewall,
|
||||||
igd,
|
igd,
|
||||||
stun,
|
autodiscovery,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -153,7 +153,7 @@ impl RuntimeConfigIgd {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl RuntimeConfigStun {
|
impl RuntimeConfigAutoDiscovery {
|
||||||
pub(super) fn new(opts: &ConfigOptsBase) -> Result<Self> {
|
pub(super) fn new(opts: &ConfigOptsBase) -> Result<Self> {
|
||||||
let mut stun_server_v4 = None;
|
let mut stun_server_v4 = None;
|
||||||
let mut stun_server_v6 = None;
|
let mut stun_server_v6 = None;
|
||||||
|
|
|
@ -3,15 +3,15 @@ use futures::future::FutureExt;
|
||||||
use tokio::try_join;
|
use tokio::try_join;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
config::ConfigOpts, consul_actor::ConsulActor, fw_actor::FirewallActor, igd_actor::IgdActor,
|
autodiscovery_actor::AutoDiscoveryActor, config::ConfigOpts, consul_actor::ConsulActor,
|
||||||
stun_actor::StunActor,
|
fw_actor::FirewallActor, igd_actor::IgdActor,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub struct Diplonat {
|
pub struct Diplonat {
|
||||||
consul: ConsulActor,
|
consul: ConsulActor,
|
||||||
firewall: FirewallActor,
|
firewall: FirewallActor,
|
||||||
igd: Option<IgdActor>,
|
igd: Option<IgdActor>,
|
||||||
stun: StunActor,
|
autodiscovery: AutoDiscoveryActor,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Diplonat {
|
impl Diplonat {
|
||||||
|
@ -43,13 +43,17 @@ impl Diplonat {
|
||||||
None => None,
|
None => None,
|
||||||
};
|
};
|
||||||
|
|
||||||
let sa = StunActor::new(&rt_cfg.consul, &rt_cfg.stun, &rt_cfg.consul.node_name);
|
let ad = AutoDiscoveryActor::new(
|
||||||
|
&rt_cfg.consul,
|
||||||
|
&rt_cfg.autodiscovery,
|
||||||
|
&rt_cfg.consul.node_name,
|
||||||
|
);
|
||||||
|
|
||||||
let ctx = Self {
|
let ctx = Self {
|
||||||
consul: ca,
|
consul: ca,
|
||||||
igd: ia,
|
igd: ia,
|
||||||
firewall: fw,
|
firewall: fw,
|
||||||
stun: sa,
|
autodiscovery: ad,
|
||||||
};
|
};
|
||||||
|
|
||||||
Ok(ctx)
|
Ok(ctx)
|
||||||
|
@ -70,7 +74,9 @@ impl Diplonat {
|
||||||
self.firewall
|
self.firewall
|
||||||
.listen()
|
.listen()
|
||||||
.map(|x| x.context("Run firewall actor")),
|
.map(|x| x.context("Run firewall actor")),
|
||||||
self.stun.listen().map(|x| x.context("Run STUN actor")),
|
self.autodiscovery
|
||||||
|
.listen()
|
||||||
|
.map(|x| x.context("Run autodiscovery actor")),
|
||||||
)?;
|
)?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
mod autodiscovery_actor;
|
||||||
mod config;
|
mod config;
|
||||||
mod consul;
|
mod consul;
|
||||||
mod consul_actor;
|
mod consul_actor;
|
||||||
|
@ -6,7 +7,6 @@ mod fw;
|
||||||
mod fw_actor;
|
mod fw_actor;
|
||||||
mod igd_actor;
|
mod igd_actor;
|
||||||
mod messages;
|
mod messages;
|
||||||
mod stun_actor;
|
|
||||||
|
|
||||||
use diplonat::Diplonat;
|
use diplonat::Diplonat;
|
||||||
use log::*;
|
use log::*;
|
||||||
|
|
Loading…
Add table
Reference in a new issue