mirror of
https://github.com/async-rs/async-std.git
synced 2025-02-16 01:09:40 +00:00
Inline TryFutureExt logic for src/io/timeout.rs (#317)
This commit is contained in:
parent
1819408b46
commit
c7f6543502
1 changed files with 48 additions and 2 deletions
|
@ -1,6 +1,9 @@
|
|||
use std::pin::Pin;
|
||||
use std::task::{Context, Poll};
|
||||
use std::time::Duration;
|
||||
|
||||
use futures_timer::TryFutureExt;
|
||||
use futures_core::future::TryFuture;
|
||||
use futures_timer::Delay;
|
||||
|
||||
use crate::future::Future;
|
||||
use crate::io;
|
||||
|
@ -33,5 +36,48 @@ pub async fn timeout<F, T>(dur: Duration, f: F) -> io::Result<T>
|
|||
where
|
||||
F: Future<Output = io::Result<T>>,
|
||||
{
|
||||
f.timeout(dur).await
|
||||
let f = TimeoutFuture {
|
||||
timeout: Delay::new(dur),
|
||||
future: f,
|
||||
};
|
||||
f.await
|
||||
}
|
||||
|
||||
// Future returned by the [`io::timeout`](./fn.timeout.html) function.
|
||||
#[derive(Debug)]
|
||||
pub struct TimeoutFuture<F, T>
|
||||
where
|
||||
F: Future<Output = io::Result<T>>,
|
||||
{
|
||||
future: F,
|
||||
timeout: Delay,
|
||||
}
|
||||
|
||||
impl<F, T> TimeoutFuture<F, T>
|
||||
where
|
||||
F: Future<Output = io::Result<T>>,
|
||||
{
|
||||
pin_utils::unsafe_pinned!(future: F);
|
||||
pin_utils::unsafe_pinned!(timeout: Delay);
|
||||
}
|
||||
|
||||
impl<F, T> Future for TimeoutFuture<F, T>
|
||||
where
|
||||
F: Future<Output = io::Result<T>>,
|
||||
{
|
||||
type Output = io::Result<T>;
|
||||
|
||||
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
||||
match self.as_mut().future().try_poll(cx) {
|
||||
Poll::Pending => {}
|
||||
other => return other,
|
||||
}
|
||||
|
||||
if self.timeout().poll(cx).is_ready() {
|
||||
let err = Err(io::Error::new(io::ErrorKind::TimedOut, "future timed out").into());
|
||||
Poll::Ready(err)
|
||||
} else {
|
||||
Poll::Pending
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue