Browse Source

WIP consul actor

main
Quentin 1 year ago
parent
commit
efbbc17f82
  1. 2
      README.md
  2. 44
      src/consul_actor.rs
  3. 1
      src/main.rs
  4. 4
      src/messages.rs

2
README.md

@ -59,6 +59,6 @@ To test the Consul Catalog part, you can do:
```bash
consul agent -dev #in a separate terminal, if not already running
consul services register -name=example -port=1337
consul services register -name=example -port=1337 -tag="diplonat_pub_port=1337"
consul services -id=example
```

44
src/consul_actor.rs

@ -1,15 +1,51 @@
use crate::consul::Consul;
use tokio::sync::watch;
use tokio::time::delay_for;
use crate::messages;
use anyhow::Result;
use std::cmp;
use std::time::Duration;
use log::*;
pub struct ConsulActor {
consul: Consul
out:
pub rx_open_ports: watch::Receiver<messages::OpenPorts>,
consul: Consul,
node: String,
retries: u32,
tx_open_ports: watch::Sender<messages::OpenPorts>
}
fn retry_to_time(retries: u32, max_time: Duration) -> Duration {
// 1.2^x seems to be a good value to exponentially increase time at a good pace
// eg. 1.2^32 = 341 seconds ~= 5 minutes - ie. after 32 retries we wait 5 minutes
return Duration::from_secs(cmp::max(max_time.as_secs(), 1.2f64.powf(retries as f64) as u64))
}
impl ConsulActor {
fn new(url: &str) -> Self {
fn new(url: &str, node: &str) -> Self {
let (tx, rx) = watch::channel(messages::OpenPorts{ports: Vec::new() });
return Self {
consul: Consul::new(url);
consul: Consul::new(url),
rx_open_ports: rx,
tx_open_ports: tx,
node: node.to_string(),
retries: 0,
};
}
async fn listen(&mut self) -> Result<()> {
loop {
let catalog = match self.consul.watch_node(&self.node).await {
Ok(c) => c,
Err(e) => {
self.retries = cmp::min(u32::MAX - 1, self.retries) + 1;
let will_retry_in = retry_to_time(self.retries, Duration::from_secs(600));
error!("Failed to query consul. Will retry in {}s. {}", will_retry_in.as_secs(), e);
delay_for(will_retry_in).await;
continue;
}
};
}
}
}

1
src/main.rs

@ -1,3 +1,4 @@
mod messages;
mod consul;
mod consul_actor;

4
src/messages.rs

@ -0,0 +1,4 @@
#[derive(Debug, Clone)]
pub struct OpenPorts {
pub ports: Vec<u16>
}
Loading…
Cancel
Save