use lasso::{Spur, ThreadedRodeo};
use serde::{Deserialize, Deserializer, Serialize, Serializer};
use std::fmt;
use std::sync::LazyLock;
static INTERNER: LazyLock<ThreadedRodeo> = LazyLock::new(ThreadedRodeo::default);
#[derive(Clone, Copy, PartialEq, Eq, Hash)]
pub struct NodeId(Spur);
impl NodeId {
pub fn intern(s: &str) -> Self {
NodeId(INTERNER.get_or_intern(s))
}
pub fn as_str(&self) -> &str {
INTERNER.resolve(&self.0)
}
pub fn anonymous(kind: &str) -> Self {
Self::with_prefix(&format!("_{kind}"))
}
pub fn with_prefix(prefix: &str) -> Self {
use std::sync::atomic::{AtomicU64, Ordering};
static COUNTER: AtomicU64 = AtomicU64::new(0);
let n = COUNTER.fetch_add(1, Ordering::Relaxed);
Self::intern(&format!("{prefix}_{n}"))
}
}
impl fmt::Debug for NodeId {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "@{}", self.as_str())
}
}
impl fmt::Display for NodeId {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "@{}", self.as_str())
}
}
impl Serialize for NodeId {
fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
serializer.serialize_str(self.as_str())
}
}
impl<'de> Deserialize<'de> for NodeId {
fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
let s = String::deserialize(deserializer)?;
Ok(NodeId::intern(&s))
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn interning_roundtrip() {
let a = NodeId::intern("login_form");
let b = NodeId::intern("login_form");
assert_eq!(a, b);
assert_eq!(a.as_str(), "login_form");
}
#[test]
fn anonymous_ids_are_unique() {
let a = NodeId::anonymous("rect");
let b = NodeId::anonymous("rect");
assert_ne!(a, b);
}
}