ax_core 0.3.2

Core library implementing the functions of ax
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

use ax_types::{
    service::{SubscribeRequest, SubscribeResponse},
    AppId,
};
use futures::{
    stream::{self, BoxStream},
    FutureExt, StreamExt,
};
use wsrpc::Service;

use crate::api::events::service::EventService;

pub struct Subscribe {
    event_service: EventService,
}

impl Service for Subscribe {
    type Req = SubscribeRequest;
    type Resp = SubscribeResponse;
    type Error = String;
    type Ctx = AppId;

    fn serve(&self, app_id: AppId, req: Self::Req) -> BoxStream<'static, Result<Self::Resp, Self::Error>> {
        let service = self.event_service.clone();
        async move {
            service
                .subscribe(app_id, req)
                .map(move |x| match x {
                    Ok(stream) => stream.map(Ok).left_stream(),
                    Err(e) => stream::once(futures::future::err(e.to_string())).right_stream(),
                })
                .await
        }
        .flatten_stream()
        .boxed()
    }
}

pub fn service(event_service: EventService) -> Subscribe {
    Subscribe { event_service }
}