gloo_worker/reactor/
spawner.rs

1use serde::de::Deserialize;
2use serde::ser::Serialize;
3
4use super::bridge::ReactorBridge;
5use super::scope::ReactorScoped;
6use super::traits::Reactor;
7use super::worker::ReactorWorker;
8use crate::actor::WorkerSpawner;
9use crate::codec::{Bincode, Codec};
10
11/// A spawner to create oneshot workers.
12#[derive(Debug, Default)]
13pub struct ReactorSpawner<R, CODEC = Bincode>
14where
15    R: Reactor + 'static,
16    CODEC: Codec,
17{
18    inner: WorkerSpawner<ReactorWorker<R>, CODEC>,
19}
20
21impl<R, CODEC> ReactorSpawner<R, CODEC>
22where
23    R: Reactor + 'static,
24    CODEC: Codec,
25{
26    /// Creates a ReactorSpawner.
27    pub const fn new() -> Self {
28        Self {
29            inner: WorkerSpawner::<ReactorWorker<R>, CODEC>::new(),
30        }
31    }
32
33    /// Sets a new message encoding.
34    pub const fn encoding<C>(&self) -> ReactorSpawner<R, C>
35    where
36        C: Codec,
37    {
38        ReactorSpawner {
39            inner: WorkerSpawner::<ReactorWorker<R>, C>::new(),
40        }
41    }
42
43    /// Spawns a reactor worker.
44    pub fn spawn(mut self, path: &str) -> ReactorBridge<R>
45    where
46        <R::Scope as ReactorScoped>::Input: Serialize + for<'de> Deserialize<'de>,
47        <R::Scope as ReactorScoped>::Output: Serialize + for<'de> Deserialize<'de>,
48    {
49        let rx = ReactorBridge::register_callback(&mut self.inner);
50
51        let inner = self.inner.spawn(path);
52
53        ReactorBridge::new(inner, rx)
54    }
55
56    /// Spawns a Reactor Worker with a loader shim script.
57    pub fn spawn_with_loader(mut self, loader_path: &str) -> ReactorBridge<R>
58    where
59        <R::Scope as ReactorScoped>::Input: Serialize + for<'de> Deserialize<'de>,
60        <R::Scope as ReactorScoped>::Output: Serialize + for<'de> Deserialize<'de>,
61    {
62        let rx = ReactorBridge::register_callback(&mut self.inner);
63
64        let inner = self.inner.spawn_with_loader(loader_path);
65
66        ReactorBridge::new(inner, rx)
67    }
68}