tower_layer/
stack.rs

1use super::Layer;
2use core::fmt;
3
4/// Two [`Layer`]s chained together.
5///
6/// # Examples
7///
8/// ```rust
9/// use tower_layer::{Stack, layer_fn, Layer};
10///
11/// let inner = layer_fn(|service| service+2);
12/// let outer = layer_fn(|service| service*2);
13///
14/// let inner_outer_stack = Stack::new(inner, outer);
15///
16/// // (4 + 2) * 2 = 12
17/// // (4 * 2) + 2 = 10
18/// assert_eq!(inner_outer_stack.layer(4), 12);
19/// ```
20#[derive(Clone)]
21pub struct Stack<Inner, Outer> {
22    inner: Inner,
23    outer: Outer,
24}
25
26impl<Inner, Outer> Stack<Inner, Outer> {
27    /// Creates a new [`Stack`].
28    ///
29    /// # Examples
30    ///
31    /// ```rust
32    /// use tower_layer::{Stack, Identity};
33    ///
34    /// let stack = Stack::new(Identity::new(), Identity::new());
35    /// ```
36    pub const fn new(inner: Inner, outer: Outer) -> Self {
37        Stack { inner, outer }
38    }
39}
40
41impl<S, Inner, Outer> Layer<S> for Stack<Inner, Outer>
42where
43    Inner: Layer<S>,
44    Outer: Layer<Inner::Service>,
45{
46    type Service = Outer::Service;
47
48    fn layer(&self, service: S) -> Self::Service {
49        let inner = self.inner.layer(service);
50
51        self.outer.layer(inner)
52    }
53}
54
55impl<Inner, Outer> fmt::Debug for Stack<Inner, Outer>
56where
57    Inner: fmt::Debug,
58    Outer: fmt::Debug,
59{
60    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
61        // The generated output of nested `Stack`s is very noisy and makes
62        // it harder to understand what is in a `ServiceBuilder`.
63        //
64        // Instead, this output is designed assuming that a `Stack` is
65        // usually quite nested, and inside a `ServiceBuilder`. Therefore,
66        // this skips using `f.debug_struct()`, since each one would force
67        // a new layer of indentation.
68        //
69        // - In compact mode, a nested stack ends up just looking like a flat
70        //   list of layers.
71        //
72        // - In pretty mode, while a newline is inserted between each layer,
73        //   the `DebugStruct` used in the `ServiceBuilder` will inject padding
74        //   to that each line is at the same indentation level.
75        //
76        // Also, the order of [outer, inner] is important, since it reflects
77        // the order that the layers were added to the stack.
78        if f.alternate() {
79            // pretty
80            write!(f, "{:#?},\n{:#?}", self.outer, self.inner)
81        } else {
82            write!(f, "{:?}, {:?}", self.outer, self.inner)
83        }
84    }
85}