feat: 开放RootRoute, 引入root_middleware概念, Route添加root_hook相关方法

silent/Cargo.toml

@@ -41,7 +41,7 @@ tokio = { version = "1.38.0", optional = true }
 bytes = "1.6.0"
 http-body-util = "0.1.2"
 tracing = "0.1.40"
-tracing-subscriber = { version = "0.3.18", features = ["local-time"] }
+tracing-subscriber = { version = "0.3.18", features = ["chrono"] }
 async-trait = "0.1.80"
 serde = { version = "1.0.203", features = ["derive"] }
 serde_json = "1.0.117"

silent/src/lib.rs

@@ -63,7 +63,7 @@ pub mod prelude {
     #[cfg(feature = "upgrade")]
     pub use crate::route::handler_append::WSHandlerAppend;
     pub use crate::route::handler_append::{HandlerAppend, HandlerGetter};
-    pub use crate::route::{Route, RouteService, RouterAdapt};
+    pub use crate::route::{RootRoute, Route, RouteService, RouterAdapt};
     #[cfg(feature = "scheduler")]
     pub use crate::scheduler::Task;
     #[cfg(feature = "security")]

silent/src/route/mod.rs

@@ -28,6 +28,7 @@ pub struct Route {
     pub handler: HashMap<Method, Arc<dyn Handler>>,
     pub children: Vec<Route>,
     pub middlewares: Vec<Arc<dyn MiddleWareHandler>>,
+    pub root_middlewares: Vec<Arc<dyn MiddleWareHandler>>,
     special_match: bool,
     create_path: String,
 }
@@ -80,6 +81,7 @@ impl Route {
             handler: HashMap::new(),
             children: Vec::new(),
             middlewares: Vec::new(),
+            root_middlewares: Vec::new(),
             special_match: first_path.starts_with('<') && first_path.ends_with('>'),
             create_path: path.to_string(),
         };
@@ -90,6 +92,7 @@ impl Route {
         }
     }
     fn append_route(mut self, route: Route) -> Self {
+        self.root_middlewares.extend(route.root_middlewares.clone());
         self.children.push(route);
         self
     }
@@ -110,6 +113,7 @@ impl Route {
     }
     pub fn append<R: RouterAdapt>(mut self, route: R) -> Self {
         let mut route = route.into_router();
+        self.root_middlewares.extend(route.root_middlewares.clone());
         self.middlewares
             .iter()
             .cloned()
@@ -118,6 +122,14 @@ impl Route {
         real_route.children.push(route);
         self
     }
+    pub fn root_hook(mut self, handler: impl MiddleWareHandler + 'static) -> Self {
+        self.root_middlewares.push(Arc::new(handler));
+        self
+    }
+    pub fn root_hook_first(mut self, handler: impl MiddleWareHandler + 'static) -> Self {
+        self.root_middlewares.insert(0, Arc::new(handler));
+        self
+    }
     pub fn hook(mut self, handler: impl MiddleWareHandler + 'static) -> Self {
         self.middleware_hook(Arc::new(handler));
         self

silent/src/route/root.rs

@@ -80,6 +80,7 @@ impl RootRoute {
 
     pub fn push(&mut self, route: Route) {
         self.middlewares.clone_from(&route.middlewares);
+        self.middlewares.extend(route.root_middlewares.clone());
         self.children.push(route);
     }