84 lines
2.4 KiB
Rust
84 lines
2.4 KiB
Rust
|
use crate::constants;
|
||
|
|
use crate::devices::device::{Device, DeviceId, DeviceRelayID};
|
||
|
|
use crate::energy::consumption::EnergyConsumption;
|
||
|
|
use crate::utils::time_utils::time_secs;
|
||
|
|
use prettytable::{row, Table};
|
||
|
|
use std::collections::HashMap;
|
||
|
|
|
||
|
|
#[derive(Default)]
|
||
|
|
pub struct DeviceState {
|
||
|
|
pub last_ping: u64,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl DeviceState {
|
||
|
|
pub fn record_ping(&mut self) {
|
||
|
|
self.last_ping = time_secs();
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn is_online(&self) -> bool {
|
||
|
|
(time_secs() - self.last_ping) < constants::DEVICE_MAX_PING_TIME
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Default, Clone)]
|
||
|
|
struct RelayState {
|
||
|
|
on: bool,
|
||
|
|
since: usize,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Default)]
|
||
|
|
pub struct EnergyEngine {
|
||
|
|
devices_state: HashMap<DeviceId, DeviceState>,
|
||
|
|
relays_state: HashMap<DeviceRelayID, RelayState>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl EnergyEngine {
|
||
|
|
pub fn device_state(&mut self, dev_id: &DeviceId) -> &mut DeviceState {
|
||
|
|
if !self.devices_state.contains_key(dev_id) {
|
||
|
|
self.devices_state
|
||
|
|
.insert(dev_id.clone(), Default::default());
|
||
|
|
}
|
||
|
|
|
||
|
|
self.devices_state.get_mut(dev_id).unwrap()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn relay_state(&mut self, relay_id: DeviceRelayID) -> &mut RelayState {
|
||
|
|
if !self.relays_state.contains_key(&relay_id) {
|
||
|
|
self.relays_state.insert(relay_id, Default::default());
|
||
|
|
}
|
||
|
|
|
||
|
|
self.relays_state.get_mut(&relay_id).unwrap()
|
||
|
|
}
|
||
|
|
|
||
|
|
fn print_summary(&mut self, curr_consumption: EnergyConsumption, devices: &[Device]) {
|
||
|
|
log::info!("Current consumption: {curr_consumption}");
|
||
|
|
|
||
|
|
let mut table = Table::new();
|
||
|
|
table.add_row(row!["Device", "Relay", "On", "Since"]);
|
||
|
|
for d in devices {
|
||
|
|
for r in &d.relays {
|
||
|
|
let status = self.relay_state(r.id);
|
||
|
|
table.add_row(row![d.name, r.name, status.on.to_string(), status.since]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
table.printstd();
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn refresh(&mut self, curr_consumption: EnergyConsumption, devices: &[Device]) {
|
||
|
|
let new_relays_state = self.relays_state.clone();
|
||
|
|
|
||
|
|
// Forcefully turn off relays that belongs to offline devices
|
||
|
|
|
||
|
|
// Forcefully turn off relays with dependency conflicts
|
||
|
|
|
||
|
|
// Virtually turn off all relays that can be stopped
|
||
|
|
|
||
|
|
// Turn on relays based on priority / dependencies
|
||
|
|
|
||
|
|
// Turn on relays with running constraints
|
||
|
|
|
||
|
|
// Commit changes
|
||
|
|
self.print_summary(curr_consumption, devices); // TODO :replace
|
||
|
|
}
|
||
|
|
}
|