spaceflake 1.3.0

⛄ A distributed generator to create unique IDs with ease in Rust; inspired by Twitter's Snowflake
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

extern crate spaceflake;

#[cfg(test)]
mod tests {
    use std::collections::HashMap;
    use std::thread;
    use std::time::Duration;

    use spaceflake::Spaceflake;

    #[test]
    fn bulk_generation() {
        let mut spaceflakes: HashMap<String, Spaceflake> = HashMap::new();
        let settings = spaceflake::BulkGeneratorSettings::new(1_000_000);
        let bulk = spaceflake::bulk_generate(settings).expect("Failed generating the Spaceflakes");
        for spaceflake in bulk {
            if spaceflakes.contains_key(spaceflake.to_string().as_str()) {
                panic!("Spaceflake ID {} is a duplicate", spaceflake.id);
            }
            spaceflakes.insert(spaceflake.to_string(), spaceflake);
        }
    }

    #[test]
    fn bulk_generation_node() {
        let mut spaceflakes: HashMap<String, Spaceflake> = HashMap::new();
        let node = spaceflake::Node::new(1);
        let bulk = node
            .bulk_generate(1_000_000)
            .expect("Failed generating the Spaceflakes");
        for spaceflake in bulk {
            if spaceflakes.contains_key(spaceflake.to_string().as_str()) {
                panic!("Spaceflake ID {} is a duplicate", spaceflake.id);
            }
            spaceflakes.insert(spaceflake.to_string(), spaceflake);
        }
    }

    #[test]
    fn bulk_generation_worker() {
        let mut spaceflakes: HashMap<String, Spaceflake> = HashMap::new();
        let mut node = spaceflake::Node::new(1);
        let worker = node.new_worker();
        let bulk = worker
            .bulk_generate(1_000_000)
            .expect("Failed generating the Spaceflakes");
        for spaceflake in bulk {
            if spaceflakes.contains_key(spaceflake.to_string().as_str()) {
                panic!("Spaceflake ID {} is a duplicate", spaceflake.id);
            }
            spaceflakes.insert(spaceflake.to_string(), spaceflake);
        }
    }

    #[test]
    fn generate_at() {
        let mut node = spaceflake::Node::new(1);
        let worker = node.new_worker();
        let sf = worker.generate_at(1532180612064).unwrap();
        assert_eq!(sf.time(), 1532180612064);
    }

    #[test]
    fn generate_future() {
        let mut node = spaceflake::Node::new(1);
        let worker = node.new_worker();
        let sf = worker.generate_at(2662196938000).unwrap_err();
        assert_eq!(
            sf,
            "The current time must be greater than the time you want to generate the Spaceflake at"
        );
    }

    #[test]
    fn worker_unique() {
        let mut spaceflakes: HashMap<String, Spaceflake> = HashMap::new();
        let mut node = spaceflake::Node::new(1);
        let worker = node.new_worker();

        for _ in 0..1000 {
            let sf = worker.generate().expect("Failed generating the Spaceflake");
            if spaceflakes.contains_key(sf.to_string().as_str()) {
                panic!("Spaceflake ID {} is a duplicate", sf.id);
            }
            spaceflakes.insert(sf.to_string(), sf);
        }
    }

    #[test]
    fn generate_unique() {
        let mut spaceflakes: HashMap<String, Spaceflake> = HashMap::new();
        let settings = spaceflake::GeneratorSettings::default();

        for _ in 0..1000 {
            let sf = spaceflake::generate(settings).expect("Failed generating the Spaceflake");
            if spaceflakes.contains_key(sf.to_string().as_str()) {
                panic!("Spaceflake ID {} is a duplicate", sf.id);
            }
            spaceflakes.insert(sf.to_string(), sf);
            // When using random there is a chance that the sequence will be twice the same due to Rust's speed, hence using a worker is better. We wait a millisecond to make sure it's different.
            thread::sleep(Duration::from_millis(1));
        }
    }
}