unsync 0.1.2

Unsynchronized synchronization primitives for async 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

use tokio::task;
use unsync::spsc;

#[cfg(not(miri))]
const SIZE: u32 = 100_000;

#[cfg(miri)]
const SIZE: u32 = 10;

#[tokio::test]
async fn test_spsc() -> Result<(), Box<dyn std::error::Error>> {
    let local = task::LocalSet::new();

    let (mut tx, mut rx) = spsc::channel(2);

    let (a, b) = local
        .run_until(async move {
            let a = task::spawn_local(async move {
                let mut out = Vec::new();

                while let Some(value) = rx.recv().await {
                    out.push(value);

                    if value % 3 == 0 {
                        task::yield_now().await;
                    }
                }

                out
            });

            let b = task::spawn_local(async move {
                for n in 0..SIZE {
                    let _ = tx.send(n).await;

                    if n % 5 == 0 {
                        task::yield_now().await;
                    }
                }
            });

            tokio::join!(a, b)
        })
        .await;

    let actual = a?;
    b?;

    let expected = (0..SIZE).collect::<Vec<_>>();

    assert_eq!(actual, expected);
    Ok(())
}

#[tokio::test]
async fn test_try_send() -> Result<(), task::JoinError> {
    let (mut tx, mut rx) = spsc::channel(3);
    assert!(tx.try_send(1).is_ok());
    assert!(tx.try_send(2).is_ok());
    assert!(tx.try_send(3).is_ok());
    assert!(tx.try_send(4).is_err());

    let first = rx.recv().await;
    assert_eq!(first, Some(1));
    assert!(tx.try_send(5).is_ok());
    assert!(tx.try_send(6).is_err());

    let mut collected = Vec::new();

    drop(tx);

    while let Some(value) = rx.recv().await {
        collected.push(value);
    }

    assert_eq!(collected, vec![2, 3, 5]);
    Ok(())
}