hala-io 0.1.6

hala asynchronous io programming primitive types
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

use std::{io, time::Duration};

use super::*;
use futures::*;

/// Add timeout feature for exists `Fut`
pub async fn timeout<'a, Fut, R>(fut: Fut, expired: Option<Duration>) -> io::Result<R>
where
    Fut: Future<Output = io::Result<R>> + 'a,
{
    if let Some(expired) = expired {
        if !expired.is_zero() {
            select! {
                r = fut.fuse() => {
                    return r
                }
                _ = sleep(expired).fuse() => {
                    return Err(io::Error::new(io::ErrorKind::TimedOut, "timeout expired, duration={expired}",));
                }

            }
        } else {
            fut.await
        }
    } else {
        fut.await
    }
}