xet-runtime 1.5.3

use std::fmt;
use std::sync::atomic::{AtomicU64, Ordering};

static NEXT_ID: AtomicU64 = AtomicU64::new(1);

#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord)]
#[cfg_attr(feature = "python", pyo3::pyclass(from_py_object))]
pub struct UniqueId(pub u64);

impl UniqueId {
    pub fn new() -> Self {
        Self(NEXT_ID.fetch_add(1, Ordering::Relaxed))
    }

    pub fn null() -> Self {
        Self(0)
    }
}

impl Default for UniqueId {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(feature = "python")]
#[pyo3::pymethods]
impl UniqueId {
    fn __repr__(&self) -> String {
        format!("UniqueId({})", self.0)
    }

    fn __hash__(&self) -> u64 {
        self.0
    }

    fn __eq__(&self, other: &Self) -> bool {
        self.0 == other.0
    }
}

impl fmt::Display for UniqueId {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.0)
    }
}

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

    use super::*;

    #[test]
    fn test_unique_id_basics() {
        let id1 = UniqueId::new();
        let id2 = UniqueId::new();
        assert_ne!(id1, id2);

        let cloned = id1;
        assert_eq!(id1, cloned);
    }

    #[test]
    fn test_unique_id_display() {
        let id = UniqueId::new();
        let s = id.to_string();
        assert!(!s.is_empty());
    }

    #[test]
    fn test_unique_id_hash() {
        let id = UniqueId::new();
        let mut map = HashMap::new();
        map.insert(id, 42);
        assert_eq!(map[&id], 42);
    }

    #[test]
    fn test_unique_id_null() {
        let null_id = UniqueId::null();
        let new_id = UniqueId::new();
        assert_ne!(null_id, new_id);
    }
}