87 lines
1.9 KiB
Rust
87 lines
1.9 KiB
Rust
#![allow(dead_code)]
|
|
|
|
use std::ptr;
|
|
use std::sync::{Arc, Weak};
|
|
|
|
use once_cell::sync::Lazy;
|
|
use parking_lot::Mutex;
|
|
use rustc_hash::FxHashMap;
|
|
|
|
pub struct Registry<T: Send + Sync + ?Sized>(Lazy<Mutex<FxHashMap<usize, Weak<T>>>>);
|
|
|
|
impl<T: Send + Sync + ?Sized> Registry<T> {
|
|
pub const fn new() -> Self {
|
|
Registry(Lazy::new(|| Mutex::new(FxHashMap::default())))
|
|
}
|
|
pub fn add(&self, t: T) -> Arc<T>
|
|
where
|
|
T: Sized,
|
|
{
|
|
let t_arc = Arc::new(t);
|
|
self.add_raw(&t_arc);
|
|
t_arc
|
|
}
|
|
pub fn add_raw(&self, t: &Arc<T>) {
|
|
self.0
|
|
.lock()
|
|
.insert(Arc::as_ptr(t) as *const () as usize, Arc::downgrade(t));
|
|
}
|
|
pub fn contains(&self, t: &T) -> bool {
|
|
self.0
|
|
.lock()
|
|
.contains_key(&(ptr::addr_of!(*t) as *const () as usize))
|
|
}
|
|
pub fn get_valid_contents(&self) -> Vec<Arc<T>> {
|
|
self.0
|
|
.lock()
|
|
.iter()
|
|
.filter_map(|pair| pair.1.upgrade())
|
|
.collect()
|
|
}
|
|
pub fn remove(&self, t: &T) {
|
|
self.0
|
|
.lock()
|
|
.remove(&(ptr::addr_of!(*t) as *const () as usize));
|
|
}
|
|
pub fn clear(&self) {
|
|
self.0.lock().clear();
|
|
}
|
|
}
|
|
|
|
pub struct OwnedRegistry<T: Send + Sync + ?Sized>(Lazy<Mutex<FxHashMap<usize, Arc<T>>>>);
|
|
|
|
impl<T: Send + Sync + ?Sized> OwnedRegistry<T> {
|
|
pub const fn new() -> Self {
|
|
OwnedRegistry(Lazy::new(|| Mutex::new(FxHashMap::default())))
|
|
}
|
|
pub fn add(&self, t: T) -> Arc<T>
|
|
where
|
|
T: Sized,
|
|
{
|
|
let t_arc = Arc::new(t);
|
|
self.add_raw(t_arc.clone());
|
|
t_arc
|
|
}
|
|
pub fn add_raw(&self, t: Arc<T>) {
|
|
self.0
|
|
.lock()
|
|
.insert(Arc::as_ptr(&t) as *const () as usize, t);
|
|
}
|
|
pub fn get_vec(&self) -> Vec<Arc<T>> {
|
|
self.0.lock().values().cloned().collect::<Vec<_>>()
|
|
}
|
|
pub fn contains(&self, t: &T) -> bool {
|
|
self.0
|
|
.lock()
|
|
.contains_key(&(ptr::addr_of!(*t) as *const () as usize))
|
|
}
|
|
pub fn remove(&self, t: &T) {
|
|
self.0
|
|
.lock()
|
|
.remove(&(ptr::addr_of!(*t) as *const () as usize));
|
|
}
|
|
pub fn clear(&self) {
|
|
self.0.lock().clear();
|
|
}
|
|
}
|