intrepid-model 0.3.0

Manage complex async business logic with ease
Documentation 
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

use std::collections::HashMap;

use futures::Stream;
use intrepid_core::Frame;
use tokio_stream::StreamMap;
use tower::Service;

use crate::EventRepo;

use super::{SubscriptionConfig, SubscriptionWorker};

/// A map of subscription processes which can be run as a stream.
pub struct SubscriptionWorkerMap {
    inner: HashMap<String, (SubscriptionConfig, SubscriptionWorker)>,
}

impl SubscriptionWorkerMap {
    /// Create a new subscription collection.
    pub fn new() -> Self {
        Self {
            inner: HashMap::new(),
        }
    }

    /// Insert a subscription process into the map using a config and a frame service.
    /// The subscription process will be started immediately, but won't do anything until
    /// it is polled.
    pub fn insert<Repo, FrameService>(
        &mut self,
        repo: &Repo,
        config: impl Into<SubscriptionConfig>,
        service: FrameService,
    ) where
        Repo: EventRepo + Clone + Send + Sync + 'static,
        FrameService: Service<Frame> + Clone + Send + Sync + 'static,
        FrameService::Future: Send,
        <FrameService as Service<Frame>>::Response: Into<Frame>,
        <FrameService as Service<Frame>>::Error: Into<Frame>,
    {
        let config = config.into();
        let subscription = config.subscription(repo).unwrap();
        let process = SubscriptionWorker::new(subscription, service.clone());

        self.inner.insert(config.stream_id(), (config, process));
    }

    /// Consume the process map and collect all the workers from it into a single stream.
    pub fn stream(self) -> impl Stream<Item = (String, Frame)> {
        let mut stream_map = StreamMap::new();

        for (id, (config, process)) in self.inner {
            stream_map.insert(id, Box::pin(config.worker(process)));
        }

        stream_map
    }
}

impl Default for SubscriptionWorkerMap {
    fn default() -> Self {
        Self::new()
    }
}