yew_agent/reactor/
provider.rs

1use std::any::type_name;
2use std::cell::RefCell;
3use std::fmt;
4use std::rc::Rc;
5
6use gloo_worker::{Bincode, Codec};
7use serde::{Deserialize, Serialize};
8use yew::prelude::*;
9
10use super::{Reactor, ReactorBridge, ReactorScoped, ReactorSpawner};
11use crate::Reach;
12use crate::utils::get_next_id;
13use crate::worker::WorkerProviderProps;
14
15pub(crate) struct ReactorProviderState<T>
16where
17    T: Reactor + 'static,
18{
19    id: usize,
20    spawn_bridge_fn: Rc<dyn Fn() -> ReactorBridge<T>>,
21    reach: Reach,
22    held_bridge: Rc<RefCell<Option<ReactorBridge<T>>>>,
23}
24
25impl<T> fmt::Debug for ReactorProviderState<T>
26where
27    T: Reactor,
28{
29    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
30        f.debug_struct(type_name::<Self>()).finish_non_exhaustive()
31    }
32}
33
34impl<T> ReactorProviderState<T>
35where
36    T: Reactor,
37{
38    fn get_held_bridge(&self) -> ReactorBridge<T> {
39        let mut held_bridge = self.held_bridge.borrow_mut();
40
41        match held_bridge.as_mut() {
42            Some(m) => m.fork(),
43            None => {
44                let bridge = (self.spawn_bridge_fn)();
45                *held_bridge = Some(bridge.fork());
46                bridge
47            }
48        }
49    }
50
51    /// Creates a bridge, uses "fork" for public agents.
52    pub fn create_bridge(&self) -> ReactorBridge<T> {
53        match self.reach {
54            Reach::Public => {
55                let held_bridge = self.get_held_bridge();
56                held_bridge.fork()
57            }
58            Reach::Private => (self.spawn_bridge_fn)(),
59        }
60    }
61}
62
63impl<T> Clone for ReactorProviderState<T>
64where
65    T: Reactor,
66{
67    fn clone(&self) -> Self {
68        Self {
69            id: self.id,
70            spawn_bridge_fn: self.spawn_bridge_fn.clone(),
71            reach: self.reach,
72            held_bridge: self.held_bridge.clone(),
73        }
74    }
75}
76
77impl<T> PartialEq for ReactorProviderState<T>
78where
79    T: Reactor,
80{
81    fn eq(&self, rhs: &Self) -> bool {
82        self.id == rhs.id
83    }
84}
85
86/// The Reactor Agent Provider.
87///
88/// This component provides its children access to a reactor agent.
89#[component]
90pub fn ReactorProvider<R, C = Bincode>(props: &WorkerProviderProps) -> Html
91where
92    R: 'static
93        + Reactor<
94            Scope: ReactorScoped<
95                Input: Serialize + for<'de> Deserialize<'de> + 'static,
96                Output: Serialize + for<'de> Deserialize<'de> + 'static,
97            >,
98        >,
99    C: Codec + 'static,
100{
101    let WorkerProviderProps {
102        children,
103        path,
104        lazy,
105        module,
106        reach,
107    } = props.clone();
108
109    // Creates a spawning function so Codec is can be erased from contexts.
110    let spawn_bridge_fn: Rc<dyn Fn() -> ReactorBridge<R>> = {
111        let path = path.clone();
112        Rc::new(move || {
113            ReactorSpawner::<R>::new()
114                .as_module(module)
115                .encoding::<C>()
116                .spawn(&path)
117        })
118    };
119
120    let state = {
121        use_memo((path, lazy, reach), move |(_path, lazy, reach)| {
122            let state = ReactorProviderState::<R> {
123                id: get_next_id(),
124                spawn_bridge_fn,
125                reach: *reach,
126                held_bridge: Rc::default(),
127            };
128
129            if *reach == Reach::Public && !*lazy {
130                state.get_held_bridge();
131            }
132            state
133        })
134    };
135
136    html! {
137        <ContextProvider<ReactorProviderState<R>> context={(*state).clone()}>
138            {children}
139        </ContextProvider<ReactorProviderState<R>>>
140    }
141}
yew_agent/reactor/provider.rs

yew_agent/reactor/
provider.rs
