Use cached consumption value

This commit is contained in:
2024-09-23 20:30:34 +02:00
parent d98be00a40
commit 0e0da14fde
4 changed files with 124 additions and 12 deletions

View File

@ -6,6 +6,7 @@ use crate::devices::device::{
use crate::devices::devices_list::DevicesList;
use crate::energy::consumption;
use crate::energy::consumption::EnergyConsumption;
use crate::energy::consumption_cache::ConsumptionCache;
use crate::energy::engine::EnergyEngine;
use crate::utils::time_utils::time_secs;
use actix::prelude::*;
@ -13,34 +14,55 @@ use openssl::x509::X509Req;
use std::time::Duration;
pub struct EnergyActor {
curr_consumption: EnergyConsumption,
consumption_cache: ConsumptionCache,
devices: DevicesList,
engine: EnergyEngine,
last_engine_refresh: u64,
}
impl EnergyActor {
pub async fn new() -> anyhow::Result<Self> {
let consumption_cache_size =
AppConfig::get().refresh_interval / AppConfig::get().energy_fetch_interval;
let curr_consumption = consumption::get_curr_consumption().await?;
let mut consumption_cache = ConsumptionCache::new(consumption_cache_size as usize);
consumption_cache.add_value(curr_consumption);
if consumption_cache_size < 1 {
panic!("Energy fetch interval must be equal or smaller than refresh interval!");
}
Ok(Self {
curr_consumption: consumption::get_curr_consumption().await?,
consumption_cache,
devices: DevicesList::load()?,
engine: EnergyEngine::default(),
last_engine_refresh: 0,
})
}
async fn refresh(&mut self) -> anyhow::Result<()> {
// Refresh energy
self.curr_consumption = consumption::get_curr_consumption()
.await
.unwrap_or_else(|e| {
log::error!(
self.consumption_cache
.add_value(
consumption::get_curr_consumption()
.await
.unwrap_or_else(|e| {
log::error!(
"Failed to fetch latest consumption value, will use fallback value! {e}"
);
constants::FALLBACK_PRODUCTION_VALUE
});
constants::FALLBACK_PRODUCTION_VALUE
}),
);
if self.last_engine_refresh + AppConfig::get().refresh_interval > time_secs() {
return Ok(());
}
self.last_engine_refresh = time_secs();
let devices_list = self.devices.full_list();
self.engine.refresh(self.curr_consumption, &devices_list);
self.engine
.refresh(self.consumption_cache.median_value(), &devices_list);
self.engine.persist_relays_state(&devices_list)?;
@ -55,7 +77,7 @@ impl Actor for EnergyActor {
log::info!("Energy actor successfully started!");
ctx.run_interval(
Duration::from_secs(AppConfig::get().refresh_interval),
Duration::from_secs(AppConfig::get().energy_fetch_interval),
|act, _ctx| {
log::info!("Performing energy refresh operation");
if let Err(e) = futures::executor::block_on(act.refresh()) {
@ -81,7 +103,7 @@ impl Handler<GetCurrConsumption> for EnergyActor {
type Result = EnergyConsumption;
fn handle(&mut self, _msg: GetCurrConsumption, _ctx: &mut Context<Self>) -> Self::Result {
self.curr_consumption
self.consumption_cache.median_value()
}
}