49 lines
1.1 KiB
Rust
49 lines
1.1 KiB
Rust
use crate::events::Event;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::collections::HashSet;
|
|
use yew_agent::{Agent, AgentLink, Context, HandlerId};
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub enum Request {
|
|
EventBusMsg(Event),
|
|
}
|
|
|
|
pub struct EventBus {
|
|
link: AgentLink<EventBus>,
|
|
subscribers: HashSet<HandlerId>,
|
|
}
|
|
|
|
impl Agent for EventBus {
|
|
type Reach = Context<Self>;
|
|
type Message = ();
|
|
type Input = Request;
|
|
type Output = Event;
|
|
|
|
fn create(link: AgentLink<Self>) -> Self {
|
|
Self {
|
|
link,
|
|
subscribers: HashSet::new(),
|
|
}
|
|
}
|
|
|
|
fn update(&mut self, _msg: Self::Message) {}
|
|
|
|
fn connected(&mut self, id: HandlerId) {
|
|
self.subscribers.insert(id);
|
|
}
|
|
|
|
fn handle_input(&mut self, msg: Self::Input, _id: HandlerId) {
|
|
match msg {
|
|
Request::EventBusMsg(s) => {
|
|
for sub in self.subscribers.iter() {
|
|
self.link.respond(*sub, s.clone());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn disconnected(&mut self, id: HandlerId) {
|
|
self.subscribers.remove(&id);
|
|
}
|
|
}
|