feat(volo-http): support any for routing any methods (#298)

* feat(volo-http): add `any` for routing any methods * chore(volo-http): re-export `Router`, `Query` and `Form` --------- Signed-off-by: Yu Li <[email protected]>
cloudwego · Dec 26, 2023 · 4208ccd · 4208ccd
1 parent 7b6bab2
commit 4208ccd
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/volo-http/src/lib.rs b/volo-http/src/lib.rs
@@ -25,10 +25,11 @@ pub use volo::net::Address;
 
 pub use crate::{
     context::{ConnectionInfo, HttpContext},
-    extract::{Json, MaybeInvalid, State},
+    extract::{Form, Json, MaybeInvalid, Query, State},
     param::Params,
     request::Request,
     response::Response,
+    route::Router,
     server::Server,
 };
 

diff --git a/volo-http/src/route.rs b/volo-http/src/route.rs
@@ -397,6 +397,15 @@ where
 
     for_all_methods!(impl_method_register_for_builder);
 
+    pub fn fallback<H, T>(mut self, handler: H) -> Self
+    where
+        for<'a> H: Handler<T, S> + Clone + Send + Sync + 'a,
+        for<'a> T: 'a,
+    {
+        self.router.fallback = Fallback::Handler(DynHandler::new(handler));
+        self
+    }
+
     pub fn build(self) -> MethodRouter<S> {
         self.router
     }
@@ -419,6 +428,15 @@ macro_rules! impl_method_register {
 
 for_all_methods!(impl_method_register);
 
+pub fn any<H, T, S>(h: H) -> MethodRouter<S>
+where
+    for<'a> H: Handler<T, S> + Clone + Send + Sync + 'a,
+    for<'a> T: 'a,
+    S: Clone + Send + Sync + 'static,
+{
+    MethodRouterBuilder::new().fallback(h).build()
+}
+
 #[derive(Clone, Default)]
 pub enum MethodEndpoint<S> {
     #[default]