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 std::task::{Context, Poll};
use futures::future::{self, Ready};
use tower::Service;
use crate::state_machine::events::{EventListener, EventSubscriber, MaskLengthUpdate};
pub struct MaskLengthRequest;
pub type MaskLengthResponse = Option<usize>;
pub struct MaskLengthService(EventListener<MaskLengthUpdate>);
impl MaskLengthService {
pub fn new(events: &EventSubscriber) -> Self {
Self(events.mask_length_listener())
}
}
impl Service<MaskLengthRequest> for MaskLengthService {
type Response = MaskLengthResponse;
type Error = ::std::convert::Infallible;
type Future = 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: MaskLengthRequest) -> Self::Future {
future::ready(match self.0.get_latest().event {
MaskLengthUpdate::Invalidate => Ok(None),
MaskLengthUpdate::New(mask_length) => Ok(Some(mask_length)),
})
}
}