1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use core::{future::Future, pin::Pin, time::Duration};
use std::time::Instant;

#[cfg(feature = "timeout")]
pub mod timeout;
#[cfg(feature = "timeout")]
pub use self::timeout::{timeout, timeout_at};

#[cfg(feature = "impl_async_io")]
pub mod impl_async_io;
#[cfg(feature = "impl_async_timer")]
pub mod impl_async_timer;
#[cfg(feature = "impl_tokio")]
pub mod impl_tokio;

//
pub trait Sleepble {
    fn sleep(dur: Duration) -> Self;

    fn wait(self) -> Pin<Box<dyn Future<Output = ()> + Send + 'static>>;
}

pub fn sleep<SLEEP>(dur: Duration) -> impl Future<Output = ()>
where
    SLEEP: Sleepble,
{
    SLEEP::sleep(dur).wait()
}

pub fn sleep_until<SLEEP>(deadline: Instant) -> impl Future<Output = ()>
where
    SLEEP: Sleepble,
{
    let dur = deadline
        .checked_duration_since(Instant::now())
        .unwrap_or_default();

    sleep::<SLEEP>(dur)
}