use http::Request;
#[macro_export]
macro_rules! request_parser_joiner {
($name:ident ,$($T:ty), *) => {
struct $name;
impl <B> RequestParser<B> for $name
where $($T: RequestParser<B>, )*
{
fn parse_operation_id(request: &Request<B>) -> Option<&'static str> {
__impl_request_parser_joiner!(request, $($T), *)
}
}
};
}
#[macro_export]
#[doc(hidden)]
macro_rules! __impl_request_parser_joiner {
($argname:expr, $head:ty) => {<$head as RequestParser<B>>::parse_operation_id(&$argname)};
($argname:expr, $head:ty, $( $tail:ty), *) => {
match <$head as RequestParser<B>>::parse_operation_id(&$argname) {
Some(s) => Some(s),
None => __impl_request_parser_joiner!($argname, $( $tail), *),
}
};
}
pub trait RequestParser<B> {
fn parse_operation_id(req: &Request<B>) -> Option<&'static str>;
}
#[cfg(test)]
mod context_tests {
use super::*;
use bytes::Bytes;
use http::Uri;
use http_body_util::Full;
use std::str::FromStr;
struct TestParser1;
impl RequestParser<Full<Bytes>> for TestParser1 {
fn parse_operation_id(request: &Request<Full<Bytes>>) -> Option<&'static str> {
match request.uri().path() {
"/test/t11" => Some("t11"),
"/test/t12" => Some("t12"),
_ => None,
}
}
}
struct TestParser2;
impl RequestParser<Full<Bytes>> for TestParser2 {
fn parse_operation_id(request: &Request<Full<Bytes>>) -> Option<&'static str> {
match request.uri().path() {
"/test/t21" => Some("t21"),
"/test/t22" => Some("t22"),
_ => None,
}
}
}
#[test]
fn test_macros() {
let uri = Uri::from_str("https://www.rust-lang.org/test/t11").unwrap();
let req1: Request<Full<Bytes>> = Request::get(uri).body(Full::default()).unwrap();
let uri = Uri::from_str("https://www.rust-lang.org/test/t22").unwrap();
let req2: Request<Full<Bytes>> = Request::get(uri).body(Full::default()).unwrap();
let uri = Uri::from_str("https://www.rust-lang.org/test/t33").unwrap();
let req3: Request<Full<Bytes>> = Request::get(uri).body(Full::default()).unwrap();
request_parser_joiner!(JoinedReqParser, TestParser1, TestParser2);
assert_eq!(JoinedReqParser::parse_operation_id(&req1), Some("t11"));
assert_eq!(JoinedReqParser::parse_operation_id(&req2), Some("t22"));
assert_eq!(JoinedReqParser::parse_operation_id(&req3), None);
}
}