trapeze 0.2.0

A rust implementation of ttrpc
Documentation 
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

use std::future::Future;
use std::ops::Deref;
use std::time::Duration;

use crate::{Result, Status};

#[derive(Clone, Copy, Debug)]
pub enum Timeout {
    None,
    Duration(Duration),
}

impl Default for Timeout {
    fn default() -> Self {
        Self::None
    }
}

impl Deref for Timeout {
    type Target = Duration;
    fn deref(&self) -> &Self::Target {
        match self {
            Timeout::None => &Duration::ZERO,
            Timeout::Duration(t) => t,
        }
    }
}

pub trait WithTimeout {
    type Ouput;
    fn with_timeout(self, t: impl Into<Timeout>) -> impl Future<Output = Result<Self::Ouput>>;
}

impl<Output, F: Future<Output = Result<Output>>> WithTimeout for F {
    type Ouput = Output;
    async fn with_timeout(self, t: impl Into<Timeout>) -> Result<Self::Ouput> {
        let Timeout::Duration(t) = t.into() else {
            return self.await;
        };

        let Ok(output) = tokio::time::timeout(t, self).await else {
            return Err(Status::timeout());
        };

        output
    }
}

const MAX_TIMEOUT: Duration = Duration::from_nanos(i64::MAX as u64);

impl From<Option<Duration>> for Timeout {
    fn from(value: Option<Duration>) -> Self {
        match value {
            Some(t) => t.into(),
            _ => Timeout::None,
        }
    }
}

impl From<Duration> for Timeout {
    fn from(t: Duration) -> Self {
        if t.is_zero() {
            return Timeout::None;
        }
        Timeout::Duration(t.min(MAX_TIMEOUT))
    }
}

impl Timeout {
    pub fn from_nanos(nanos: i64) -> Self {
        Some(Duration::from_nanos(nanos.max(0) as u64)).into()
    }

    pub fn as_nanos(&self) -> i64 {
        self.deref().as_nanos().min(i64::MAX as u128) as i64
    }
}