use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;
use tracing::debug;
#[derive(Clone)]
pub struct KillSwitch {
alive: Arc<AtomicBool>,
}
impl Default for KillSwitch {
fn default() -> Self {
KillSwitch {
alive: Arc::new(AtomicBool::new(true)),
}
}
}
impl KillSwitch {
pub fn kill(&self) {
debug!("kill-switch-activated");
self.alive.store(false, Ordering::Relaxed);
}
pub fn is_alive(&self) -> bool {
self.alive.load(Ordering::Relaxed)
}
pub fn is_dead(&self) -> bool {
!self.is_alive()
}
}
#[cfg(test)]
mod tests {
use super::KillSwitch;
#[test]
fn test_kill_switch() {
let kill_switch = KillSwitch::default();
assert!(kill_switch.is_alive());
assert!(!kill_switch.is_dead());
kill_switch.kill();
assert!(!kill_switch.is_alive());
assert!(kill_switch.is_dead());
kill_switch.kill();
assert!(!kill_switch.is_alive());
assert!(kill_switch.is_dead());
}
}