rs-zero 0.2.6

Rust-first microservice framework inspired by go-zero engineering practices
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use axum::Router;

use crate::rest::{RestConfig, middleware::apply_default_layers};

/// Tower-first REST layer stack builder.
///
/// The current implementation preserves the existing rs-zero REST middleware
/// order while exposing an explicit stack object. Future releases can move
/// individual middleware pieces behind concrete Tower [`tower::Layer`] types
/// without changing [`RestServer`](crate::rest::RestServer) callers.
#[derive(Debug, Clone)]
pub struct RestLayerStack {
    config: RestConfig,
}

impl RestLayerStack {
    /// Creates a REST layer stack from runtime configuration.
    pub fn new(config: RestConfig) -> Self {
        Self { config }
    }

    /// Creates a production-oriented REST layer stack.
    pub fn production_defaults(name: impl Into<String>) -> Self {
        Self::new(RestConfig::production_defaults(name))
    }

    /// Creates a production-oriented REST layer stack.
    #[allow(deprecated)]
    #[deprecated(note = "use production_defaults instead")]
    pub fn go_zero_defaults(name: impl Into<String>) -> Self {
        Self::production_defaults(name)
    }

    /// Applies the stack to an axum router.
    pub fn layer(&self, router: Router) -> Router {
        apply_default_layers(router, self.config.clone())
    }

    /// Returns an axum-compatible layer value.
    pub fn into_layer(self) -> RestRouterLayer {
        RestRouterLayer {
            config: self.config,
        }
    }
}

/// Axum router layer produced by [`RestLayerStack`].
#[derive(Debug, Clone)]
pub struct RestRouterLayer {
    config: RestConfig,
}

impl tower::Layer<Router> for RestRouterLayer {
    type Service = Router;

    fn layer(&self, inner: Router) -> Self::Service {
        apply_default_layers(inner, self.config.clone())
    }
}