summaryrefslogtreecommitdiffstats
path: root/core/src/runtime.rs
blob: 31234d110bba28e8a436a5fef5e794fd09758c97 (plain) (blame)
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
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,
    subscriptions: subscription::Tracker<Hasher, Event>,
    receiver: Receiver,
    _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(receiver: Receiver) -> Self {
        Self {
            executor: Executor::new(),
            subscriptions: subscription::Tracker::new(),
            receiver,
            _message: PhantomData,
        }
    }

    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);
    }
}