changed branding to nws
This commit is contained in:
parent
cec9ceff7b
commit
061e74bc1a
GPG key ID:
838827D8C4611687
11 changed files with 286 additions and 225 deletions
|
|
@ -1,12 +1,12 @@
|
|||
use std::collections::HashMap;
|
||||
use tokio::time::{sleep};
|
||||
use std::sync::{Arc, Mutex};
|
||||
use std::time::{Duration};
|
||||
use chrono::{Datelike, NaiveDate};
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
use anyhow::anyhow;
|
||||
use anyhow::Context;
|
||||
use anyhow::{anyhow};
|
||||
use chrono::{Datelike, NaiveDate};
|
||||
use log::*;
|
||||
use std::collections::HashMap;
|
||||
use std::sync::{Arc, Mutex};
|
||||
use std::time::Duration;
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
use tokio::time::sleep;
|
||||
|
||||
#[macro_use]
|
||||
use dotenv::dotenv;
|
||||
|
|
@ -17,7 +17,7 @@ pub enum UptimeType {
|
|||
Provider,
|
||||
Service,
|
||||
Datacenter,
|
||||
Unknown
|
||||
Unknown,
|
||||
}
|
||||
|
||||
#[derive(Debug, PartialEq, Clone)]
|
||||
|
|
@ -25,7 +25,7 @@ pub enum UptimeStatus {
|
|||
Up,
|
||||
Down,
|
||||
Maintenance,
|
||||
Unknown
|
||||
Unknown,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
|
|
@ -34,28 +34,29 @@ pub struct Uptime {
|
|||
pub uptime: String,
|
||||
pub response_time: String,
|
||||
pub status: UptimeStatus,
|
||||
pub uptime_type: UptimeType,
|
||||
pub url: String
|
||||
pub uptime_type: UptimeType,
|
||||
pub url: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct UptimeServiceState {
|
||||
uptimes: Vec<Uptime>,
|
||||
last_updated: SystemTime
|
||||
last_updated: SystemTime,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct UptimeService {
|
||||
state: Arc<Mutex<UptimeServiceState>>
|
||||
state: Arc<Mutex<UptimeServiceState>>,
|
||||
}
|
||||
|
||||
impl UptimeService {
|
||||
const UPDATE_SECONDS: u64 = 300;
|
||||
|
||||
pub fn new() -> Self {
|
||||
let init_state = Arc::new(Mutex::new(
|
||||
UptimeServiceState { uptimes: vec![], last_updated: UNIX_EPOCH }
|
||||
));
|
||||
let init_state = Arc::new(Mutex::new(UptimeServiceState {
|
||||
uptimes: vec![],
|
||||
last_updated: UNIX_EPOCH,
|
||||
}));
|
||||
Self { state: init_state }
|
||||
}
|
||||
|
||||
|
|
@ -63,18 +64,18 @@ impl UptimeService {
|
|||
info!("Starting UptimeService");
|
||||
let cloned_state = Arc::clone(&self.state);
|
||||
tokio::spawn(async move {
|
||||
loop {
|
||||
let clonedx_state = Arc::clone(&cloned_state);
|
||||
let res = Self::update_data(clonedx_state).await;
|
||||
match res {
|
||||
Err(err) => {
|
||||
error!("{}", err);
|
||||
},
|
||||
_ => {}
|
||||
loop {
|
||||
let clonedx_state = Arc::clone(&cloned_state);
|
||||
let res = Self::update_data(clonedx_state).await;
|
||||
match res {
|
||||
Err(err) => {
|
||||
error!("{}", err);
|
||||
}
|
||||
sleep(tokio::time::Duration::from_secs(Self::UPDATE_SECONDS)).await;
|
||||
_ => {}
|
||||
}
|
||||
});
|
||||
sleep(tokio::time::Duration::from_secs(Self::UPDATE_SECONDS)).await;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
pub fn get_data(&self) -> Vec<Uptime> {
|
||||
|
|
@ -90,7 +91,6 @@ impl UptimeService {
|
|||
}
|
||||
|
||||
async fn update_data(arc_state: Arc<Mutex<UptimeServiceState>>) -> ::anyhow::Result<()> {
|
||||
|
||||
debug!("Starting data update for UptimeService");
|
||||
|
||||
let mut request_vars = HashMap::new();
|
||||
|
|
@ -102,9 +102,10 @@ impl UptimeService {
|
|||
|
||||
let current_year = chrono::Utc::today().year();
|
||||
let january_1st = NaiveDate::from_ymd(current_year, 1, 1).and_hms(0, 0, 0);
|
||||
let duration = january_1st.signed_duration_since(NaiveDate::from_ymd(1970, 1, 1).and_hms(0, 0, 0));
|
||||
let duration =
|
||||
january_1st.signed_duration_since(NaiveDate::from_ymd(1970, 1, 1).and_hms(0, 0, 0));
|
||||
let year_start = UNIX_EPOCH + Duration::from_secs(duration.num_seconds() as u64);
|
||||
|
||||
|
||||
//let ranges = &format!(
|
||||
// "{}_{}-{}_{}",
|
||||
// thirty_days_ago.duration_since(SystemTime::UNIX_EPOCH)?.as_secs(),
|
||||
|
|
@ -122,107 +123,124 @@ impl UptimeService {
|
|||
request_vars.insert("response_times", "1");
|
||||
|
||||
let client = reqwest::Client::new();
|
||||
let res = client.post("https://api.uptimerobot.com/v2/getMonitors")
|
||||
let res = client
|
||||
.post("https://api.uptimerobot.com/v2/getMonitors")
|
||||
.form(&request_vars)
|
||||
.send()
|
||||
.await?;
|
||||
|
||||
let resp = res.json::<serde_json::Value>()
|
||||
.await?;
|
||||
let resp = res.json::<serde_json::Value>().await?;
|
||||
|
||||
let monitors = resp.get("monitors")
|
||||
.context("Response did not have a monitors subobject")?
|
||||
.as_array()
|
||||
.context("Monitors subobject was not an array")?;
|
||||
let monitors = resp
|
||||
.get("monitors")
|
||||
.context("Response did not have a monitors subobject")?
|
||||
.as_array()
|
||||
.context("Monitors subobject was not an array")?;
|
||||
|
||||
|
||||
let mut state = match arc_state.lock(){
|
||||
let mut state = match arc_state.lock() {
|
||||
Ok(val) => val,
|
||||
Err(_) => {return Err(anyhow!("Could not lock shared state"));}
|
||||
Err(_) => {
|
||||
return Err(anyhow!("Could not lock shared state"));
|
||||
}
|
||||
};
|
||||
state.uptimes.clear();
|
||||
for monitor in monitors {
|
||||
let monitor_fqn = monitor.get("friendly_name")
|
||||
.context("Monitor did not have property 'friendly_name'")?;
|
||||
|
||||
let monitor_fqn = monitor
|
||||
.get("friendly_name")
|
||||
.context("Monitor did not have property 'friendly_name'")?;
|
||||
|
||||
debug!("Monitor '{}' processing", monitor_fqn);
|
||||
|
||||
let split_str: Vec<&str> = monitor_fqn.as_str()
|
||||
.context("Expected 'friendly_name' to be a string")?
|
||||
.split(".").collect();
|
||||
let split_str: Vec<&str> = monitor_fqn
|
||||
.as_str()
|
||||
.context("Expected 'friendly_name' to be a string")?
|
||||
.split(".")
|
||||
.collect();
|
||||
if split_str.len() != 2 {
|
||||
debug!("Monitor '{}' excluded due to bad format", monitor_fqn);
|
||||
continue;
|
||||
}
|
||||
|
||||
let monitor_nt = String::from(*split_str.get(0).context("Expected name to have first part")?);
|
||||
let monitor_name = String::from(*split_str.get(1).context("Expected name to have second part")?);
|
||||
let monitor_nt = String::from(
|
||||
*split_str
|
||||
.get(0)
|
||||
.context("Expected name to have first part")?,
|
||||
);
|
||||
let monitor_name = String::from(
|
||||
*split_str
|
||||
.get(1)
|
||||
.context("Expected name to have second part")?,
|
||||
);
|
||||
let monitor_type = match monitor_nt.as_str() {
|
||||
"datacenter" => UptimeType::Datacenter,
|
||||
"service" => UptimeType::Service,
|
||||
"competitor" => UptimeType::Provider,
|
||||
_ => UptimeType::Unknown
|
||||
_ => UptimeType::Unknown,
|
||||
};
|
||||
|
||||
|
||||
if monitor_type == UptimeType::Unknown {
|
||||
debug!("Monitor '{}' excluded due to unknown type", monitor_fqn);
|
||||
continue;
|
||||
}
|
||||
|
||||
let monitor_status_num = monitor.get("status")
|
||||
.context("Expected monitor to have 'status' property")?
|
||||
.as_u64()
|
||||
.context("Expected 'status' property to be u64")?;
|
||||
let monitor_status_num = monitor
|
||||
.get("status")
|
||||
.context("Expected monitor to have 'status' property")?
|
||||
.as_u64()
|
||||
.context("Expected 'status' property to be u64")?;
|
||||
|
||||
let monitor_status = match monitor_status_num {
|
||||
0 => UptimeStatus::Maintenance,
|
||||
1 | 8 | 9 => UptimeStatus::Down,
|
||||
2 => UptimeStatus::Up,
|
||||
_ => UptimeStatus::Unknown
|
||||
_ => UptimeStatus::Unknown,
|
||||
};
|
||||
|
||||
if monitor_status == UptimeStatus::Unknown {
|
||||
debug!("Monitor '{}' excluded due to unknown status (status was {})", monitor_fqn, monitor_status_num);
|
||||
debug!(
|
||||
"Monitor '{}' excluded due to unknown status (status was {})",
|
||||
monitor_fqn, monitor_status_num
|
||||
);
|
||||
continue;
|
||||
}
|
||||
|
||||
let monitor_rt_val = monitor.get("average_response_time")
|
||||
.context("Expected monitor to have property 'average_response_time'")?;
|
||||
|
||||
let monitor_rt_val = monitor
|
||||
.get("average_response_time")
|
||||
.context("Expected monitor to have property 'average_response_time'")?;
|
||||
|
||||
// Because UptimeRobot has the world's worst API ever
|
||||
// and decided that it's okay to return multiple datatypes
|
||||
// and decided that it's okay to return multiple datatypes
|
||||
// for one property based on how they're feeling
|
||||
let monitor_rt = match monitor_rt_val.as_str() {
|
||||
Some(string) => format!("{}ms", string),
|
||||
_ => format!("N/A")
|
||||
_ => format!("N/A"),
|
||||
};
|
||||
|
||||
let monitor_uptime = format!("{}%",
|
||||
monitor.get("custom_uptime_ranges")
|
||||
.context("Expected monitor to have property 'custom_uptime_ranges'")?
|
||||
.as_str()
|
||||
.context("Expected 'custom_uptime_ranges' to be String")?
|
||||
);
|
||||
let monitor_uptime = format!(
|
||||
"{}%",
|
||||
monitor
|
||||
.get("custom_uptime_ranges")
|
||||
.context("Expected monitor to have property 'custom_uptime_ranges'")?
|
||||
.as_str()
|
||||
.context("Expected 'custom_uptime_ranges' to be String")?
|
||||
);
|
||||
|
||||
let monitor_url = String::from(monitor.get("url")
|
||||
.context("Expected monitor to have property 'url'")?
|
||||
.as_str()
|
||||
.context("Expected 'url' to be String")?);
|
||||
let monitor_url = String::from(
|
||||
monitor
|
||||
.get("url")
|
||||
.context("Expected monitor to have property 'url'")?
|
||||
.as_str()
|
||||
.context("Expected 'url' to be String")?,
|
||||
);
|
||||
|
||||
;
|
||||
|
||||
state.uptimes.push(
|
||||
Uptime {
|
||||
name: monitor_name,
|
||||
uptime: monitor_uptime,
|
||||
response_time: monitor_rt,
|
||||
status: monitor_status,
|
||||
uptime_type: monitor_type,
|
||||
url: monitor_url
|
||||
}
|
||||
);
|
||||
state.uptimes.push(Uptime {
|
||||
name: monitor_name,
|
||||
uptime: monitor_uptime,
|
||||
response_time: monitor_rt,
|
||||
status: monitor_status,
|
||||
uptime_type: monitor_type,
|
||||
url: monitor_url,
|
||||
});
|
||||
}
|
||||
|
||||
state.last_updated = SystemTime::now();
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue