tower_buffer/
layer.rs

1use crate::{error::Error, service::Buffer};
2use std::{fmt, marker::PhantomData};
3use tower_layer::Layer;
4use tower_service::Service;
5
6/// Buffer requests with a bounded buffer
7pub struct BufferLayer<Request> {
8    bound: usize,
9    _p: PhantomData<fn(Request)>,
10}
11
12impl<Request> BufferLayer<Request> {
13    /// Create a new `BufferLayer` with the provided `bound`.
14    pub fn new(bound: usize) -> Self {
15        BufferLayer {
16            bound,
17            _p: PhantomData,
18        }
19    }
20}
21
22impl<S, Request> Layer<S> for BufferLayer<Request>
23where
24    S: Service<Request> + Send + 'static,
25    S::Future: Send,
26    S::Error: Into<Error> + Send + Sync,
27    Request: Send + 'static,
28{
29    type Service = Buffer<S, Request>;
30
31    fn layer(&self, service: S) -> Self::Service {
32        Buffer::new(service, self.bound)
33    }
34}
35
36impl<Request> fmt::Debug for BufferLayer<Request> {
37    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
38        f.debug_struct("BufferLayer")
39            .field("bound", &self.bound)
40            .finish()
41    }
42}
tower_buffer/layer.rs

tower_buffer/
layer.rs
