nio 0.1.4

Async runtime for Rust
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use std::{
    future::{Future, poll_fn},
    pin::Pin,
    time::{Duration, Instant},
};

use futures_lite::FutureExt;
use nio::{Sleep, sleep, spawn_local, test, timeout};

#[test]
async fn smoke() {
    let start = Instant::now();
    Sleep::at(start + Duration::from_secs(1)).await;
    assert!(start.elapsed() >= Duration::from_secs(1));
}

#[test]
async fn interval() {
    let period = Duration::from_secs(1);
    let jitter = Duration::from_millis(500);
    let start = Instant::now();
    let mut timer = nio::Interval::at(start + period, period);

    timer.tick().await;

    let elapsed = start.elapsed();
    assert!(elapsed >= period);
    assert!(elapsed - period < jitter);

    timer.tick().await;

    let elapsed = start.elapsed();
    assert!(elapsed >= period * 2);
    assert!(elapsed - period * 2 < jitter);
}

#[test]
async fn poll_across_tasks() {
    let start = Instant::now();
    let (sender, mut receiver) = tokio::sync::mpsc::channel(1);

    let task1 = spawn_local(async move {
        let mut timer = Sleep::at(Instant::now() + Duration::from_secs(1));

        async {
            (&mut timer).await;
            panic!("timer should not be ready")
        }
        .or(async {})
        .await;

        sender.send(timer).await.ok();
    });

    let task2 = spawn_local(async move {
        let timer = receiver.recv().await.unwrap();
        timer.await;
    });

    task1.await.unwrap();
    task2.await.unwrap();

    assert!(start.elapsed() >= Duration::from_secs(1));
}

#[test]
async fn set() {
    let start = Instant::now();
    let mut timer = Sleep::at(start + Duration::from_secs(1));

    poll_fn(|cx| {
        let _ = Pin::new(&mut timer).poll(cx);
        std::task::Poll::Ready(())
    })
    .await;

    timer.reset_at(Instant::now() + Duration::from_secs(1));

    (&mut timer).await;

    assert!(timer.is_elapsed());
    assert!(start.elapsed() >= Duration::from_secs(1));
    assert!(start.elapsed() < Duration::from_secs(5));
}

#[test]
async fn test_timeout() {
    let timer = timeout(Duration::from_secs(1), sleep(Duration::from_secs(2)));
    assert!(timer.await.is_none());

    let timer = timeout(Duration::from_secs(2), sleep(Duration::from_secs(1)));
    assert!(timer.await.is_some());
}