kit_rs/middleware/
registry.rs

1//! Middleware registry for global middleware
2//!
3//! Configure global middleware in `bootstrap.rs` using the `global_middleware!` macro,
4//! or use `Server::middleware()` for manual configuration.
5
6use super::{into_boxed, BoxedMiddleware, Middleware};
7use std::sync::{OnceLock, RwLock};
8
9/// Global middleware registry (populated via `global_middleware!` macro in bootstrap.rs)
10static GLOBAL_MIDDLEWARE: OnceLock<RwLock<Vec<BoxedMiddleware>>> = OnceLock::new();
11
12/// Register a global middleware that runs on every request
13///
14/// Called by the `global_middleware!` macro. Middleware runs in registration order.
15///
16/// # Example
17///
18/// ```rust,ignore
19/// // In bootstrap.rs
20/// global_middleware!(LoggingMiddleware);
21/// global_middleware!(CorsMiddleware);
22/// ```
23pub fn register_global_middleware<M: Middleware + 'static>(middleware: M) {
24    let registry = GLOBAL_MIDDLEWARE.get_or_init(|| RwLock::new(Vec::new()));
25    if let Ok(mut vec) = registry.write() {
26        vec.push(into_boxed(middleware));
27    }
28}
29
30/// Get all registered global middleware
31///
32/// Used internally by `Server::from_config()` to apply middleware.
33pub fn get_global_middleware() -> Vec<BoxedMiddleware> {
34    GLOBAL_MIDDLEWARE
35        .get()
36        .and_then(|lock| lock.read().ok())
37        .map(|vec| vec.clone())
38        .unwrap_or_default()
39}
40
41/// Registry for global middleware that runs on every request
42///
43/// # Example
44///
45/// ```rust,ignore
46/// Server::from_config(router)
47///     .middleware(LoggingMiddleware)  // Global middleware
48///     .middleware(CorsMiddleware)
49///     .run()
50///     .await;
51/// ```
52pub struct MiddlewareRegistry {
53    /// Middleware that runs on every request (in order)
54    global: Vec<BoxedMiddleware>,
55}
56
57impl MiddlewareRegistry {
58    /// Create a new empty middleware registry
59    pub fn new() -> Self {
60        Self { global: Vec::new() }
61    }
62
63    /// Create a registry pre-populated with globally registered middleware
64    ///
65    /// This pulls middleware registered via `global_middleware!` in bootstrap.rs.
66    pub fn from_global() -> Self {
67        Self {
68            global: get_global_middleware(),
69        }
70    }
71
72    /// Append global middleware that runs on every request
73    ///
74    /// Global middleware runs in the order they are added, before any
75    /// route-specific middleware.
76    ///
77    /// # Example
78    ///
79    /// ```rust,ignore
80    /// m.append(LoggingMiddleware)
81    ///  .append(CorsMiddleware)
82    /// ```
83    pub fn append<M: Middleware + 'static>(mut self, middleware: M) -> Self {
84        self.global.push(into_boxed(middleware));
85        self
86    }
87
88    /// Get the list of global middleware
89    pub fn global_middleware(&self) -> &[BoxedMiddleware] {
90        &self.global
91    }
92}
93
94impl Default for MiddlewareRegistry {
95    fn default() -> Self {
96        Self::new()
97    }
98}
kit_rs/middleware/registry.rs

kit_rs/middleware/
registry.rs
