use std::{
thread::sleep,
time::{Duration, Instant},
};
#[derive(Clone, Debug)]
pub struct Delay {
time: Option<Instant>,
delay: Duration,
}
impl Delay {
pub fn new(delay: Duration) -> Self {
Delay {
time: Some(Instant::now()),
delay,
}
}
pub fn remaining(&self) -> Duration {
self.time
.as_ref()
.and_then(|time| self.delay.checked_sub(time.elapsed()))
.unwrap_or(Duration::default())
}
pub fn sleep(&mut self) {
if let Some(delay) = self.time.take().and_then(|time| self.delay.checked_sub(time.elapsed())) {
sleep(delay);
}
}
}
impl Default for Delay {
fn default() -> Self {
Delay {
time: None,
delay: Default::default(),
}
}
}