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
use std::{
sync::Arc,
task::{Context, Poll},
};
use futures::future::{self, Ready};
use tower::Service;
use tracing_futures::{Instrument, Instrumented};
use xaynet_core::mask::Model;
use crate::state_machine::events::{EventListener, EventSubscriber, ModelUpdate};
#[derive(Default, Clone, Eq, PartialEq, Debug)]
pub struct ModelRequest;
pub type ModelResponse = Option<Arc<Model>>;
pub struct ModelService(EventListener<ModelUpdate>);
impl ModelService {
pub fn new(events: &EventSubscriber) -> Self {
Self(events.model_listener())
}
}
impl Service<ModelRequest> for ModelService {
type Response = ModelResponse;
type Error = ::std::convert::Infallible;
type Future = Instrumented<Ready<Result<Self::Response, Self::Error>>>;
fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
Poll::Ready(Ok(()))
}
fn call(&mut self, _req: ModelRequest) -> Self::Future {
future::ready(match self.0.get_latest().event {
ModelUpdate::Invalidate => Ok(None),
ModelUpdate::New(model) => Ok(Some(model)),
})
.instrument(error_span!("model_fetch_request"))
}
}