63 lines
1.9 KiB
Rust
63 lines
1.9 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use parking_lot::RwLock;
|
|
|
|
use crate::actor::ActorAddress;
|
|
use crate::{AddrBuildHasher, AddrMap};
|
|
|
|
/// Named actor registry — maps human-readable names to actor addresses.
|
|
pub struct NameRegistry {
|
|
names: RwLock<HashMap<String, ActorAddress>>,
|
|
reverse: RwLock<AddrMap<String>>,
|
|
}
|
|
|
|
impl Default for NameRegistry {
|
|
fn default() -> Self {
|
|
Self::new()
|
|
}
|
|
}
|
|
|
|
impl NameRegistry {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
names: RwLock::new(HashMap::new()),
|
|
reverse: RwLock::new(HashMap::with_hasher(AddrBuildHasher)),
|
|
}
|
|
}
|
|
|
|
/// Register a name → address mapping. Returns `Err` if the name is already taken.
|
|
pub fn register(&self, name: String, addr: ActorAddress) -> Result<(), crate::Error> {
|
|
let mut names = self.names.write();
|
|
if names.contains_key(&name) {
|
|
return Err(crate::Error::from("Name already registered"));
|
|
}
|
|
names.insert(name.clone(), addr);
|
|
drop(names);
|
|
self.reverse.write().insert(addr, name);
|
|
Ok(())
|
|
}
|
|
|
|
/// Look up an actor address by name.
|
|
pub fn lookup(&self, name: &str) -> Option<ActorAddress> {
|
|
self.names.read().get(name).copied()
|
|
}
|
|
|
|
/// Unregister a name, returning the address it was bound to.
|
|
pub fn unregister(&self, name: &str) -> Option<ActorAddress> {
|
|
let addr = self.names.write().remove(name)?;
|
|
self.reverse.write().remove(&addr);
|
|
Some(addr)
|
|
}
|
|
|
|
/// Remove a name by address (called on actor death for auto-cleanup).
|
|
pub fn unregister_by_addr(&self, addr: &ActorAddress) {
|
|
if let Some(name) = self.reverse.write().remove(addr) {
|
|
self.names.write().remove(&name);
|
|
}
|
|
}
|
|
|
|
/// Return all registered names.
|
|
pub fn registered_names(&self) -> Vec<String> {
|
|
self.names.read().keys().cloned().collect()
|
|
}
|
|
}
|