fastsend 1.2.3

快速生成适用于分布式环境的 ID 和序列号
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

use fastsend::ID;
use fastsend::{Serial, Serialer, TimeSerialer};
use futures::future;
use std::collections::HashSet;
use std::error::Error;
use std::result::Result as StdResult;

type Result<T> = StdResult<T, Box<dyn Error>>;

const TOP: usize = 10999;

macro_rules! into_hashset {
    (@it $iter:expr) => {{
        IntoIterator::into_iter($iter).collect::<HashSet<_>>()
    }};

    (@ok $iter:expr) => {{
        IntoIterator::into_iter($iter)
            .collect::<StdResult<Vec<_>, _>>()?
            .into_iter()
            .collect::<HashSet<_>>()
    }};
}

#[tokio::test]
async fn test_unique_id() {
    let set = into_hashset!(@it
        future::join_all((0..TOP).map(|_| async { fastsend::next_token().await.id() })).await
    );

    assert_eq!(set.len(), TOP);
}

#[tokio::test(flavor = "multi_thread", worker_threads = 10)]
async fn test_unique_id_concurrently() -> Result<()> {
    let set = into_hashset!(@ok
        future::join_all(
            (0..TOP).map(|_| {
                tokio::spawn(async move { fastsend::next_token().await.id() })
            })
        )
        .await
    );

    assert_eq!(set.len(), TOP);

    Ok(())
}

macro_rules! serial {
    ($expr:expr => $serialer:ty) => {{
        let mut serialer = <$serialer>::default();
        $expr.serial(&mut serialer);
        serialer.build().await
    }};
}

#[tokio::test(flavor = "multi_thread", worker_threads = 10)]
async fn test_unique_serial_concurrently() -> Result<()> {
    let set = into_hashset!(@ok
        future::join_all(
            (0..TOP).map(|_| {
                tokio::spawn(async move { serial!(fastsend::next_token().await => TimeSerialer) })
            })
        )
        .await
    );

    assert_eq!(set.len(), TOP);

    Ok(())
}