rbatis 4.9.5

The Rust SQL Toolkit and ORM Library. An async, pure Rust SQL crate featuring compile-time Dynamic SQL
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

#[cfg(test)]
mod test {
    use dark_std::sync::WaitGroup;
    use rbatis::plugin::snowflake::new_snowflake_id;
    use rbatis::snowflake::Snowflake;
    use std::collections::HashMap;
    use std::thread::sleep;
    use std::time::Duration;

    #[test]
    fn test_new_snowflake_id() {
        println!("{}", new_snowflake_id());
        println!("{}", new_snowflake_id());
    }

    #[test]
    fn test_snowflake_generation() {
        let sf = Snowflake::default();
        let id = sf.generate_id();
        assert_ne!(id, 0);
    }

    #[test]
    fn test_gen() {
        let id = Snowflake::new(1, 1, 0);
        println!("{}", id.generate_id());
        sleep(Duration::from_secs(1));
        println!("{}", id.generate_id());
    }

    #[test]
    fn test_gen1() {
        let id = Snowflake::new(1, 1, 1);
        println!("{}", id.generate_id());
        println!("{}", id.generate_id());
        sleep(Duration::from_secs(1));
        println!("{}", id.generate_id());
        println!("{}", id.generate_id());
    }

    #[test]
    fn test_race() {
        let id_generator_generator = Snowflake::new(1, 1, 0);
        let size = 1000000;
        let mut v1: Vec<i64> = Vec::with_capacity(size);
        let mut v2: Vec<i64> = Vec::with_capacity(size);
        let mut v3: Vec<i64> = Vec::with_capacity(size);
        let mut v4: Vec<i64> = Vec::with_capacity(size);
        let wg = WaitGroup::new();
        std::thread::scope(|s| {
            s.spawn(|| {
                let wg1 = wg.clone();
                for _ in 0..size {
                    v1.push(id_generator_generator.generate_id());
                }
                drop(wg1);
            });
            s.spawn(|| {
                let wg1 = wg.clone();
                for _ in 0..size {
                    v2.push(id_generator_generator.generate_id());
                }
                drop(wg1);
            });
            s.spawn(|| {
                let wg1 = wg.clone();
                for _ in 0..size {
                    v3.push(id_generator_generator.generate_id());
                }
                drop(wg1);
            });
            s.spawn(|| {
                let wg1 = wg.clone();
                for _ in 0..size {
                    v4.push(id_generator_generator.generate_id());
                }
                drop(wg1);
            });
        });

        wg.wait();

        println!(
            "v1 len:{},v2 len:{},v3 len:{},v4 len:{}",
            v1.len(),
            v2.len(),
            v3.len(),
            v4.len()
        );
        let mut all: Vec<i64> = Vec::with_capacity(size * 4);
        all.append(&mut v1);
        all.append(&mut v2);
        all.append(&mut v3);
        all.append(&mut v4);

        let mut id_map: HashMap<i64, i64> = HashMap::with_capacity(all.len());
        for id in all {
            id_map
                .entry(id)
                .and_modify(|count| *count += 1)
                .or_insert(1);
        }
        for (_, v) in id_map {
            assert!(v <= 1);
        }
    }

    #[test]
    fn test_generate_no_clock_back() {
        let snowflake = Snowflake::default();
        let id1 = snowflake.generate_id();
        let id2 = snowflake.generate_id();
        assert_ne!(id1, id2);
    }

    #[test]
    fn test_generate_clock_rollback() {
        let id_generator_generator = Snowflake::new(1, 1, 0);
        let initial_id = id_generator_generator.generate_id();
        println!("initial_id={}", initial_id);

        let new_id = id_generator_generator.generate_id();
        println!("new_id____={}", new_id);
        assert!(new_id > initial_id);
    }
}