1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
//! Run commands and subscriptions.
mod executor;
pub use executor::Executor;
use crate::{subscription, Command, Subscription};
use futures::Sink;
use std::marker::PhantomData;
#[derive(Debug)]
pub struct Runtime<Hasher, Event, Executor, Receiver, Message> {
executor: Executor,
receiver: Receiver,
subscriptions: subscription::Tracker<Hasher, Event>,
_message: PhantomData<Message>,
}
impl<Hasher, Event, Executor, Receiver, Message>
Runtime<Hasher, Event, Executor, Receiver, Message>
where
Hasher: std::hash::Hasher + Default,
Event: Send + Clone + 'static,
Executor: self::Executor,
Receiver: Sink<Message, Error = core::convert::Infallible>
+ Unpin
+ Send
+ Clone
+ 'static,
Message: Send + 'static,
{
pub fn new(executor: Executor, receiver: Receiver) -> Self {
Self {
executor,
receiver,
subscriptions: subscription::Tracker::new(),
_message: PhantomData,
}
}
pub fn enter<R>(&self, f: impl FnOnce() -> R) -> R {
self.executor.enter(f)
}
pub fn spawn(&mut self, command: Command<Message>) {
use futures::{FutureExt, SinkExt};
let futures = command.futures();
for future in futures {
let mut receiver = self.receiver.clone();
self.executor.spawn(future.then(|message| {
async move {
let _ = receiver.send(message).await;
()
}
}));
}
}
pub fn track(
&mut self,
subscription: Subscription<Hasher, Event, Message>,
) {
let futures = self
.subscriptions
.update(subscription, self.receiver.clone());
for future in futures {
self.executor.spawn(future);
}
}
pub fn broadcast(&mut self, event: Event) {
self.subscriptions.broadcast(event);
}
}
|
