use super::RequestContext;
use crate::data_map::SharedDataMap;
use hyper::{Body, HeaderMap, Method, Request, Uri, Version};
use std::fmt::{self, Debug, Formatter};
use std::sync::Arc;
#[derive(Clone)]
pub struct RequestInfo {
pub(crate) req_info_inner: Arc<RequestInfoInner>,
pub(crate) shared_data_maps: Option<Vec<SharedDataMap>>,
pub(crate) context: RequestContext,
}
#[derive(Debug)]
pub(crate) struct RequestInfoInner {
headers: HeaderMap,
method: Method,
uri: Uri,
version: Version,
}
impl RequestInfo {
pub(crate) fn new_from_req(req: &Request<Body>, ctx: RequestContext) -> Self {
let inner = RequestInfoInner {
headers: req.headers().clone(),
method: req.method().clone(),
uri: req.uri().clone(),
version: req.version(),
};
RequestInfo {
req_info_inner: Arc::new(inner),
shared_data_maps: None,
context: ctx,
}
}
pub fn headers(&self) -> &HeaderMap {
&self.req_info_inner.headers
}
pub fn method(&self) -> &Method {
&self.req_info_inner.method
}
pub fn uri(&self) -> &Uri {
&self.req_info_inner.uri
}
pub fn version(&self) -> Version {
self.req_info_inner.version
}
pub fn data<T: Send + Sync + 'static>(&self) -> Option<&T> {
if let Some(ref shared_data_maps) = self.shared_data_maps {
for shared_data_map in shared_data_maps.iter() {
if let Some(data) = shared_data_map.inner.get::<T>() {
return Some(data);
}
}
}
None
}
pub fn context<T: Send + Sync + Clone + 'static>(&self) -> Option<T> {
self.context.get::<T>()
}
}
impl Debug for RequestInfo {
fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
write!(f, "{:?}", self.req_info_inner)
}
}