spwnmsg_core/
snowflake.rs

1use chrono::Utc;
2use parking_lot::Mutex;
3use std::sync::Arc;
4
5#[derive(Debug)]
6pub struct Snowflake {
7    epoch: i64,
8    worker_id: i64,
9    datacenter_id: i64,
10    sequence: i64,
11    time: Arc<Mutex<i64>>,
12}
13
14impl Default for Snowflake {
15    fn default() -> Snowflake {
16        Snowflake {
17            epoch: 1_573_948_800,
18            worker_id: 1,
19            datacenter_id: 1,
20            sequence: 0,
21            time: Arc::new(Mutex::new(0)),
22        }
23    }
24}
25
26impl Snowflake {
27    pub fn new(epoch: i64, worker_id: i64, datacenter_id: i64) -> Snowflake {
28        Self {
29            epoch,
30            worker_id,
31            datacenter_id,
32            ..Default::default()
33        }
34    }
35
36    pub fn generate(&mut self) -> i64 {
37        let mut last_timestamp = self.time.lock();
38        let mut timestamp = self.get_time();
39        if timestamp == *last_timestamp {
40            self.sequence = (self.sequence + 1) & (-1 ^ (-1 << 12));
41            if self.sequence == 0 && timestamp <= *last_timestamp {
42                timestamp = self.get_time();
43            }
44        } else {
45            self.sequence = 0;
46        }
47        *last_timestamp = timestamp;
48        (timestamp << 22) | (self.worker_id << 17) | (self.datacenter_id << 12) | self.sequence
49    }
50
51    pub fn generate_u8_u64(&mut self) -> [u8; 8] {
52        self.generate().to_le_bytes()
53    }
54
55    fn get_time(&self) -> i64 {
56        Utc::now().timestamp_millis() - self.epoch
57    }
58}
spwnmsg_core/snowflake.rs

spwnmsg_core/
snowflake.rs
