mirror of https://github.com/Trivernis/vented.git
parent
d5d8875e79
commit
c550acc129
@ -0,0 +1,49 @@
|
|||||||
|
use crate::event::Event;
|
||||||
|
use serde::de::DeserializeOwned;
|
||||||
|
use serde::Serialize;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests;
|
||||||
|
|
||||||
|
/// A handler for events
|
||||||
|
pub struct EventHandler<T> {
|
||||||
|
event_handlers: HashMap<String, Vec<Box<dyn Fn(&Event<T>)>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T> EventHandler<T>
|
||||||
|
where
|
||||||
|
T: Serialize + DeserializeOwned,
|
||||||
|
{
|
||||||
|
/// Creates a new vented event_handler
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
event_handlers: HashMap::new(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Adds a handler for the given event
|
||||||
|
pub fn on<F: 'static>(&mut self, event_name: &str, handler: F)
|
||||||
|
where
|
||||||
|
F: Fn(&Event<T>),
|
||||||
|
{
|
||||||
|
match self.event_handlers.get_mut(event_name) {
|
||||||
|
Some(handlers) => handlers.push(Box::new(handler)),
|
||||||
|
None => {
|
||||||
|
self.event_handlers
|
||||||
|
.insert(event_name.to_string(), vec![Box::new(handler)]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handles a single event
|
||||||
|
pub fn handle_event(&mut self, event: Event<T>) -> bool {
|
||||||
|
if let Some(handlers) = self.event_handlers.get(&event.name) {
|
||||||
|
handlers.iter().for_each(|handler| handler(&event));
|
||||||
|
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,28 @@
|
|||||||
|
use crate::event::Event;
|
||||||
|
use crate::event_handler::EventHandler;
|
||||||
|
use std::sync::atomic::{AtomicUsize, Ordering};
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn it_handles_events() {
|
||||||
|
let mut handler = EventHandler::new();
|
||||||
|
let call_count = Arc::new(AtomicUsize::new(0));
|
||||||
|
{
|
||||||
|
let call_count = Arc::clone(&call_count);
|
||||||
|
handler.on("test", move |_event| {
|
||||||
|
call_count.fetch_add(1, Ordering::Relaxed);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
{
|
||||||
|
let call_count = Arc::clone(&call_count);
|
||||||
|
handler.on("test2", move |_event| {
|
||||||
|
call_count.fetch_add(1, Ordering::Relaxed);
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
handler.handle_event(Event::new("test".to_string()));
|
||||||
|
handler.handle_event(Event::new("test".to_string()));
|
||||||
|
handler.handle_event(Event::new("test2".to_string()));
|
||||||
|
|
||||||
|
assert_eq!(call_count.load(Ordering::Relaxed), 3)
|
||||||
|
}
|
@ -1,2 +1,3 @@
|
|||||||
pub mod event;
|
pub mod event;
|
||||||
|
pub mod event_handler;
|
||||||
pub mod result;
|
pub mod result;
|
Loading…
Reference in New Issue