Add Query extractor

2 files changed, 23 insertions, 0 deletions

diff --git a/src/extract/mod.rs b/src/extract/mod.rs
index b46a610..e16b9d4 100644
--- a/src/extract/mod.rs
+++ b/src/extract/mod.rs
@@ -1,5 +1,7 @@
 mod auth;
 mod json;
+mod query;
 
 pub use auth::Auth;
 pub use json::ApiJson;
+pub use query::ApiQuery;
diff --git a/src/extract/query.rs b/src/extract/query.rs
new file mode 100644
index 0000000..53c8993
--- /dev/null
+++ b/src/extract/query.rs
@@ -0,0 +1,21 @@
+use axum::{
+    extract::{FromRequestParts, Query, rejection::QueryRejection},
+    http::request::Parts,
+};
+
+use crate::ErrorResponse;
+
+pub struct ApiQuery<T>(pub T);
+
+impl<S, T> FromRequestParts<S> for ApiQuery<T>
+where
+    Query<T>: FromRequestParts<S, Rejection = QueryRejection>,
+    S: Send + Sync,
+{
+    type Rejection = ErrorResponse;
+
+    #[inline]
+    async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
+        Ok(Self(Query::from_request_parts(parts, state).await?.0))
+    }
+}