use super::{Instant, Wait};
use std::future::IntoFuture;
use std::ops::{Add, AddAssign, Sub, SubAssign};
use wasi::clocks::monotonic_clock;
#[derive(Debug, PartialEq, PartialOrd, Ord, Eq, Hash, Clone, Copy)]
pub struct Duration(pub(crate) monotonic_clock::Duration);
impl Duration {
#[must_use]
#[inline]
pub fn new(secs: u64, nanos: u32) -> Duration {
std::time::Duration::new(secs, nanos).into()
}
#[must_use]
#[inline]
pub fn from_secs(secs: u64) -> Duration {
std::time::Duration::from_secs(secs).into()
}
#[must_use]
#[inline]
pub fn from_millis(millis: u64) -> Self {
std::time::Duration::from_millis(millis).into()
}
#[must_use]
#[inline]
pub fn from_micros(micros: u64) -> Self {
std::time::Duration::from_micros(micros).into()
}
#[must_use]
#[inline]
pub fn from_secs_f64(secs: f64) -> Duration {
std::time::Duration::from_secs_f64(secs).into()
}
#[must_use]
#[inline]
pub fn from_secs_f32(secs: f32) -> Duration {
std::time::Duration::from_secs_f32(secs).into()
}
}
impl std::ops::Deref for Duration {
type Target = monotonic_clock::Duration;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl std::ops::DerefMut for Duration {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
}
impl From<std::time::Duration> for Duration {
fn from(inner: std::time::Duration) -> Self {
Self(
inner
.as_nanos()
.try_into()
.expect("only dealing with durations that can fit in u64"),
)
}
}
impl Into<std::time::Duration> for Duration {
fn into(self) -> std::time::Duration {
std::time::Duration::from_nanos(self.0)
}
}
impl Add<Duration> for Duration {
type Output = Self;
fn add(self, rhs: Duration) -> Self::Output {
Self(self.0 + rhs.0)
}
}
impl AddAssign<Duration> for Duration {
fn add_assign(&mut self, rhs: Duration) {
*self = Self(self.0 + rhs.0)
}
}
impl Sub<Duration> for Duration {
type Output = Self;
fn sub(self, rhs: Duration) -> Self::Output {
Self(self.0 - rhs.0)
}
}
impl SubAssign<Duration> for Duration {
fn sub_assign(&mut self, rhs: Duration) {
*self = Self(self.0 - rhs.0)
}
}
impl IntoFuture for Duration {
type Output = Instant;
type IntoFuture = Wait;
fn into_future(self) -> Self::IntoFuture {
crate::task::sleep(self)
}
}