88 lines
2.3 KiB
Rust
88 lines
2.3 KiB
Rust
use std::time::Duration;
|
|
|
|
use actix::Addr;
|
|
use chrono::Local;
|
|
use emgauwa_lib::constants::RELAYS_RETRY_TIMEOUT;
|
|
use emgauwa_lib::errors::EmgauwaError;
|
|
use emgauwa_lib::models::Controller;
|
|
use futures::pin_mut;
|
|
use tokio::time;
|
|
use tokio::time::timeout;
|
|
use utils::app_state_get_controller_notifier;
|
|
|
|
use crate::app_state::AppState;
|
|
use crate::utils;
|
|
|
|
pub async fn run_relays_loop(app_state: Addr<AppState>) {
|
|
log::debug!("Spawned relays loop");
|
|
loop {
|
|
let run_result = run_relays(&app_state).await;
|
|
if let Err(err) = run_result {
|
|
log::error!("Error running relays: {}", err);
|
|
}
|
|
time::sleep(RELAYS_RETRY_TIMEOUT).await;
|
|
}
|
|
}
|
|
|
|
async fn run_relays(app_state: &Addr<AppState>) -> Result<(), EmgauwaError> {
|
|
let default_duration = Duration::new(10, 0);
|
|
let notifier = &*app_state_get_controller_notifier(app_state).await?;
|
|
|
|
let mut last_weekday = emgauwa_lib::utils::get_weekday();
|
|
let mut this = utils::app_state_get_this(app_state).await?;
|
|
let mut relay_states: Vec<Option<bool>> = Vec::new();
|
|
init_relay_states(&mut relay_states, &this);
|
|
|
|
loop {
|
|
let notifier_future = notifier.notified();
|
|
pin_mut!(notifier_future);
|
|
let timeout_result = timeout(default_duration, &mut notifier_future).await;
|
|
let mut changed = timeout_result.is_ok();
|
|
|
|
let current_weekday = emgauwa_lib::utils::get_weekday();
|
|
if current_weekday != last_weekday {
|
|
log::debug!("Weekday changed");
|
|
last_weekday = current_weekday;
|
|
utils::app_state_reload(app_state).await?;
|
|
changed = true;
|
|
}
|
|
|
|
if changed {
|
|
log::debug!("Reloading controller in relay loop");
|
|
this = utils::app_state_get_this(app_state).await?;
|
|
}
|
|
|
|
let mut relay_debug = String::new();
|
|
let now = Local::now().time();
|
|
this.relays
|
|
.iter()
|
|
.zip(relay_states.iter_mut())
|
|
.for_each(|(relay, state)| {
|
|
*state = Some(relay.active_schedule.is_on(&now));
|
|
|
|
relay_debug.push_str(&format!(
|
|
"{}{}: {} ; ",
|
|
if relay.active_schedule.is_on(&now) {
|
|
"+"
|
|
} else {
|
|
"-"
|
|
},
|
|
relay.r.name,
|
|
relay.active_schedule.name
|
|
));
|
|
});
|
|
log::debug!(
|
|
"Relay loop at {}: {}",
|
|
Local::now().naive_local().time(),
|
|
relay_debug
|
|
);
|
|
utils::app_state_update_relays_on(app_state, relay_states.clone()).await?;
|
|
}
|
|
}
|
|
|
|
fn init_relay_states(relay_states: &mut Vec<Option<bool>>, this: &Controller) {
|
|
relay_states.clear();
|
|
for _ in 0..this.c.relay_count {
|
|
relay_states.push(None);
|
|
}
|
|
}
|