2 releases
0.1.1 | Aug 7, 2021 |
---|---|
0.1.0 | Aug 7, 2021 |
#20 in #distributed-id
31KB
547 lines
Infallible-SonyFlake
A distributed unique ID generator inspired by Twitter's Snowflake.
This is a Rust implementation of the original sony/sonyflake, which is written in Go.
English | 简体中文
A Sonyflake ID is composed of
39 bits for time in units of 10 msec
8 bits for a sequence number
16 bits for a machine id
Install
Add the following to your Cargo.toml
:
[dependencies]
infallible-sonyflake = "0.1"
Quickstart
- Fallible SonyFlake
Sonyflake
may fail to generate a unique ID when we callnext_id
if time overflows.use infallible_sonyflake::{SonyFlake, Settings}; use chrono::Utc; fn main() { let now = Utc::now(); let mut sf = Settings::new().set_start_time(now).into_sonyflake().unwrap(); let next_id = sf.next_id().unwrap(); println!("{}", next_id); }
- Infallible SonyFlake
InfaillibleSonyFlake
will always generate a unique ID when we callnext_id
if time overflow happens, it will refresh thestart_time
to the current time.use infallible_sonyflake::{InfallibleSonyFlake, Settings}; use chrono::Utc; fn main() { let now = Utc::now(); let mut sf = Settings::new().set_start_time(now).into_infallible_sonyflake().unwrap(); let next_id = sf.next_id(); println!("{}", next_id); }
- Custom machine ID and machine ID checker
use infallible_sonyflake::{InfallibleSonyFlake, Settings, MachineID, MachineIDChecker, IDParts, Error}; use chrono::Utc; struct CustomMachineID { counter: u64, id: u16, } impl MachineID for CustomMachineID { fn machine_id(&mut self) -> Result<u16, Box<dyn std::error::Error + Send + Sync + 'static>> { self.counter += 1; if self.counter % 2 != 0 { Ok(self.id) } else { Err(Box::new("NaN".parse::<u32>().unwrap_err())) } } } struct CustomMachineIDChecker; impl MachineIDChecker for CustomMachineIDChecker { fn check_machine_id(&self, id: u16) -> bool { if id % 2 != 0 { true } else { false } } } fn main() { let mut sf = Settings::new() .set_machine_id(Box::new(CustomMachineID { counter: 0, id: 1 })) .set_check_machine_id(Box::new(CustomMachineIDChecker {})) .into_infallible_sonyflake().unwrap(); let id = sf.next_id(); let parts = IDParts::decompose(id); assert_eq!(parts.get_machine_id(), 1); let err = Settings::new() .set_machine_id(Box::new(CustomMachineID { counter: 0, id: 2 })) .set_check_machine_id(Box::new(CustomMachineIDChecker {})) .into_infallible_sonyflake().unwrap_err(); assert_eq!(format!("{}", err), Error::InvalidMachineID(2).to_string()); }
License
Licensed under Apache License, Version 2.0.Unless you explicitly state otherwise, any contribution intentionally submitted for inclusion in this crate by you, as defined in the Apache-2.0 license.
Dependencies
~5.5–7.5MB
~139K SLoC