FabInfra-Mirror/spacermake
1
0
Fork 0
mirror of https://github.com/LastExceed/spacermake.git synced 2025-03-12 15:01:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

cancel scheduled shsutdowns upon reactivation

Browse Source
This commit is contained in:
LastExceed 2024-02-13 22:58:34 +01:00
parent cdfd93674b
commit b497436ec4
1 changed files with 27 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/state/listener.rs
View File

@ -151,11 +151,13 @@ impl State<Listener> {
.filter(|slave| if SLAVE_PROPERTIES[slave][index::RUNS_CONTINUOUSLY] { long_slaves } else { short_slaves }); .filter(|slave| if SLAVE_PROPERTIES[slave][index::RUNS_CONTINUOUSLY] { long_slaves } else { short_slaves });
for slave in slaves_to_update { for slave in slaves_to_update {
if !power && SLAVE_PROPERTIES[&slave][index::NEEDS_TRAILING_TIME] { if SLAVE_PROPERTIES[&slave][index::NEEDS_TRAILING_TIME] {
dark_grey_ln!("scheduling delayed shutdown for {}", slave); if power {
let shutdown_timestamp = Instant::now() + Duration::from_secs(30); self.cancel_scheduled_shutdown(&slave).await;
self.scheduled_shutdowns.write().await.push_back((shutdown_timestamp, slave)); } else {
continue; self.schedule_shutdown(slave).await;
continue;
}
} }
self.set_power_state(&slave, power).await; self.set_power_state(&slave, power).await;
@ -163,4 +165,24 @@ impl State<Listener> {
Ok(()) Ok(())
} }
async fn schedule_shutdown(&self, slave: String) {
dark_grey_ln!("scheduling delayed shutdown for {}", slave);
let shutdown_timestamp = Instant::now() + Duration::from_secs(30);
self.scheduled_shutdowns
.write()
.await
.push_back((shutdown_timestamp, slave));
}
async fn cancel_scheduled_shutdown(&self, slave: &String) {
let mut schedule = self.scheduled_shutdowns.write().await;
if let Some(index) = schedule.iter().position(|(_, name)| name == slave) {
dark_grey_ln!("cancelling scheduling shutdown for {}", slave);
schedule.remove(index);
}
}
} }