Compare commits

...

2 Commits

Author SHA1 Message Date
Edward Shen f42c70b27e
errors are a vector instead of a single string 2019-05-03 01:05:02 -04:00
Edward Shen 17a0730275
refactor results into own file 2019-05-03 00:54:43 -04:00
5 changed files with 128 additions and 132 deletions

View File

@ -60,21 +60,23 @@
<p>{{ results.timestamp_str }}</p>
</header>
{% for group in results.groups -%}
<h2>{{ group.label }}</h2>
{% for status in group.endpoints -%}
<section>
<aside class="indicator {% if status.status == 0 %}ok{% elif status.status == 1 %}warn{% else %}error{% endif %}"></aside>
<main>
<div class="info">
<h3>{{ status.endpoint }}</h3>
<div class="spacer"></div>
{% if status.status != 2 %}<p>{{ status.rtt }}</p>{% endif %}
</div>
<a href="{{ status.location }}" target="_blank">{{ status.location }}</a>
{% if status.error %}<p class="error-msg">{{ status.error }}</p>{% endif %}
</main>
</section>
{% endfor -%}
<h2>{{ group.label }}</h2>
{% for status in group.endpoints -%}
<section>
<aside class="indicator {% if status.status == 0 %}ok{% elif status.status == 1 %}warn{% else %}error{% endif %}"></aside>
<main>
<div class="info">
<h3>{{ status.endpoint }}</h3>
<div class="spacer"></div>
{% if status.status != 2 %}<p>{{ status.rtt }}</p>{% endif %}
</div>
<a href="{{ status.location }}" target="_blank">{{ status.location }}</a>
{% for msg in status.errors -%}
<p class="error-msg">{{ msg }}</p>
{% endfor -%}
</main>
</section>
{% endfor -%}
{% endfor -%}
</body>
</html>

View File

