wtx 0.50.0

A collection of different transport implementations and related tools focused primarily on web technologies.
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

use crate::futures::Sleep;
use core::{
  pin::{Pin, pin},
  task::{Context, Poll},
  time::Duration,
};

pin_project_lite::pin_project! {
  /// Requires a `Future` to complete within a specified duration.
  #[derive(Debug)]
  pub struct Timeout<F> {
    #[pin]
    future: F,
    #[pin]
    sleep: Sleep,
  }
}

impl<F> Timeout<F> {
  /// New instance from a duration that creates an owned [`Sleep`].
  #[inline]
  pub fn new(future: F, duration: Duration) -> crate::Result<Self> {
    Ok(Self { future, sleep: Sleep::new(duration)? })
  }
}

impl<F> Future for Timeout<F>
where
  F: Future,
{
  type Output = crate::Result<Option<F::Output>>;

  #[allow(unused_mut, reason = "depends on feature")]
  #[inline]
  fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
    let this = self.project();
    if let Poll::Ready(output) = pin!(this.future).as_mut().poll(cx) {
      return Poll::Ready(Ok(Some(output)));
    }
    let poll = this.sleep.poll(cx);
    match poll {
      Poll::Ready(Ok(_)) => Poll::Ready(Ok(None)),
      Poll::Ready(Err(err)) => Poll::Ready(Err(err)),
      Poll::Pending => Poll::Pending,
    }
  }
}

#[cfg(all(not(feature = "tokio"), test))]
mod tests {
  use crate::futures::{Sleep, Timeout};
  use core::{pin::pin, time::Duration};

  #[wtx::test]
  async fn timeout() {
    assert_eq!(
      Timeout::new(pin!(async { 1 }), Duration::from_millis(10)).unwrap().await.unwrap(),
      Some(1)
    );
    assert_eq!(
      Timeout::new(
        pin!(async {
          Sleep::new(Duration::from_millis(20)).unwrap().await.unwrap();
          1
        }),
        Duration::from_millis(10)
      )
      .unwrap()
      .await
      .unwrap(),
      None
    );
  }
}