diff --git a/src/handlers.rs b/src/handlers.rs index bd61483..10b22bc 100644 --- a/src/handlers.rs +++ b/src/handlers.rs @@ -20,17 +20,32 @@ pub struct Status { } #[derive(Serialize)] -pub struct FetchResults { +pub struct QueryResults { pub last_update: EpochTimestamp, pub refresh_time: u64, pub config: Config, pub statuses: Vec, } -type StatusState = Arc>; +pub struct State(Arc>); -pub fn index(tmpl: Data, state: Data) -> WebResult { - let state = update_state(state.lock().unwrap()); +impl State { + pub fn new(config: &Config) -> Self { + State(Arc::from(Mutex::from(QueryResults { + last_update: EpochTimestamp::now(), + refresh_time: config.refresh_time.clone(), + config: config.clone(), + statuses: vec![], + }))) + } + + fn lock(&self) -> MutexGuard { + self.0.lock().unwrap() + } +} + +pub fn index(tmpl: Data, state: Data) -> WebResult { + let state = update_state(state.lock()); let mut ctx = Context::new(); ctx.insert("results", &*state); let s = tmpl.render("index.html", &ctx).map_err(|e| { @@ -40,12 +55,12 @@ pub fn index(tmpl: Data, state: Data) -> WebResult) -> HttpResponse { - let state = update_state(state.lock().unwrap()); +pub fn json_endpoint(state: Data) -> HttpResponse { + let state = update_state(state.lock()); HttpResponse::Ok().json(&state.statuses) } -pub fn update_state(mut state: MutexGuard) -> MutexGuard { +fn update_state(mut state: MutexGuard) -> MutexGuard { if EpochTimestamp::now() - state.last_update >= state.refresh_time { state.last_update = EpochTimestamp::now(); state.statuses = update_status(&state.config); diff --git a/src/main.rs b/src/main.rs index 7aab7c9..4bcdc4d 100644 --- a/src/main.rs +++ b/src/main.rs @@ -12,14 +12,9 @@ mod utils; use self::config::*; use self::handlers::*; -use self::utils::EpochTimestamp; use actix_web::{middleware::Logger, web::resource, App, HttpServer}; use ron::de::from_str; -use std::{ - error::Error, - fs::read_to_string, - sync::{Arc, Mutex}, -}; +use std::{error::Error, fs::read_to_string}; fn main() -> Result<(), Box> { let config = from_str::(&read_to_string("./endstat_conf.ron")?)?; @@ -28,16 +23,10 @@ fn main() -> Result<(), Box> { env_logger::init(); HttpServer::new(move || { - let state = Arc::from(Mutex::from(FetchResults { - last_update: EpochTimestamp::now(), - refresh_time: config.refresh_time.clone(), - config: config.clone(), - statuses: vec![], - })); let tera = compile_templates!(concat!(env!("CARGO_MANIFEST_DIR"), "/templates/**/*")); App::new() - .data(state) + .data(State::new(&config)) .data(tera) .wrap(Logger::default()) .service(resource("/").to(index))