65 lines
1.8 KiB
Rust
65 lines
1.8 KiB
Rust
|
use crate::constants;
|
||
|
|
use crate::energy::consumption;
|
||
|
|
use crate::energy::consumption::EnergyConsumption;
|
||
|
|
use actix::prelude::*;
|
||
|
|
|
||
|
|
pub struct EnergyActor {
|
||
|
|
curr_consumption: EnergyConsumption,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl EnergyActor {
|
||
|
|
pub async fn new() -> anyhow::Result<Self> {
|
||
|
|
Ok(Self {
|
||
|
|
curr_consumption: consumption::get_curr_consumption().await?,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn refresh(&mut self) -> anyhow::Result<()> {
|
||
|
|
// Refresh energy
|
||
|
|
self.curr_consumption = 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
|
||
|
|
});
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Actor for EnergyActor {
|
||
|
|
type Context = Context<Self>;
|
||
|
|
|
||
|
|
fn started(&mut self, ctx: &mut Self::Context) {
|
||
|
|
log::info!("Energy actor successfully started!");
|
||
|
|
|
||
|
|
ctx.run_interval(constants::ENERGY_REFRESH_INTERVAL, |act, _ctx| {
|
||
|
|
log::info!("Performing energy refresh operation");
|
||
|
|
if let Err(e) = futures::executor::block_on(act.refresh()) {
|
||
|
|
log::error!("Energy refresh failed! {e}")
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
fn stopped(&mut self, _ctx: &mut Self::Context) {
|
||
|
|
log::info!("Energy actor successfully stopped!");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub type EnergyActorAddr = Addr<EnergyActor>;
|
||
|
|
|
||
|
|
/// Get current consumption
|
||
|
|
#[derive(Message)]
|
||
|
|
#[rtype(result = "EnergyConsumption")]
|
||
|
|
pub struct GetCurrConsumption;
|
||
|
|
|
||
|
|
impl Handler<GetCurrConsumption> for EnergyActor {
|
||
|
|
type Result = EnergyConsumption;
|
||
|
|
|
||
|
|
fn handle(&mut self, _msg: GetCurrConsumption, _ctx: &mut Context<Self>) -> Self::Result {
|
||
|
|
self.curr_consumption
|
||
|
|
}
|
||
|
|
}
|