Routing with dynamic parts!

2026-08-24 00:00:16 +02:00 · 2021-05-30 15:44:26 +02:00
parent 7328127a3d
commit 763d4e8d21
5 changed files with 182 additions and 30 deletions
@@ -6,8 +6,10 @@ use http_body::Body as _;
 use pin_project::pin_project;
 use serde::de::DeserializeOwned;
 use std::{
+    collections::HashMap,
    future::Future,
    pin::Pin,
+    str::FromStr,
    task::{Context, Poll},
 };

@@ -181,3 +183,31 @@ impl<const N: u64> FromRequest for BytesMaxLength<N> {
        })
    }
 }
+
+pub struct UrlParams(HashMap<String, String>);
+
+impl UrlParams {
+    pub fn get(&self, key: &str) -> Result<&str, Error> {
+        if let Some(value) = self.0.get(key) {
+            Ok(value)
+        } else {
+            Err(Error::UnknownUrlParam(key.to_string()))
+        }
+    }
+}
+
+impl FromRequest for UrlParams {
+    type Future = future::Ready<Result<Self, Error>>;
+
+    fn from_request(req: &mut Request<Body>) -> Self::Future {
+        if let Some(params) = req
+            .extensions_mut()
+            .get_mut::<Option<crate::routing::UrlParams>>()
+        {
+            let params = params.take().expect("params already taken").0;
+            future::ok(Self(params.into_iter().collect()))
+        } else {
+            panic!("no url params found for matched route. This is a bug in tower-web")
+        }
+    }
+}