@ -1,96 +1,10 @@
use crate::{config::*, updater::update_status, State};
use crate::State;
use actix_web::{
error::ErrorInternalServerError, web::Data, Error as WebError, HttpResponse,
Result as WebResult,
};
use chrono::prelude::*;
use serde::Serialize;
use tera::{Context, Tera};
#[derive(Clone, Serialize, Default, Debug)]
pub struct EndpointStatus {
pub status: u8,
pub location: String,
pub endpoint: String,
pub rtt: Option<String>,
pub error: Option<String>,
}
impl EndpointStatus {
pub fn ok(location: String, endpoint: String, rtt: String) -> Self {
EndpointStatus {
status: 0,
location,
endpoint,
rtt: Some(rtt),
error: None,
}
}
pub fn warn(location: String, endpoint: String, rtt: String, error: Option<String>) -> Self {
EndpointStatus {
status: 1,
location,
endpoint,
rtt: Some(rtt),
error,
}
}
pub fn error(location: String, endpoint: String, error: Option<String>) -> Self {
EndpointStatus {
status: 2,
location,
endpoint,
rtt: None,
error,
}
}
}
#[derive(Serialize, Debug)]
pub struct StatusGroup {
pub label: String,
pub endpoints: Vec<EndpointStatus>,
}
#[derive(Serialize, Debug)]
pub struct QueryResults {
pub timestamp: DateTime<Utc>,
pub timestamp_str: String,
pub refresh_time: u64,
pub config: Config,
pub groups: Vec<StatusGroup>,
}
impl QueryResults {
pub fn new(config: Config) -> Self {
let time = Utc::now();
QueryResults {
timestamp: time,
timestamp_str: Self::format_timestamp(time),
refresh_time: config.refresh_time,
config: config.clone(),
groups: update_status(&config),
}
}
pub fn update(&mut self, updated_groups: Vec<StatusGroup>) {
self.update_timestamp();
self.groups = updated_groups;
}
fn update_timestamp(&mut self) {
let current_time = Utc::now();
self.timestamp = current_time;
self.timestamp_str = Self::format_timestamp(current_time);
}
fn format_timestamp(timestamp: DateTime<Utc>) -> String {
timestamp.format("%Y-%m-%d %H:%M:%S").to_string()
}
}
pub fn index(tmpl: Data<Tera>, state: Data<State>) -> WebResult<HttpResponse, WebError> {
let state = state.read().unwrap();
let mut ctx = Context::new();

View File

@ -15,8 +15,9 @@ extern crate ring;
mod config;
mod handlers;
mod updater;
mod results;
use self::{config::*, handlers::*, updater::*};
use self::{config::*, handlers::*, updater::*, results::QueryResults};
use actix::System;
use actix_web::{middleware::Logger, web::resource, App, HttpServer};
use ron::de::from_str;

87
src/results.rs Normal file
View File

@ -0,0 +1,87 @@
use crate::{config::*, updater::update_status};
use chrono::prelude::*;
use serde::Serialize;
#[derive(Clone, Serialize, Default, Debug)]
pub struct EndpointStatus {
pub status: u8,
pub location: String,
pub endpoint: String,
pub rtt: Option<String>,
pub errors: Vec<String>,
}
impl EndpointStatus {
pub fn ok(location: String, endpoint: String, rtt: String) -> Self {
EndpointStatus {
status: 0,
location,
endpoint,
rtt: Some(rtt),
errors: vec![],
}
}
pub fn warn(location: String, endpoint: String, rtt: String, errors: Vec<String>) -> Self {
EndpointStatus {
status: 1,
location,
endpoint,
rtt: Some(rtt),
errors,
}
}
pub fn error(location: String, endpoint: String, errors: Vec<String>) -> Self {
EndpointStatus {
status: 2,
location,
endpoint,
rtt: None,
errors,
}
}
}
#[derive(Serialize, Debug)]
pub struct StatusGroup {
pub label: String,
pub endpoints: Vec<EndpointStatus>,
}
#[derive(Serialize, Debug)]
pub struct QueryResults {
pub timestamp: DateTime<Utc>,
pub timestamp_str: String,
pub refresh_time: u64,
pub config: Config,
pub groups: Vec<StatusGroup>,
}
impl QueryResults {
pub fn new(config: Config) -> Self {
let time = Utc::now();
QueryResults {
timestamp: time,
timestamp_str: Self::format_timestamp(time),
refresh_time: config.refresh_time,
config: config.clone(),
groups: update_status(&config),
}
}
pub fn update(&mut self, updated_groups: Vec<StatusGroup>) {
self.update_timestamp();
self.groups = updated_groups;
}
fn update_timestamp(&mut self) {
let current_time = Utc::now();
self.timestamp = current_time;
self.timestamp_str = Self::format_timestamp(current_time);
}
fn format_timestamp(timestamp: DateTime<Utc>) -> String {
timestamp.format("%Y-%m-%d %H:%M:%S").to_string()
}
}

View File

@ -1,6 +1,6 @@
use crate::{
config::*,
handlers::{EndpointStatus, StatusGroup},
results::{EndpointStatus, StatusGroup},
State,
};
use chrono::Utc;
@ -62,43 +62,42 @@ fn get_result(
match ping_result {
Ok(mut res) => {
let res_body = res.text().expect("could not get body of request");
let mut error = None;
let mut errors = vec![];
if res.status() != code {
error = append_err_msg(
error,
format!(
"Status code mismatch: {} != {}",
res.status().as_u16(),
code
),
);
errors.push(format!(
"Status code mismatch: {} != {}",
res.status().as_u16(),
code
));
}
if let Some(expected_hash) = &endpoint.body_hash {
let expected = from_hex(expected_hash).unwrap();
let actual = digest(&SHA256, String::from(res_body).as_bytes());
if &expected != &actual.as_ref() {
error = append_err_msg(error, String::from("Body hash mismatch."));
errors.push(String::from("Body hash mismatch."));
}
} else if !body.is_empty() && res_body != body {
error = append_err_msg(
error,
format!("Body mismatch: {} != {}", res_body.len(), body.len()),
);
errors.push(format!(
"Body mismatch: {} != {}",
res_body.len(),
body.len()
));
}
if let Some(max_rtt) = endpoint.max_rtt {
if rtt.num_milliseconds() > max_rtt {
error = append_err_msg(
error,
format!("RTT too long: {} > {}s", rtt_string, max_rtt as f64 / 1000.),
);
errors.push(format!(
"RTT too long: {} > {}s",
rtt_string,
max_rtt as f64 / 1000.
));
}
}
if error.is_some() {
EndpointStatus::warn(url, label, rtt_string, error)
if !errors.is_empty() {
EndpointStatus::warn(url, label, rtt_string, errors)
} else {
EndpointStatus::ok(url, label, rtt_string)
}
@ -107,7 +106,7 @@ fn get_result(
if let Some(true) = endpoint.should_err {
EndpointStatus::ok(url, label, rtt_string)
} else {
EndpointStatus::error(url, label, Some(format!("{}", e)))
EndpointStatus::error(url, label, vec![format!("{}", e)])
}
}
}
@ -124,10 +123,3 @@ fn get_url(base: &Option<String>, path: &String, port: Option<u16>) -> Result<St
}
Ok(url.into_string())
}
fn append_err_msg(optional: Option<String>, to_append: String) -> Option<String> {
Some(match optional {
Some(e) => format!("{}\n{}", e, to_append),
None => to_append,
})
}