StardustXR/server
1
0
Fork 0
Code Issues 11 Pull Requests 3 Actions Packages Projects Releases 5 Wiki Activity

feat(registry): get_valid_contents

Browse Source
This commit is contained in:
Nova
2022-06-14 21:05:20 -04:00
parent 46a1581e1b
commit 956431bd00
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/core/registry.rs
View File

@@ -1,6 +1,6 @@
use anyhow::{anyhow, Result}; use anyhow::{anyhow, Result};
use parking_lot::RwLock; use parking_lot::RwLock;
use slab::{Iter, Slab}; use slab::Slab;
use std::sync::{Arc, Weak}; use std::sync::{Arc, Weak};
pub struct Registry<T>(RwLock<Slab<Weak<T>>>); pub struct Registry<T>(RwLock<Slab<Weak<T>>>);
@@ -11,9 +11,12 @@ impl<T> Registry<T> {
self.0.write().insert(Arc::downgrade(&t_arc)); self.0.write().insert(Arc::downgrade(&t_arc));
Ok(t_arc) Ok(t_arc)
} }
pub fn iterate<F: FnOnce(Iter<'_, Weak<T>>)>(&self, closure: F) -> Result<()> { pub fn get_valid_contents(&self) -> Vec<Arc<T>> {
closure(self.0.read().iter()); self.0
Ok(()) .read()
.iter()
.filter_map(|(_, item)| item.upgrade())
.collect()
} }
pub fn remove(&self, t: &T) -> Result<()> { pub fn remove(&self, t: &T) -> Result<()> {
let mut del_idx: Option<usize> = None; let mut del_idx: Option<usize> = None;