refactor(registry): use globals instead of storing in event loop

This commit is contained in:
Nova
2022-06-12 00:51:12 -04:00
parent 3ef45a1041
commit 01971b5048
4 changed files with 39 additions and 47 deletions

View File

@@ -1,12 +1,16 @@
use super::core::Node;
use crate::core::eventloop::EventLoop;
use crate::core::registry::{Registry, RegistryEntry};
use anyhow::{ensure, Result};
use lazy_static::lazy_static;
use rccell::RcCell;
use std::sync::{Arc, RwLock, Weak};
use std::sync::RwLock;
lazy_static! {
static ref PULSE_SENDER_REGISTRY: Registry<PulseSender> = Default::default();
}
pub struct PulseSender {
event_loop: Weak<EventLoop>,
registry_idx: RwLock<Option<usize>>,
registry_idx: RwLock<usize>,
}
impl PulseSender {
@@ -16,32 +20,23 @@ impl PulseSender {
"Node does not have a spatial attached!"
);
let sender = Arc::new(PulseSender {
event_loop: node.borrow().get_client().map_or(Weak::new(), |client| {
Arc::downgrade(&client.get_event_loop())
}),
registry_idx: RwLock::new(None),
});
let idx = sender
.event_loop
.upgrade()
.and_then(|event_loop| event_loop.pulse_senders.add(sender.clone()).ok());
*sender.registry_idx.write().unwrap() = idx;
let sender = PulseSender {
registry_idx: Default::default(),
};
let sender = PULSE_SENDER_REGISTRY.add(sender)?;
node.borrow_mut().pulse_sender = Some(sender);
Ok(())
}
}
impl Drop for PulseSender {
fn drop(&mut self) {
let event_loop = self.event_loop.upgrade();
let idx = self
.registry_idx
.write()
.ok()
.and_then(|registry_idx| registry_idx.clone());
event_loop
.zip(idx)
.and_then(|(event_loop, idx)| event_loop.pulse_senders.remove(idx).ok());
impl RegistryEntry for PulseSender {
fn store_idx(&self, store_idx: usize) {
*self.registry_idx.write().unwrap() = store_idx;
}
}
impl Drop for PulseSender {
fn drop(&mut self) {
let _ = PULSE_SENDER_REGISTRY.remove(*self.registry_idx.read().unwrap());
}
}