| use dyn_any::DynAny; |
| pub use uuid_generation::*; |
|
|
| #[derive(Clone, Copy, serde::Serialize, serde::Deserialize, specta::Type)] |
| pub struct Uuid( |
| #[serde(with = "u64_string")] |
| #[specta(type = String)] |
| u64, |
| ); |
|
|
| mod u64_string { |
| use serde::{self, Deserialize, Deserializer, Serializer}; |
| use std::str::FromStr; |
|
|
| |
| |
| |
| |
| |
| |
| |
| pub fn serialize<S>(value: &u64, serializer: S) -> Result<S::Ok, S::Error> |
| where |
| S: Serializer, |
| { |
| serializer.serialize_str(&value.to_string()) |
| } |
|
|
| |
| |
| |
| |
| |
| |
| |
| pub fn deserialize<'de, D>(deserializer: D) -> Result<u64, D::Error> |
| where |
| D: Deserializer<'de>, |
| { |
| let s = String::deserialize(deserializer)?; |
| u64::from_str(&s).map_err(serde::de::Error::custom) |
| } |
| } |
|
|
| mod uuid_generation { |
| use rand_chacha::ChaCha20Rng; |
| use rand_chacha::rand_core::{RngCore, SeedableRng}; |
| use std::cell::Cell; |
| use std::sync::Mutex; |
|
|
| static RNG: Mutex<Option<ChaCha20Rng>> = Mutex::new(None); |
| thread_local! { |
| pub static UUID_SEED: Cell<Option<u64>> = const { Cell::new(None) }; |
| } |
|
|
| pub fn set_uuid_seed(random_seed: u64) { |
| UUID_SEED.with(|seed| seed.set(Some(random_seed))) |
| } |
|
|
| pub fn generate_uuid() -> u64 { |
| let Ok(mut lock) = RNG.lock() else { panic!("UUID mutex poisoned") }; |
| if lock.is_none() { |
| UUID_SEED.with(|seed| { |
| let random_seed = seed.get().unwrap_or(42); |
| *lock = Some(ChaCha20Rng::seed_from_u64(random_seed)); |
| }) |
| } |
| lock.as_mut().map(ChaCha20Rng::next_u64).expect("UUID mutex poisoned") |
| } |
| } |
|
|
| #[repr(transparent)] |
| #[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash, PartialOrd, Ord, serde::Serialize, serde::Deserialize, specta::Type, DynAny)] |
| pub struct NodeId(pub u64); |
|
|
| impl NodeId { |
| pub fn new() -> Self { |
| Self(generate_uuid()) |
| } |
| } |
|
|
| impl std::fmt::Display for NodeId { |
| fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { |
| write!(f, "{}", self.0) |
| } |
| } |
|
|
